Skip to content

Commit bcf5a10

Browse files
authored
Merge pull request #2586 from AllanZyne/review/yang/fix_kernel_release
[DeviceASAN] Fix kernel release order
2 parents d3aa07a + 9a56768 commit bcf5a10

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
@@ -1418,12 +1418,12 @@ __urdlllocal ur_result_t urKernelRelease(
14181418
}
14191419

14201420
getContext()->logger.debug("==== urKernelRelease");
1421-
UR_CALL(pfnRelease(hKernel));
14221421

14231422
auto &KernelInfo = getAsanInterceptor()->getOrCreateKernelInfo(hKernel);
14241423
if (--KernelInfo.RefCount == 0) {
14251424
UR_CALL(getAsanInterceptor()->eraseKernelInfo(hKernel));
14261425
}
1426+
UR_CALL(pfnRelease(hKernel));
14271427

14281428
return UR_RESULT_SUCCESS;
14291429
}

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

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

13221322
getContext()->logger.debug("==== urKernelRelease");
1323-
UR_CALL(pfnRelease(hKernel));
13241323

13251324
auto &KernelInfo = getMsanInterceptor()->getOrCreateKernelInfo(hKernel);
13261325
if (--KernelInfo.RefCount == 0) {
13271326
UR_CALL(getMsanInterceptor()->eraseKernelInfo(hKernel));
13281327
}
1328+
UR_CALL(pfnRelease(hKernel));
13291329

13301330
return UR_RESULT_SUCCESS;
13311331
}

0 commit comments

Comments
 (0)