Skip to content

Commit fb8005d

Browse files
authored
Merge pull request #1426 from IntelPython/remove/barrier_in_reduction_template
Update reduction_template to use kapi.group_barrier
2 parents 6e0d9bb + 799b1bc commit fb8005d

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

numba_dpex/core/parfors/kernel_templates/reduction_template.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@
88
import dpnp
99
from numba.core import compiler
1010

11-
import numba_dpex as dpex
11+
import numba_dpex.kernel_api as kapi
1212

1313
from .kernel_template_iface import KernelTemplateInterface
1414

@@ -112,7 +112,7 @@ def _generate_kernel_stub_as_string(self):
112112
gufunc_txt += (
113113
" stride0 = local_size0 // 2\n"
114114
+ " while stride0 > 0:\n"
115-
+ " dpex.barrier(dpex.LOCAL_MEM_FENCE)\n"
115+
+ " kapi.group_barrier(group)\n"
116116
+ " if local_id0 < stride0:\n"
117117
)
118118

@@ -158,7 +158,7 @@ def _generate_kernel_ir(self):
158158
Returns: The Numba functionIR object for the compiled kernel_txt string.
159159
160160
"""
161-
globls = {"dpnp": dpnp, "dpex": dpex}
161+
globls = {"dpnp": dpnp, "kapi": kapi}
162162
locls = {}
163163
exec(self._kernel_txt, globls, locls)
164164
kernel_fn = locls[self._kernel_name]
@@ -317,7 +317,7 @@ def _generate_kernel_ir(self):
317317
318318
"""
319319

320-
globls = {"dpnp": dpnp, "dpex": dpex}
320+
globls = {"dpnp": dpnp, "kapi": kapi}
321321
locls = {}
322322
exec(self._kernel_txt, globls, locls)
323323
kernel_fn = locls[self._kernel_name]

0 commit comments

Comments
 (0)