From 5115ea67f99ac8fcc8705dce0b70319c8ead2728 Mon Sep 17 00:00:00 2001 From: "Zhao, Yang2" Date: Mon, 19 May 2025 05:22:45 +0200 Subject: [PATCH] fix cpu crash --- libdevice/sanitizer/asan_rtl.cpp | 3 ++- libdevice/sanitizer/msan_rtl.cpp | 3 ++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/libdevice/sanitizer/asan_rtl.cpp b/libdevice/sanitizer/asan_rtl.cpp index 65b27506598b7..a2f50add2fc61 100644 --- a/libdevice/sanitizer/asan_rtl.cpp +++ b/libdevice/sanitizer/asan_rtl.cpp @@ -905,7 +905,8 @@ static __SYCL_CONSTANT__ const char __asan_print_private_base[] = DEVICE_EXTERN_C_NOINLINE void __asan_set_private_base(__SYCL_PRIVATE__ void *ptr) { auto launch_info = (__SYCL_GLOBAL__ const AsanRuntimeData *)__AsanLaunchInfo; - if (!launch_info || launch_info->PrivateShadowOffset == 0) + if (!launch_info || launch_info->PrivateShadowOffset == 0 || + launch_info->PrivateBase == 0) return; // Only set on the first sub-group item if (__spirv_BuiltInSubgroupLocalInvocationId != 0) diff --git a/libdevice/sanitizer/msan_rtl.cpp b/libdevice/sanitizer/msan_rtl.cpp index bcdb8db35d612..864a6c0c2c4af 100644 --- a/libdevice/sanitizer/msan_rtl.cpp +++ b/libdevice/sanitizer/msan_rtl.cpp @@ -578,7 +578,8 @@ static __SYCL_CONSTANT__ const char __msan_print_private_base[] = DEVICE_EXTERN_C_NOINLINE void __msan_set_private_base(__SYCL_PRIVATE__ void *ptr) { - if (!GetMsanLaunchInfo || GetMsanLaunchInfo->PrivateShadowOffset == 0) + if (!GetMsanLaunchInfo || GetMsanLaunchInfo->PrivateShadowOffset == 0 || + GetMsanLaunchInfo->PrivateBase == 0) return; // Only set on the first sub-group item if (__spirv_BuiltInSubgroupLocalInvocationId != 0)