Supporting "where" for unary operations (#1061)

cunumeric/_ufunc/ufunc.py

@@ -1,4 +1,4 @@
-# Copyright 2021-2022 NVIDIA Corporation
+# Copyright 2021-2023 NVIDIA Corporation
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
 # you may not use this file except in compliance with the License.
@@ -14,12 +14,17 @@
 #
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any, Dict, Sequence, Union
+from typing import TYPE_CHECKING, Any, Dict, Optional, Sequence, Union
 
 import numpy as np
 from legate.core.utils import OrderedSet
 
-from ..array import check_writeable, convert_to_cunumeric_ndarray, ndarray
+from ..array import (
+    add_boilerplate,
+    check_writeable,
+    convert_to_cunumeric_ndarray,
+    ndarray,
+)
 from ..config import BinaryOpCode, UnaryOpCode, UnaryRedCode
 from ..types import NdShape
 
@@ -680,6 +685,7 @@ def __call__(
 
         return self._maybe_cast_output(out, result)
 
+    @add_boilerplate("array")
     def reduce(
         self,
         array: ndarray,
@@ -688,7 +694,7 @@ def reduce(
         out: Union[ndarray, None] = None,
         keepdims: bool = False,
         initial: Union[Any, None] = None,
-        where: bool = True,
+        where: Optional[ndarray] = None,
     ) -> ndarray:
         """
         reduce(array, axis=0, dtype=None, out=None, keepdims=False, initial=<no
@@ -742,16 +748,10 @@ def reduce(
         --------
         numpy.ufunc.reduce
         """
-        array = convert_to_cunumeric_ndarray(array)
-
         if self._red_code is None:
             raise NotImplementedError(
                 f"reduction for {self} is not yet implemented"
             )
-        if not isinstance(where, bool) or not where:
-            raise NotImplementedError(
-                "the 'where' keyword is not yet supported"
-            )
 
         # NumPy seems to be using None as the default axis value for scalars
         if array.ndim == 0 and axis == 0: