Skip to content

Commit 899600b

Browse files
[SYCL][NFC] Unify setting kernel arguments (#1379)
Signed-off-by: Alexey Sachkov <[email protected]>
1 parent ce53521 commit 899600b

File tree

1 file changed

+7
-9
lines changed

1 file changed

+7
-9
lines changed

sycl/source/detail/scheduler/commands.cpp

+7-9
Original file line numberDiff line numberDiff line change
@@ -1649,16 +1649,14 @@ cl_int ExecCGCommand::enqueueImp() {
16491649
case kernel_param_kind_t::kind_accessor: {
16501650
Requirement *Req = (Requirement *)(Arg.MPtr);
16511651
AllocaCommandBase *AllocaCmd = getAllocaForReq(Req);
1652-
#if USE_PI_CUDA
1653-
pi_mem MemArg = (pi_mem)AllocaCmd->getMemAllocation();
1654-
Plugin.call<PiApiKind::piextKernelSetArgMemObj>(Kernel, Arg.MIndex, &MemArg);
1655-
#else
16561652
RT::PiMem MemArg = (RT::PiMem)AllocaCmd->getMemAllocation();
1657-
Plugin.call<PiApiKind::piKernelSetArg>(Kernel, Arg.MIndex,
1658-
sizeof(RT::PiMem), &MemArg);
1659-
Plugin.call<PiApiKind::piKernelSetArg>(Kernel, Arg.MIndex,
1660-
sizeof(RT::PiMem), &MemArg);
1661-
#endif
1653+
if (RT::useBackend(pi::Backend::SYCL_BE_PI_OPENCL)) {
1654+
Plugin.call<PiApiKind::piKernelSetArg>(Kernel, Arg.MIndex,
1655+
sizeof(RT::PiMem), &MemArg);
1656+
} else {
1657+
Plugin.call<PiApiKind::piextKernelSetArgMemObj>(Kernel, Arg.MIndex,
1658+
&MemArg);
1659+
}
16621660
break;
16631661
}
16641662
case kernel_param_kind_t::kind_std_layout: {

0 commit comments

Comments
 (0)