Skip to content

Commit 4f4e6b8

Browse files
committed
Remove multi-device condition
1 parent fc827a1 commit 4f4e6b8

File tree

1 file changed

+3
-7
lines changed
  • unified-runtime/source/adapters/level_zero/v2

1 file changed

+3
-7
lines changed

unified-runtime/source/adapters/level_zero/v2/kernel.cpp

Lines changed: 3 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -426,14 +426,10 @@ ur_result_t urKernelSetArgPointer(
426426
TRACK_SCOPE_LATENCY("urKernelSetArgPointer");
427427

428428
std::scoped_lock<ur_shared_mutex> guard(hKernel->Mutex);
429-
// In multi-device context store the raw pointer value and defer setting the
429+
// Store the raw pointer value and defer setting the
430430
// argument until we know the device where kernel is being submitted.
431-
if (hKernel->getContext()->getDevices().size() > 1) {
432-
hKernel->addPendingPointerArgument(argIndex, pArgValue);
433-
return UR_RESULT_SUCCESS;
434-
}
435-
436-
return hKernel->setArgPointer(argIndex, pProperties, pArgValue);
431+
hKernel->addPendingPointerArgument(argIndex, pArgValue);
432+
return UR_RESULT_SUCCESS;
437433
} catch (...) {
438434
return exceptionToResult(std::current_exception());
439435
}

0 commit comments

Comments
 (0)