fixing eager logic for var

nv-legate · Nov 30, 2023 · 18a6fe8 · 18a6fe8
1 parent 4ef0749
commit 18a6fe8
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/cunumeric/eager.py b/cunumeric/eager.py
@@ -1532,7 +1532,9 @@ def unary_reduction(
                 squared,
                 out=self.array,
                 axis=orig_axis,
-                where=where,
+                where=where
+                if not isinstance(where, EagerArray)
+                else where.array,
                 keepdims=keepdims,
             )
         elif op == UnaryRedCode.VARIANCE:
@@ -1542,7 +1544,9 @@ def unary_reduction(
             np.sum(
                 squares,
                 axis=orig_axis,
-                where=where,
+                where=where
+                if not isinstance(where, EagerArray)
+                else where.array,
                 keepdims=keepdims,
                 out=self.array,
             )