From b8eb838ac878b7e423718f3cc47f85a6ab8a2d8e Mon Sep 17 00:00:00 2001
From: MaheshRavishankar <mahesh.ravishankar@gmail.com>
Date: Fri, 6 Dec 2024 14:39:30 -0800
Subject: [PATCH] Address comments.

Signed-off-by: MaheshRavishankar <mahesh.ravishankar@gmail.com>
---
 compiler/src/iree/compiler/Codegen/LLVMGPU/KernelConfig.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/compiler/src/iree/compiler/Codegen/LLVMGPU/KernelConfig.cpp b/compiler/src/iree/compiler/Codegen/LLVMGPU/KernelConfig.cpp
index debbf0fefcd6..67e7236c2e21 100644
--- 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;