[DevSAN] Fix logic error in function to set private base (#19063)

early return may cause barrier function not work.

Author: zhaomaosu

libdevice/sanitizer/asan_rtl.cpp

@@ -921,12 +921,12 @@ __asan_set_private_base(__SYCL_PRIVATE__ void *ptr) {
       launch_info->PrivateBase == 0)
     return;
   // Only set on the first sub-group item
-  if (__spirv_BuiltInSubgroupLocalInvocationId != 0)
-    return;
-  const size_t sid = SubGroupLinearId();
-  launch_info->PrivateBase[sid] = (uptr)ptr;
+  if (__spirv_BuiltInSubgroupLocalInvocationId == 0) {
+    const size_t sid = SubGroupLinearId();
+    launch_info->PrivateBase[sid] = (uptr)ptr;
+    ASAN_DEBUG(__spirv_ocl_printf(__asan_print_private_base, sid, ptr));
+  }
   SubGroupBarrier();
-  ASAN_DEBUG(__spirv_ocl_printf(__asan_print_private_base, sid, ptr));
 }
 
 #endif // __SPIR__ || __SPIRV__

libdevice/sanitizer/msan_rtl.cpp

@@ -592,12 +592,12 @@ __msan_set_private_base(__SYCL_PRIVATE__ void *ptr) {
       GetMsanLaunchInfo->PrivateBase == 0)
     return;
   // Only set on the first sub-group item
-  if (__spirv_BuiltInSubgroupLocalInvocationId != 0)
-    return;
-  const size_t sid = SubGroupLinearId();
-  GetMsanLaunchInfo->PrivateBase[sid] = (uptr)ptr;
+  if (__spirv_BuiltInSubgroupLocalInvocationId == 0) {
+    const size_t sid = SubGroupLinearId();
+    GetMsanLaunchInfo->PrivateBase[sid] = (uptr)ptr;
+    MSAN_DEBUG(__spirv_ocl_printf(__msan_print_private_base, sid, ptr));
+  }
   SubGroupBarrier();
-  MSAN_DEBUG(__spirv_ocl_printf(__msan_print_private_base, sid, ptr));
 }
 
 static __SYCL_CONSTANT__ const char __msan_print_strided_copy_unsupport_type[] =