Reorganize new features that belong to experimental module.

IntelPython · Nov 2, 2023 · a3174fc · a3174fc
1 parent 5777b67
commit a3174fc
Show file tree

Hide file tree

Showing 8 changed files with 17 additions and 5 deletions.
diff --git a/numba_dpex/core/types/__init__.py b/numba_dpex/core/types/__init__.py
@@ -3,7 +3,6 @@
 # SPDX-License-Identifier: Apache-2.0
 
 from .array_type import Array
-from .dpcpp_types import AtomicRefType
 from .dpctl_types import DpctlSyclEvent, DpctlSyclQueue
 from .dpnp_ndarray_type import DpnpNdArray
 from .literal_intenum_type import IntEnumLiteral
@@ -35,7 +34,6 @@
 
 __all__ = [
     "Array",
-    "AtomicRefType",
     "DpctlSyclQueue",
     "DpctlSyclEvent",
     "DpnpNdArray",

diff --git a/numba_dpex/core/typing/typeof.py b/numba_dpex/core/typing/typeof.py
@@ -11,8 +11,8 @@
 from numba_dpex.core.kernel_interface.dpcpp_iface import AtomicRef
 from numba_dpex.utils import address_space
 
+from ...experimental.dpcpp_types import AtomicRefType
 from ..kernel_interface.indexers import NdRange, Range
-from ..types.dpcpp_types import AtomicRefType
 from ..types.dpctl_types import DpctlSyclEvent, DpctlSyclQueue
 from ..types.dpnp_ndarray_type import DpnpNdArray
 from ..types.range_types import NdRangeType, RangeType

diff --git a/numba_dpex/experimental/__init__.py b/numba_dpex/experimental/__init__.py
@@ -9,6 +9,8 @@
 from numba.core.imputils import Registry
 
 from .decorators import kernel
+from .dpcpp_iface import AddressSpace, AtomicRef, MemoryOrder, MemoryScope
+from .dpcpp_types import AtomicRefType
 from .kernel_dispatcher import KernelDispatcher
 from .launcher import call_kernel
 from .models import *
@@ -26,4 +28,13 @@ def dpex_dispatcher_const(context):
     return context.get_dummy_value()
 
 
-__all__ = ["kernel", "KernelDispatcher", "call_kernel"]
+__all__ = [
+    "kernel",
+    "call_kernel",
+    "AddressSpace",
+    "AtomicRef",
+    "AtomicRefType",
+    "KernelDispatcher",
+    "MemoryOrder",
+    "MemoryScope",
+]
diff --git a/.../kernel_interface/dpcpp_iface/__init__.py → ...dpex/experimental/dpcpp_iface/__init__.py b/.../kernel_interface/dpcpp_iface/__init__.py → ...dpex/experimental/dpcpp_iface/__init__.py
@@ -1,7 +1,10 @@
-# SPDX-FileCopyrightText: 2020 - 2023 Intel Corporation
+# SPDX-FileCopyrightText: 2023 Intel Corporation
 #
 # SPDX-License-Identifier: Apache-2.0
 
+"""Python classes that are analogous to dpcpp's SYCL API used to write kernels.
+"""
+
 from .atomic_ref import AtomicRef
 from .memory_enums import AddressSpace, MemoryOrder, MemoryScope
 

diff --git a/...terface/dpcpp_iface/_spv_atomic_helper.py → ...imental/dpcpp_iface/_spv_atomic_helper.py b/...terface/dpcpp_iface/_spv_atomic_helper.py → ...imental/dpcpp_iface/_spv_atomic_helper.py
diff --git a/...ernel_interface/dpcpp_iface/atomic_ref.py → ...ex/experimental/dpcpp_iface/atomic_ref.py b/...ernel_interface/dpcpp_iface/atomic_ref.py → ...ex/experimental/dpcpp_iface/atomic_ref.py
diff --git a/...nel_interface/dpcpp_iface/memory_enums.py → .../experimental/dpcpp_iface/memory_enums.py b/...nel_interface/dpcpp_iface/memory_enums.py → .../experimental/dpcpp_iface/memory_enums.py
diff --git a/numba_dpex/core/types/dpcpp_types.py → numba_dpex/experimental/dpcpp_types.py b/numba_dpex/core/types/dpcpp_types.py → numba_dpex/experimental/dpcpp_types.py