File tree Expand file tree Collapse file tree 1 file changed +3
-7
lines changed
unified-runtime/source/adapters/level_zero/v2 Expand file tree Collapse file tree 1 file changed +3
-7
lines changed Original file line number Diff line number Diff 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}
You can’t perform that action at this time.
0 commit comments