Remove utils.constants and utils.array_utils

Author: Diptorup Deb

numba_dpex/core/datamodel/models.py

@@ -17,7 +17,7 @@
     DpctlMDLocalAccessorType,
     LocalAccessorType,
 )
-from numba_dpex.utils import address_space
+from numba_dpex.kernel_api.memory_enums import AddressSpace as address_space
 
 from ..types import (
     Array,
@@ -62,7 +62,7 @@ def __init__(self, dmm, fe_type):
         adrsp = (
             fe_type.addrspace
             if fe_type.addrspace is not None
-            else address_space.GLOBAL
+            else address_space.GLOBAL.value
         )
         be_type = dmm.lookup(fe_type.dtype).get_data_type().as_pointer(adrsp)
         super(GenericPointerModel, self).__init__(dmm, fe_type, be_type)

numba_dpex/core/types/kernel_api/local_accessor.py

@@ -8,7 +8,7 @@
 from numba.np import numpy_support
 
 from numba_dpex.core.types import USMNdArray
-from numba_dpex.utils import address_space as AddressSpace
+from numba_dpex.kernel_api.memory_enums import AddressSpace
 
 
 class DpctlMDLocalAccessorType(Type):
@@ -36,14 +36,14 @@ def __init__(self, ndim, dtype):
 
         type_name = (
             f"LocalAccessor(dtype={parsed_dtype}, ndim={ndim}, "
-            f"address_space={AddressSpace.LOCAL})"
+            f"address_space={AddressSpace.LOCAL.value})"
         )
 
         super().__init__(
             ndim=ndim,
             layout="C",
             dtype=parsed_dtype,
-            addrspace=AddressSpace.LOCAL,
+            addrspace=AddressSpace.LOCAL.value,
             name=type_name,
         )
 

numba_dpex/core/types/usm_ndarray_type.py

@@ -13,7 +13,7 @@
 from numba.np.numpy_support import from_dtype
 
 from numba_dpex.core.types.dpctl_types import DpctlSyclQueue
-from numba_dpex.utils.constants import address_space
+from numba_dpex.kernel_api.memory_enums import AddressSpace as address_space
 
 
 class USMNdArray(Array):
@@ -30,7 +30,7 @@ def __init__(
         readonly=False,
         name=None,
         aligned=True,
-        addrspace=address_space.GLOBAL,
+        addrspace=address_space.GLOBAL.value,
     ):
         if (
             queue is not None

numba_dpex/core/typing/typeof.py

@@ -9,8 +9,8 @@
 from numba.np import numpy_support
 
 from numba_dpex.kernel_api import AtomicRef, Group, Item, LocalAccessor, NdItem
+from numba_dpex.kernel_api.memory_enums import AddressSpace as address_space
 from numba_dpex.kernel_api.ranges import NdRange, Range
-from numba_dpex.utils.constants import address_space
 
 from ..types.dpctl_types import DpctlSyclEvent, DpctlSyclQueue
 from ..types.dpnp_ndarray_type import DpnpNdArray
@@ -59,7 +59,7 @@ def _array_typeof_helper(val, array_class_type):
         readonly=readonly,
         usm_type=usm_type,
         queue=ty_queue,
-        addrspace=address_space.GLOBAL,
+        addrspace=address_space.GLOBAL.value,
     )
 
 

numba_dpex/kernel_api_impl/spirv/overloads/_private_array_overloads.py

@@ -20,12 +20,12 @@
 
 from numba_dpex.core.types import USMNdArray
 from numba_dpex.kernel_api import PrivateArray
+from numba_dpex.kernel_api.memory_enums import AddressSpace
 from numba_dpex.kernel_api_impl.spirv.arrayobj import (
     np_cfarray,
     require_literal,
 )
 from numba_dpex.kernel_api_impl.spirv.target import SPIRVTypingContext
-from numba_dpex.utils import address_space as AddressSpace
 
 from ._registry import lower
 
@@ -50,7 +50,7 @@ def typer(shape, dtype, fill_zeros=types.BooleanLiteral(False)):
             dtype=_ty_parse_dtype(dtype),
             ndim=_ty_parse_shape(shape),
             layout="C",
-            addrspace=AddressSpace.PRIVATE,
+            addrspace=AddressSpace.PRIVATE.value,
         )
 
     return typer

numba_dpex/kernel_api_impl/spirv/target.py

@@ -24,9 +24,9 @@
 from numba_dpex.core.types import IntEnumLiteral
 from numba_dpex.core.typing import dpnpdecl
 from numba_dpex.kernel_api.flag_enum import FlagEnum
+from numba_dpex.kernel_api.memory_enums import AddressSpace as address_space
 from numba_dpex.kernel_api_impl.spirv.arrayobj import populate_array
 from numba_dpex.ocl.mathimpl import lower_ocl_impl, sig_mapper
-from numba_dpex.utils import address_space, calling_conv
 
 from . import codegen
 from .overloads._registry import registry as spirv_registry
@@ -367,7 +367,7 @@ def declare_function(self, module, fndesc):
         if not self.enable_debuginfo:
             fn.attributes.add("alwaysinline")
         ret = super().declare_function(module, fndesc)
-        ret.calling_convention = calling_conv.CC_SPIR_FUNC
+        ret.calling_convention = CC_SPIR_FUNC
         return ret
 
     def insert_const_string(self, mod, string):
@@ -392,15 +392,15 @@ def insert_const_string(self, mod, string):
         if gv is None:
             # Not defined yet
             gv = cgutils.add_global_variable(
-                mod, text.type, name=name, addrspace=address_space.GENERIC
+                mod, text.type, name=name, addrspace=address_space.GENERIC.value
             )
             gv.linkage = "internal"
             gv.global_constant = True
             gv.initializer = text
 
         # Cast to a i8* pointer
         charty = gv.type.pointee.element
