[Testing] Add some numpy array testing for the CUDA target

numba_cuda/numba/cuda/__init__.py

@@ -68,3 +68,9 @@
     )
 
 from numba.cuda.np.ufunc import vectorize, guvectorize
+
+# Re-export typeof
+from numba.cuda.misc.special import (
+    literally,
+    literal_unroll,
+)

numba_cuda/numba/cuda/simulator/api.py

@@ -160,3 +160,7 @@ def jitwrapper(fn):
 def defer_cleanup():
     # No effect for simulator
     yield
+
+
+class grid(object):
+    pass

numba_cuda/numba/cuda/simulator/memory_management/nrt.py

@@ -2,6 +2,9 @@
 # SPDX-License-Identifier: BSD-2-Clause
 
 from numba.cuda import config
+from collections import namedtuple
+
+_nrt_mstats = namedtuple("nrt_mstats", ["alloc", "free", "mi_alloc", "mi_free"])
 
 
 class RTSys:
@@ -11,9 +14,12 @@ def __init__(self, *args, **kwargs):
     def memsys_enable_stats(self):
         pass
 
-    def get_allocation_stats(self):
+    def memsys_disable_stats(self):
         pass
 
+    def get_allocation_stats(self):
+        return _nrt_mstats(alloc=0, free=0, mi_alloc=0, mi_free=0)
+
 
 rtsys = RTSys()
 

numba_cuda/numba/cuda/target.py

@@ -36,7 +36,7 @@
 # Typing
 
 
-class CUDATypingContext(typing.BaseContext):
+class CUDATypingContext(typing.Context):
     def load_additional_registries(self):
         from . import (
             cudadecl,
@@ -58,6 +58,7 @@ def load_additional_registries(self):
         self.install_registry(vector_types.typing_registry)
         self.install_registry(fp16.typing_registry)
         self.install_registry(bf16.typing_registry)
+        self.install_registry(npydecl.registry)
 
     def resolve_value_type(self, val):
         # treat other dispatcher object as another device function
@@ -176,8 +177,13 @@ def load_additional_registries(self):
         from numba.cuda.cpython import builtins as cpython_builtins
         from numba.cuda.core import optional  # noqa: F401
         from numba.cuda.misc import cffiimpl
-        from numba.cuda.np import arrayobj, npdatetime, polynomial, arraymath
-
+        from numba.cuda.np import (
+            arrayobj,
+            npdatetime,
+            polynomial,
+            arraymath,
+            npyimpl,
+        )
         from . import (
             cudaimpl,
             fp16,
@@ -218,6 +224,7 @@ def load_additional_registries(self):
         self.install_registry(npdatetime.registry)
         self.install_registry(arrayobj.registry)
         self.install_registry(arraymath.registry)
+        self.install_registry(npyimpl.registry)
 
         # Install only implementations that are defined outside of numba (i.e.,
         # in third-party extensions) from Numba's builtin_registry.