From f1659ec6e800de1d3ebc05716b7eeaaf73cf9d9d Mon Sep 17 00:00:00 2001 From: Masahiro Masuda Date: Thu, 10 Feb 2022 02:22:36 +0900 Subject: [PATCH] [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 cc18de25ee9e..262340099cc7 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); }