Update unit test to use correct kernel launch API.

IntelPython · Dec 26, 2023 · b539970 · b539970
1 parent e5d53f7
commit b539970
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 9 deletions.
diff --git a/numba_dpex/tests/kernel_tests/test_atomic_op.py b/numba_dpex/tests/kernel_tests/test_atomic_op.py
@@ -96,7 +96,7 @@ def test_kernel_atomic_local(input_arrays, return_list_of_op):
     op_type, expected = return_list_of_op
     f = get_func_local(op_type, dtype)
     kernel = dpex.kernel(f)
-    kernel[dpex.Range(N), dpex.Range(N)](a)
+    kernel[dpex.NdRange(dpex.Range(N), dpex.Range(N))](a)
     assert a[0] == expected
 
 

diff --git a/numba_dpex/tests/kernel_tests/test_barrier.py b/numba_dpex/tests/kernel_tests/test_barrier.py
@@ -2,13 +2,11 @@
 #
 # SPDX-License-Identifier: Apache-2.0
 
-import dpctl
 import dpctl.tensor as dpt
 import numpy as np
-import pytest
 
 import numba_dpex as dpex
-from numba_dpex import float32, usm_ndarray, void
+from numba_dpex import NdRange, Range, float32, usm_ndarray, void
 
 f32arrty = usm_ndarray(ndim=1, dtype=float32, layout="C")
 
@@ -27,7 +25,7 @@ def twice(A):
     orig = dpt.asnumpy(arr)
     global_size = (N,)
     local_size = (N // 2,)
-    twice[global_size, local_size](arr)
+    twice[NdRange(global_size, local_size)](arr)
     after = dpt.asnumpy(arr)
     # The computation is correct?
     np.testing.assert_allclose(orig * 2, after)
@@ -45,7 +43,7 @@ def twice(A):
     N = 256
     arr = dpt.arange(N, dtype=dpt.float32)
     orig = dpt.asnumpy(arr)
-    twice[N](arr)
+    twice[Range(N)](arr)
     after = dpt.asnumpy(arr)
     # The computation is correct?
     np.testing.assert_allclose(orig * 2, after)
@@ -68,7 +66,7 @@ def reverse_array(A):
 
     arr = dpt.arange(blocksize, dtype=dpt.float32)
     orig = dpt.asnumpy(arr)
-    reverse_array[(blocksize,), (blocksize,)](arr)
+    reverse_array[NdRange(Range(blocksize), Range(blocksize))](arr)
     after = dpt.asnumpy(arr)
     expected = orig[::-1] + orig
     np.testing.assert_allclose(expected, after)
diff --git a/numba_dpex/tests/kernel_tests/test_caching.py b/numba_dpex/tests/kernel_tests/test_caching.py
@@ -6,7 +6,6 @@
 
 import dpctl.tensor as dpt
 import numpy as np
-import pytest
 
 import numba_dpex as dpex
 from numba_dpex.core.caching import LRUCache
@@ -130,7 +129,7 @@ def data_parallel_sum(x, y, z):
 
     d = JitKernel(data_parallel_sum)
 
-    d_launcher = d[100]
+    d_launcher = d[dpex.Range(100)]
 
     N = 10
     for i in range(N):