Skip to content

Commit 1ab07a8

Browse files
author
Diptorup Deb
committed
Use experimental target descriptor in kernel_dispatcher
1 parent 1d4248c commit 1ab07a8

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

numba_dpex/experimental/kernel_dispatcher.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,14 +16,15 @@
1616
from numba.core.typing.typeof import Purpose, typeof
1717

1818
from numba_dpex import config, spirv_generator
19-
from numba_dpex.core.descriptor import dpex_kernel_target
2019
from numba_dpex.core.exceptions import (
2120
ExecutionQueueInferenceError,
2221
UnsupportedKernelArgumentError,
2322
)
2423
from numba_dpex.core.pipelines import kernel_compiler
2524
from numba_dpex.core.types import DpnpNdArray
2625

26+
from .target import dpex_exp_kernel_target
27+
2728
_KernelModule = namedtuple("_KernelModule", ["kernel_name", "kernel_bitcode"])
2829

2930
_KernelCompileResult = namedtuple(
@@ -177,7 +178,7 @@ class KernelDispatcher(Dispatcher):
177178
178179
"""
179180

180-
targetdescr = dpex_kernel_target
181+
targetdescr = dpex_exp_kernel_target
181182
_fold_args = False
182183

183184
Dispatcher._impl_kinds["kernel"] = _KernelCompiler

0 commit comments

Comments
 (0)