Address comments.

Signed-off-by: MaheshRavishankar <[email protected]>
iree-org · Dec 6, 2024 · b8eb838 · b8eb838
1 parent cf2c77c
commit b8eb838
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/compiler/src/iree/compiler/Codegen/LLVMGPU/KernelConfig.cpp b/compiler/src/iree/compiler/Codegen/LLVMGPU/KernelConfig.cpp
@@ -1620,7 +1620,7 @@ setWarpReductionConfig(IREE::GPU::TargetAttr target,
 
   // Without any bounds on dynamic dims, we need specialization to
   // get peak performance. For now, just use the warp size.
-  if (numDynamicDims) {
+  if (numDynamicDims > 0) {
     SmallVector<int64_t> reductionTileSizes(op.getNumLoops(), 0);
     int64_t preferredSubgroupSize = target.getPreferredSubgroupSize();
     reductionTileSizes[reductionDims[0]] = preferredSubgroupSize;