File tree Expand file tree Collapse file tree 1 file changed +9
-1
lines changed
source/adapters/level_zero Expand file tree Collapse file tree 1 file changed +9
-1
lines changed Original file line number Diff line number Diff line change @@ -494,11 +494,19 @@ ur_result_t urEnqueueDeviceGlobalVariableWrite(
494
494
) {
495
495
std::scoped_lock<ur_shared_mutex> lock (Queue->Mutex );
496
496
497
+ ze_module_handle_t ZeModule{};
498
+ auto It = Program->ZeModuleMap .find (Queue->Device ->ZeDevice );
499
+ if (It != Program->ZeModuleMap .end ()) {
500
+ ZeModule = It->second ;
501
+ } else {
502
+ ZeModule = Program->ZeModule ;
503
+ }
504
+
497
505
// Find global variable pointer
498
506
size_t GlobalVarSize = 0 ;
499
507
void *GlobalVarPtr = nullptr ;
500
508
ZE2UR_CALL (zeModuleGetGlobalPointer,
501
- (Program-> ZeModule , Name, &GlobalVarSize, &GlobalVarPtr));
509
+ (ZeModule, Name, &GlobalVarSize, &GlobalVarPtr));
502
510
if (GlobalVarSize < Offset + Count) {
503
511
setErrorMessage (" Write device global variable is out of range." ,
504
512
UR_RESULT_ERROR_INVALID_VALUE,
You can’t perform that action at this time.
0 commit comments