|
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