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

Author: atmnp

numba_cuda/numba/cuda/__init__.py

@@ -60,3 +60,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/np/arrayobj.py

@@ -16,9 +16,9 @@
 
 import numpy as np
 
-from numba import pndindex, literal_unroll
-from numba.core import types, errors
-from numba.cuda import typing
+from numba import pndindex
+from numba.cuda import literal_unroll
+from numba.core import types, typing, errors
 from numba.cuda import cgutils, extending
 from numba.cuda.np.numpy_support import (
     as_dtype,

numba_cuda/numba/cuda/simulator/api.py

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

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,
@@ -46,7 +46,7 @@ def load_additional_registries(self):
             libdevicedecl,
             vector_types,
         )
-        from numba.cuda.typing import enumdecl, cffi_utils
+        from numba.cuda.typing import enumdecl, cffi_utils, npydecl
 
         self.install_registry(cudadecl.registry)
         self.install_registry(cffi_utils.registry)
@@ -57,6 +57,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
@@ -182,6 +183,8 @@ def load_additional_registries(self):
             arrayobj,
             npdatetime,
             polynomial,
+            arraymath,
+            npyimpl,
         )
         from . import (
             cudaimpl,
@@ -222,6 +225,8 @@ def load_additional_registries(self):
         self.install_registry(polynomial.registry)
         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.