Skip to content

Commit 1ef765a

Browse files
committed
Keep SPMD working
1 parent 12dae84 commit 1ef765a

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

offload/plugins-nextgen/common/src/PluginInterface.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -633,7 +633,7 @@ uint32_t GenericKernelTy::getNumThreads(GenericDeviceTy &GenericDevice,
633633
assert(ThreadLimitClause[1] == 1 && ThreadLimitClause[2] == 1 &&
634634
"Multi dimensional launch not supported yet.");
635635

636-
if (ThreadLimitClause[0] > 0 && false)
636+
if (ThreadLimitClause[0] > 0 && (isGenericMode() || isGenericSPMDMode()))
637637
ThreadLimitClause[0] += GenericDevice.getWarpSize();
638638

639639
return std::min(MaxNumThreads, (ThreadLimitClause[0] > 0)

0 commit comments

Comments
 (0)