Merge branch 'branch-24.01' of github.com:nv-legate/cunumeric into en…

…h-ext-rng

CMakeLists.txt

@@ -55,7 +55,7 @@ include(rapids-cuda)
 include(rapids-export)
 include(rapids-find)
 
-set(cunumeric_version 23.11.00)
+set(cunumeric_version 24.01.00)
 
 # For now we want the optimization flags to match on both normal make and cmake
 # builds so we override the cmake defaults here for release, this changes

cmake/versions.json

@@ -1,11 +1,11 @@
 {
   "packages" : {
     "legate_core" : {
-      "version": "23.11.00",
+      "version": "24.01.00",
       "git_url" : "https://github.com/nv-legate/legate.core.git",
       "git_shallow": false,
       "always_download": false,
-      "git_tag" : "c05ebdab30dee6ac2d6e2808fb835fad0302822d"
+      "git_tag" : "08da13fc544f3db26bf1ef7ce9bdb85e72a9d9fb"
     }
   }
 }

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: