Skip to content

Commit 21de688

Browse files
committed
Merge pull request #2586 from AllanZyne/review/yang/fix_kernel_release
[DeviceASAN] Fix kernel release order
1 parent a634393 commit 21de688

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

source/loader/layers/sanitizer/asan/asan_ddi.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1378,12 +1378,12 @@ __urdlllocal ur_result_t urKernelRelease(
13781378
}
13791379

13801380
getContext()->logger.debug("==== urKernelRelease");
1381-
UR_CALL(pfnRelease(hKernel));
13821381

13831382
auto &KernelInfo = getAsanInterceptor()->getOrCreateKernelInfo(hKernel);
13841383
if (--KernelInfo.RefCount == 0) {
13851384
UR_CALL(getAsanInterceptor()->eraseKernelInfo(hKernel));
13861385
}
1386+
UR_CALL(pfnRelease(hKernel));
13871387

13881388
return UR_RESULT_SUCCESS;
13891389
}

source/loader/layers/sanitizer/msan/msan_ddi.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1281,12 +1281,12 @@ ur_result_t urKernelRelease(
12811281
auto pfnRelease = getContext()->urDdiTable.Kernel.pfnRelease;
12821282

12831283
getContext()->logger.debug("==== urKernelRelease");
1284-
UR_CALL(pfnRelease(hKernel));
12851284

12861285
auto &KernelInfo = getMsanInterceptor()->getOrCreateKernelInfo(hKernel);
12871286
if (--KernelInfo.RefCount == 0) {
12881287
UR_CALL(getMsanInterceptor()->eraseKernelInfo(hKernel));
12891288
}
1289+
UR_CALL(pfnRelease(hKernel));
12901290

12911291
return UR_RESULT_SUCCESS;
12921292
}

0 commit comments

Comments
 (0)