File tree Expand file tree Collapse file tree 2 files changed +8
-0
lines changed
unified-runtime/source/loader/layers/sanitizer/tsan Expand file tree Collapse file tree 2 files changed +8
-0
lines changed Original file line number Diff line number Diff line change @@ -1326,6 +1326,8 @@ ur_result_t urGetKernelProcAddrTable(
1326
1326
}
1327
1327
1328
1328
pDdiTable->pfnCreate = ur_sanitizer_layer::tsan::urKernelCreate;
1329
+ pDdiTable->pfnCreateWithNativeHandle =
1330
+ ur_sanitizer_layer::tsan::urKernelCreateWithNativeHandle;
1329
1331
pDdiTable->pfnRetain = ur_sanitizer_layer::tsan::urKernelRetain;
1330
1332
pDdiTable->pfnRelease = ur_sanitizer_layer::tsan::urKernelRelease;
1331
1333
pDdiTable->pfnSetArgValue = ur_sanitizer_layer::tsan::urKernelSetArgValue;
Original file line number Diff line number Diff line change @@ -106,6 +106,12 @@ ur_result_t ShadowMemoryGPU::Destroy() {
106
106
return UR_RESULT_SUCCESS;
107
107
}
108
108
109
+ if (LocalShadowOffset != 0 ) {
110
+ UR_CALL (getContext ()->urDdiTable .USM .pfnFree (Context,
111
+ (void *)LocalShadowOffset));
112
+ LocalShadowOffset = 0 ;
113
+ }
114
+
109
115
const size_t PageSize = GetVirtualMemGranularity (Context, Device);
110
116
for (auto [MappedPtr, PhysicalMem] : VirtualMemMaps) {
111
117
UR_CALL (getContext ()->urDdiTable .VirtualMem .pfnUnmap (
You can’t perform that action at this time.
0 commit comments