[tvm] branch main updated: [OpenCL] Fix vthread_extent for warp size 1 case (#10199)

[zhaowu]

This is an automated email from the ASF dual-hosted git repository.

zhaowu pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git


The following commit(s) were added to refs/heads/main by this push:
     new 69403f1  [OpenCL] Fix vthread_extent for warp size 1 case (#10199)
69403f1 is described below

commit 69403f19db5fbb64a3e8c74c76d8b0fde124c123
Author: Masahiro Masuda <masahi...@gmail.com>
AuthorDate: Thu Feb 10 02:22:36 2022 +0900

    [OpenCL] Fix vthread_extent for warp size 1 case (#10199)
---
 src/auto_scheduler/search_task.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/auto_scheduler/search_task.cc 
b/src/auto_scheduler/search_task.cc
index cc18de2..2623400 100755
--- a/src/auto_scheduler/search_task.cc
+++ b/src/auto_scheduler/search_task.cc
@@ -127,7 +127,7 @@ HardwareParams 
HardwareParamsNode::GetDefaultHardwareParams(const Target& target
             << "Warp size 1 is not recommended for OpenCL devices. Tuning 
might crash or stuck";
       }
 
-      int max_vthread_extent = warp_size / 4;
+      int max_vthread_extent = std::max(1, warp_size / 4);
       return HardwareParams(-1, 16, 64, max_shared_memory_per_block, 
max_local_memory_per_block,
                             max_threads_per_block, max_vthread_extent, 
warp_size);
     }