-        return gv.bitcast(charty.as_pointer(address_space.GENERIC))
+        return gv.bitcast(charty.as_pointer(address_space.GENERIC.value))
 
     def addrspacecast(self, builder, src, addrspace):
         """Insert an LLVM addressspace cast instruction into the module.

numba_dpex/ocl/ocldecl.py

@@ -15,7 +15,7 @@
 
 import numba_dpex as dpex
 from numba_dpex.core.types import Array
-from numba_dpex.utils import address_space
+from numba_dpex.kernel_api import AddressSpace as address_space
 
 registry = Registry()
 intrinsic = registry.register
@@ -158,7 +158,7 @@ def typer(shape, dtype):
                     dtype=nb_dtype,
                     ndim=ndim,
                     layout="C",
-                    addrspace=address_space.LOCAL,
+                    addrspace=address_space.LOCAL.value,
                 )
 
         return typer
@@ -201,7 +201,7 @@ def typer(shape, dtype):
                     dtype=nb_dtype,
                     ndim=ndim,
                     layout="C",
-                    addrspace=address_space.PRIVATE,
+                    addrspace=address_space.PRIVATE.value,
                 )
 
         return typer

numba_dpex/ocl/oclimpl.py

@@ -16,9 +16,9 @@
 from numba_dpex import spirv_kernel_target
 from numba_dpex.core import config
 from numba_dpex.core.types import Array
+from numba_dpex.kernel_api import AddressSpace as address_space
 from numba_dpex.kernel_api_impl.spirv.codegen import SPIR_DATA_LAYOUT
 from numba_dpex.ocl.atomics import atomic_helper
-from numba_dpex.utils import address_space
 
 from . import stubs
 from ._declare_function import _declare_function
@@ -301,7 +301,7 @@ def dpex_private_array_integer(context, builder, sig, args):
         shape=(length,),
         dtype=dtype,
         symbol_name="_dpex_pmem",
-        addrspace=address_space.PRIVATE,
+        addrspace=address_space.PRIVATE.value,
     )
 
 
@@ -316,7 +316,7 @@ def dpex_private_array_tuple(context, builder, sig, args):
         shape=shape,
         dtype=dtype,
         symbol_name="_dpex_pmem",
-        addrspace=address_space.PRIVATE,
+        addrspace=address_space.PRIVATE.value,
     )
 
 
@@ -330,7 +330,7 @@ def dpex_local_array_integer(context, builder, sig, args):
         shape=(length,),
         dtype=dtype,
         symbol_name="_dpex_lmem",
-        addrspace=address_space.LOCAL,
+        addrspace=address_space.LOCAL.value,
     )
 
 
@@ -345,7 +345,7 @@ def dpex_local_array_tuple(context, builder, sig, args):
         shape=shape,
         dtype=dtype,
         symbol_name="_dpex_lmem",
-        addrspace=address_space.LOCAL,
+        addrspace=address_space.LOCAL.value,
     )
 
 
@@ -358,7 +358,7 @@ def _generic_array(context, builder, shape, dtype, symbol_name, addrspace):
     lldtype = context.get_data_type(dtype)
     laryty = llvmir.ArrayType(lldtype, elemcount)
 
-    if addrspace == address_space.LOCAL:
+    if addrspace == address_space.LOCAL.value:
         lmod = builder.module
 
         # Create global variable in the requested address-space
@@ -374,7 +374,7 @@ def _generic_array(context, builder, shape, dtype, symbol_name, addrspace):
         if dtype not in types.number_domain:
             raise TypeError("unsupported type: %s" % dtype)
 
-    elif addrspace == address_space.PRIVATE:
+    elif addrspace == address_space.PRIVATE.value:
         gvmem = cgutils.alloca_once(builder, laryty, name=symbol_name)
     else:
         raise NotImplementedError("addrspace {addrspace}".format(**locals()))
@@ -397,7 +397,7 @@ def _make_array(
     dtype,
     shape,
     layout="C",
-    addrspace=address_space.GENERIC,
+    addrspace=address_space.GENERIC.value,
 ):
     ndim = len(shape)
     # Create array object

numba_dpex/printimpl.py

@@ -8,15 +8,15 @@
 from numba.core import cgutils, types
 from numba.core.imputils import Registry
 
-from numba_dpex.utils import address_space
+from numba_dpex.kernel_api.memory_enums import AddressSpace as address_space
 
 registry = Registry()
 lower = registry.lower
 
 
 def declare_print(lmod):
     voidptrty = llvmir.PointerType(
-        llvmir.IntType(8), addrspace=address_space.GENERIC
+        llvmir.IntType(8), addrspace=address_space.GENERIC.value
     )
     printfty = llvmir.FunctionType(
         llvmir.IntType(32), [voidptrty], var_arg=True

numba_dpex/tests/core/test_itanium_mangler_extension.py

@@ -7,7 +7,7 @@
 from numba.core import types
 
 import numba_dpex.core.utils.itanium_mangler as itanium_mangler
-from numba_dpex.utils import address_space
+from numba_dpex.kernel_api import AddressSpace as address_space
 
 list_of_dtypes = [
     (int32, "i"),
@@ -25,10 +25,10 @@ def dtypes(request):
 
 
 list_of_addrspaces = [
-    (address_space.PRIVATE, "3AS0"),
-    (address_space.GLOBAL, "3AS1"),
-    (address_space.LOCAL, "3AS3"),
-    (address_space.GENERIC, "3AS4"),
+    (address_space.PRIVATE.value, "3AS0"),
+    (address_space.GLOBAL.value, "3AS1"),
+    (address_space.LOCAL.value, "3AS3"),
+    (address_space.GENERIC.value, "3AS4"),
 ]