Temporary driver for testing

Author: Diptorup Deb

driver.py

@@ -0,0 +1,35 @@
+import dpnp
+from numba.core import types
+from numba.core.extending import intrinsic, overload
+
+import numba_dpex as dpex
+import numba_dpex.experimental as dpex_exp
+from numba_dpex.experimental.target import DPEX_KERNEL_EXP_TARGET_NAME
+
+
+# pylint: disable=W0613
+def scalar_add(a, b):
+    return a + b
+
+
+@overload(scalar_add, target=DPEX_KERNEL_EXP_TARGET_NAME)
+def _ol_scalar_add(a, b):
+    def ol_scalar_add_impl(a, b):
+        return a + b
+
+    return ol_scalar_add_impl
+
+
+@dpex_exp.kernel
+def test_overload_call(a, b, c):
+    i = dpex.get_global_id(0)
+    c[i] = scalar_add(a[i], b[i])
+
+
+a = dpnp.ones(10, dtype=dpnp.int64)
+b = dpnp.ones(10, dtype=dpnp.int64)
+c = dpnp.zeros(10, dtype=dpnp.int64)
+
+dpex_exp.call_kernel(test_overload_call, dpex.Range(10), a, b, c)
+
+print(c)