fixing bug in cuda code

nv-legate · Nov 17, 2023 · f504f05 · f504f05
1 parent 2bd62eb
commit f504f05
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/cunumeric/unary/unary_red.cu b/src/cunumeric/unary/unary_red.cu
@@ -282,8 +282,8 @@ static __device__ __forceinline__ Point<DIM> local_reduce(LHS& result,
   if (!domain.contains(point)) return point;
 
   bool mask = true;
-  if constexpr (HAS_WHERE) mask = where[point];
   while (point[collapsed_dim] <= domain.hi[collapsed_dim]) {
+    if constexpr (HAS_WHERE) mask = where[point];
     if (mask) {
       LHS value = OP::convert(point, collapsed_dim, identity, in[point]);
       REDOP::template fold<true>(result, value);