Move experimental.kernel_iface into top-level module.

Diptorup Deb · Diptorup Deb · commit 803c1803b5ef · 2024-02-01T19:01:14.000-06:00
diff --git a/numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_atomic_ref_overloads.py b/numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_atomic_ref_overloads.py
@@ -15,9 +15,14 @@
 from numba_dpex.core.targets.kernel_target import CC_SPIR_FUNC, LLVM_SPIRV_ARGS
 from numba_dpex.core.types import USMNdArray
 from numba_dpex.experimental.flag_enum import FlagEnum
+from numba_dpex.kernel_iface import (
+    AddressSpace,
+    AtomicRef,
+    MemoryOrder,
+    MemoryScope,
+)
 
 from ..dpcpp_types import AtomicRefType
-from ..kernel_iface import AddressSpace, AtomicRef, MemoryOrder, MemoryScope
 from ..target import DPEX_KERNEL_EXP_TARGET_NAME
 from ._spv_atomic_inst_helper import (
     get_atomic_inst_name,
diff --git a/numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_spv_atomic_inst_helper.py b/numba_dpex/experimental/_kernel_dpcpp_spirv_overloads/_spv_atomic_inst_helper.py
@@ -8,7 +8,7 @@
 
 from numba.core import types
 
-from ..kernel_iface import MemoryOrder, MemoryScope
+from numba_dpex.kernel_iface import MemoryOrder, MemoryScope
 
 
 class _SpvScope(IntEnum):
diff --git a/numba_dpex/kernel_iface/__init__.py b/numba_dpex/kernel_iface/__init__.py
@@ -2,7 +2,8 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-"""The kernel_iface provides a set of Python classes and functions that are
+"""
+The kernel_iface provides a set of Python classes and functions that are
 analogous to the C++ SYCL API. The kernel_iface API is meant to allow
 prototyping SYCL-like kernels in pure Python before compiling them using
 numba_dpex.kernel.
diff --git a/numba_dpex/kernel_iface/atomic_ref.py b/numba_dpex/kernel_iface/atomic_ref.py
diff --git a/numba_dpex/kernel_iface/memory_enums.py b/numba_dpex/kernel_iface/memory_enums.py
diff --git a/numba_dpex/tests/experimental/kernel_iface/spv_overloads/test_atomic_fetch_phi.py b/numba_dpex/tests/experimental/kernel_iface/spv_overloads/test_atomic_fetch_phi.py
@@ -8,7 +8,7 @@
 
 import numba_dpex as dpex
 import numba_dpex.experimental as dpex_exp
-from numba_dpex.experimental.kernel_iface import AtomicRef
+from numba_dpex.kernel_iface import AtomicRef
 from numba_dpex.tests._helper import get_all_dtypes
 
 list_of_supported_dtypes = get_all_dtypes(
diff --git a/numba_dpex/tests/experimental/kernel_iface/spv_overloads/test_atomic_load_store.py b/numba_dpex/tests/experimental/kernel_iface/spv_overloads/test_atomic_load_store.py
@@ -9,7 +9,7 @@
 
 import numba_dpex as dpex
 import numba_dpex.experimental as dpex_exp
-from numba_dpex.experimental.kernel_iface import AtomicRef
+from numba_dpex.kernel_iface import AtomicRef
 from numba_dpex.tests._helper import get_all_dtypes
 
 list_of_supported_dtypes = get_all_dtypes(
diff --git a/numba_dpex/tests/experimental/kernel_iface/spv_overloads/test_atomic_ref.py b/numba_dpex/tests/experimental/kernel_iface/spv_overloads/test_atomic_ref.py
@@ -8,7 +8,7 @@
 
 import numba_dpex as dpex
 import numba_dpex.experimental as dpex_exp
-from numba_dpex.experimental.kernel_iface import AddressSpace, AtomicRef
+from numba_dpex.kernel_iface import AddressSpace, AtomicRef
 
 
 def test_atomic_ref_compilation():
diff --git a/numba_dpex/tests/experimental/kernel_iface/test_memory_enum_compilation.py b/numba_dpex/tests/experimental/kernel_iface/test_memory_enum_compilation.py
@@ -6,11 +6,7 @@
 
 import numba_dpex.experimental as exp_dpex
 from numba_dpex import Range
-from numba_dpex.experimental.kernel_iface import (
-    AddressSpace,
-    MemoryOrder,
-    MemoryScope,
-)
+from numba_dpex.kernel_iface import AddressSpace, MemoryOrder, MemoryScope
 
 
 def test_compilation_of_memory_order():

Original file line number	Diff line number	Diff line change
`@@ -2,7 +2,8 @@`
`2`	`2`	`#`
`3`	`3`	`# SPDX-License-Identifier: Apache-2.0`
`4`	`4`
`5`		`-"""The kernel_iface provides a set of Python classes and functions that are`
	`5`	`+"""`
	`6`	`+The kernel_iface provides a set of Python classes and functions that are`
`6`	`7`	`analogous to the C++ SYCL API. The kernel_iface API is meant to allow`
`7`	`8`	`prototyping SYCL-like kernels in pure Python before compiling them using`
`8`	`9`	`numba_dpex.kernel.`