Activate the NumPy fallback for cunumeric.random in CPU build (nv-leg…

…ate#485)

* Activate the NumPy fallback for cunumeric.random in CPU build

* Fix for mypy errors

cunumeric/config.py

@@ -264,6 +264,9 @@ class _CunumericSharedLib:
     CUNUMERIC_WRITE: int
     CUNUMERIC_ZIP: int
 
+    def cunumeric_has_curand(self) -> int:
+        ...
+
 
 # Load the cuNumeric library first so we have a shard object that
 # we can use to initialize all these configuration enumerations

cunumeric/random/__init__.py

@@ -15,10 +15,13 @@
 from __future__ import annotations
 
 import numpy.random as _nprandom
-from cunumeric.random.random import *
 from cunumeric.coverage import clone_module
-from cunumeric.random.bitgenerator import *
-from cunumeric.random.generator import *
+from cunumeric.runtime import runtime
+
+if runtime.has_curand:
+    from cunumeric.random.random import *
+    from cunumeric.random.bitgenerator import *
+    from cunumeric.random.generator import *
 
 clone_module(_nprandom, globals())
 

cunumeric/runtime.py

@@ -162,6 +162,8 @@ def __init__(self, legate_context: LegateContext) -> None:
         # Make sure that our CuNumericLib object knows about us so it can
         # destroy us
         cunumeric_lib.set_runtime(self)
+        assert cunumeric_lib.shared_object is not None
+        self.has_curand = cunumeric_lib.shared_object.cunumeric_has_curand()
         self._register_dtypes()
 
         self.args = parse_command_args("cunumeric", ARGS)

src/cunumeric/cunumeric.cc

@@ -83,4 +83,13 @@ void cunumeric_perform_registration(void)
     ctx, CUNUMERIC_TUNABLE_HAS_NUMAMEM, cunumeric::CuNumeric::mapper_id);
   if (fut.get_result<int32_t>() != 0) cunumeric::CuNumeric::has_numamem = true;
 }
+
+bool cunumeric_has_curand()
+{
+#ifdef LEGATE_USE_CUDA
+  return true;
+#else
+  return false;
+#endif
+}
 }

src/cunumeric/cunumeric_c.h

@@ -328,6 +328,7 @@ extern "C" {
 #endif
 
 void cunumeric_perform_registration();
+bool cunumeric_has_curand();
 
 #ifdef __cplusplus
 }

src/cunumeric/unary/isnan.h

@@ -38,6 +38,10 @@ __CUDA_HD__ bool is_nan(const complex<T>& x)
   return std::isnan(x.imag()) || std::isnan(x.real());
 }
 
-__CUDA_HD__ inline bool is_nan(const __half& x) { return isnan(x); }
+__CUDA_HD__ inline bool is_nan(const __half& x)
+{
+  using std::isnan;
+  return isnan(x);
+}
 
 }  // namespace cunumeric