Skip to content

Commit 4b4747e

Browse files
author
Diptorup Deb
committed
Add launcher overloads to CPU target. Minor changes to dispatcher internals.
1 parent 4b1101c commit 4b4747e

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

numba_dpex/experimental/kernel_dispatcher.py

Lines changed: 7 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@
2727

2828
_KernelCompileResult = namedtuple(
2929
"_KernelCompileResult",
30-
["status", "cres_or_error", "kernel_module"],
30+
["status", "cres_or_error", "entry_point"],
3131
)
3232

3333

@@ -155,7 +155,7 @@ def _compile_cached(
155155
"w",
156156
encoding="UTF-8",
157157
) as f:
158-
f.write(kernel_cres.library._final_module.__str__())
158+
f.write(kernel_cres.library._final_module)
159159

160160
except errors.TypingError as e:
161161
self._failed_cache[key] = e
@@ -246,9 +246,9 @@ def typeof_pyval(self, val):
246246
self._types_active_call.append(tp)
247247
return tp
248248

249-
def add_bitcode_overload(self, cres, kernel_module):
249+
def add_overload(self, cres):
250250
args = tuple(cres.signature.args)
251-
self.overloads[args] = kernel_module
251+
self.overloads[args] = cres.entry_point
252252

253253
def compile(self, sig) -> _KernelCompileResult:
254254
disp = self._get_dispatcher_for_current_target()
@@ -311,13 +311,11 @@ def folded(args, kws):
311311
)[1]
312312

313313
raise e.bind_fold_arguments(folded)
314-
self.add_bitcode_overload(
315-
kcres.cres_or_error, kcres.kernel_module
316-
)
314+
self.add_overload(kcres.cres_or_error)
317315

318-
# FIXME: enable caching
316+
# TODO: enable caching of kernel_module
319317

320-
return kcres.kernel_module
318+
return kcres.entry_point
321319

322320
def __getitem__(self, args):
323321
"""Square-bracket notation for configuring launch arguments is not

0 commit comments

Comments
 (0)