Skip to content

Commit 859b8e6

Browse files
committed
[SYCL][NFC] Unify setting kernel arguments
Signed-off-by: Alexey Sachkov <[email protected]>
1 parent 3baec18 commit 859b8e6

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

sycl/source/detail/scheduler/commands.cpp

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -1639,16 +1639,14 @@ cl_int ExecCGCommand::enqueueImp() {
16391639
case kernel_param_kind_t::kind_accessor: {
16401640
Requirement *Req = (Requirement *)(Arg.MPtr);
16411641
AllocaCommandBase *AllocaCmd = getAllocaForReq(Req);
1642-
#if USE_PI_CUDA
1643-
pi_mem MemArg = (pi_mem)AllocaCmd->getMemAllocation();
1644-
Plugin.call<PiApiKind::piextKernelSetArgMemObj>(Kernel, Arg.MIndex, &MemArg);
1645-
#else
16461642
RT::PiMem MemArg = (RT::PiMem)AllocaCmd->getMemAllocation();
1643+
if (!RT::useBackend(pi::Backend::SYCL_BE_PI_OPENCL)) {
1644+
Plugin.call<PiApiKind::piextKernelSetArgMemObj>(Kernel, Arg.MIndex,
1645+
&MemArg);
1646+
break;
1647+
}
16471648
Plugin.call<PiApiKind::piKernelSetArg>(Kernel, Arg.MIndex,
16481649
sizeof(RT::PiMem), &MemArg);
1649-
Plugin.call<PiApiKind::piKernelSetArg>(Kernel, Arg.MIndex,
1650-
sizeof(RT::PiMem), &MemArg);
1651-
#endif
16521650
break;
16531651
}
16541652
case kernel_param_kind_t::kind_std_layout: {

0 commit comments

Comments
 (0)