|
8 | 8 | import dpnp
|
9 | 9 | from numba.core import compiler
|
10 | 10 |
|
11 |
| -import numba_dpex as dpex |
| 11 | +import numba_dpex.kernel_api as kapi |
12 | 12 |
|
13 | 13 | from .kernel_template_iface import KernelTemplateInterface
|
14 | 14 |
|
@@ -112,7 +112,7 @@ def _generate_kernel_stub_as_string(self):
|
112 | 112 | gufunc_txt += (
|
113 | 113 | " stride0 = local_size0 // 2\n"
|
114 | 114 | + " while stride0 > 0:\n"
|
115 |
| - + " dpex.barrier(dpex.LOCAL_MEM_FENCE)\n" |
| 115 | + + " kapi.group_barrier(group)\n" |
116 | 116 | + " if local_id0 < stride0:\n"
|
117 | 117 | )
|
118 | 118 |
|
@@ -158,7 +158,7 @@ def _generate_kernel_ir(self):
|
158 | 158 | Returns: The Numba functionIR object for the compiled kernel_txt string.
|
159 | 159 |
|
160 | 160 | """
|
161 |
| - globls = {"dpnp": dpnp, "dpex": dpex} |
| 161 | + globls = {"dpnp": dpnp, "kapi": kapi} |
162 | 162 | locls = {}
|
163 | 163 | exec(self._kernel_txt, globls, locls)
|
164 | 164 | kernel_fn = locls[self._kernel_name]
|
@@ -317,7 +317,7 @@ def _generate_kernel_ir(self):
|
317 | 317 |
|
318 | 318 | """
|
319 | 319 |
|
320 |
| - globls = {"dpnp": dpnp, "dpex": dpex} |
| 320 | + globls = {"dpnp": dpnp, "kapi": kapi} |
321 | 321 | locls = {}
|
322 | 322 | exec(self._kernel_txt, globls, locls)
|
323 | 323 | kernel_fn = locls[self._kernel_name]
|
|
0 commit comments