From 77295480d9dd3b5dbef579eb992805483ea28c4f Mon Sep 17 00:00:00 2001
From: Diptorup Deb <diptorup.deb@intel.com>
Date: Tue, 26 Mar 2024 14:44:21 -0500
Subject: [PATCH] Migrate experimental.testing and delete the experimental
 module.

---
 numba_dpex/experimental/__init__.py           |  7 ----
 numba_dpex/experimental/testing.py            | 32 -------------------
 .../tests/kernel_tests/test_async_kernel.py   | 29 +++++++++++++++--
 3 files changed, 26 insertions(+), 42 deletions(-)
 delete mode 100644 numba_dpex/experimental/__init__.py
 delete mode 100644 numba_dpex/experimental/testing.py

diff --git a/numba_dpex/experimental/__init__.py b/numba_dpex/experimental/__init__.py
deleted file mode 100644
index 1266f90601..0000000000
--- a/numba_dpex/experimental/__init__.py
+++ /dev/null
@@ -1,7 +0,0 @@
-# SPDX-FileCopyrightText: 2023 - 2024 Intel Corporation
-#
-# SPDX-License-Identifier: Apache-2.0
-
-"""Contains experimental features that are meant as engineering preview and not
-yet production ready.
-"""
diff --git a/numba_dpex/experimental/testing.py b/numba_dpex/experimental/testing.py
deleted file mode 100644
index 69ce6cdfe0..0000000000
--- a/numba_dpex/experimental/testing.py
+++ /dev/null
@@ -1,32 +0,0 @@
-# SPDX-FileCopyrightText: 2023 - 2024 Intel Corporation
-#
-# SPDX-License-Identifier: Apache-2.0
-
-"""Tools for testing, not intended for regular use."""
-
-
-from numba.core import types
-from numba.extending import intrinsic
-
-from numba_dpex import dpjit
-from numba_dpex.core.runtime.context import DpexRTContext
-from numba_dpex.core.targets.dpjit_target import DPEX_TARGET_NAME
-
-
-@intrinsic(target=DPEX_TARGET_NAME)
-def _kernel_cache_size(
-    typingctx,  # pylint: disable=W0613
-):
-    sig = types.int64()
-
-    def codegen(ctx, builder, sig, llargs):  # pylint: disable=W0613
-        dpexrt = DpexRTContext(ctx)
-        return dpexrt.kernel_cache_size(builder)
-
-    return sig, codegen
-
-
-@dpjit
-def kernel_cache_size() -> int:
-    """Returns kernel cache size."""
-    return _kernel_cache_size()  # pylint: disable=E1120
diff --git a/numba_dpex/tests/kernel_tests/test_async_kernel.py b/numba_dpex/tests/kernel_tests/test_async_kernel.py
index f3801ec537..437f2184f6 100644
--- a/numba_dpex/tests/kernel_tests/test_async_kernel.py
+++ b/numba_dpex/tests/kernel_tests/test_async_kernel.py
@@ -5,13 +5,36 @@
 import dpctl
 import dpnp
 import pytest
+from numba.core import types
 from numba.core.errors import TypingError
+from numba.extending import intrinsic
 
 import numba_dpex as dpex
-from numba_dpex.experimental import testing
+from numba_dpex import dpjit
+from numba_dpex.core.runtime.context import DpexRTContext
+from numba_dpex.core.targets.dpjit_target import DPEX_TARGET_NAME
 from numba_dpex.kernel_api import Item, Range
 
 
+@intrinsic(target=DPEX_TARGET_NAME)
+def _kernel_cache_size(
+    typingctx,  # pylint: disable=W0613
+):
+    sig = types.int64()
+
+    def codegen(ctx, builder, sig, llargs):  # pylint: disable=W0613
+        dpexrt = DpexRTContext(ctx)
+        return dpexrt.kernel_cache_size(builder)
+
+    return sig, codegen
+
+
+@dpjit
+def kernel_cache_size() -> int:
+    """Returns kernel cache size."""
+    return _kernel_cache_size()  # pylint: disable=E1120
+
+
 @dpex.kernel(
     release_gil=False,
     no_compile=True,
@@ -106,8 +129,8 @@ def test_async_dependent_add():
 
 def test_async_add_from_cache():
     test_async_add()  # compile
-    old_size = testing.kernel_cache_size()
+    old_size = kernel_cache_size()
     test_async_add()  # use from cache
-    new_size = testing.kernel_cache_size()
+    new_size = kernel_cache_size()
 
     assert new_size == old_size