fix crash

Author: AllanZyne

source/loader/layers/sanitizer/asan_interceptor.cpp

@@ -670,13 +670,18 @@ ur_result_t SanitizerInterceptor::prepareLaunch(
         auto KernelInfo = getKernelInfo(Kernel);
 
         // Validate pointer arguments
-        for (const auto &[ArgIndex, PtrPair] : KernelInfo->PointerArgs) {
-            auto Ptr = PtrPair.first;
-            if (auto ValidateResult = ValidateUSMPointer(
-                    Context, DeviceInfo->Handle, (uptr)Ptr)) {
-                ReportInvalidKernelArgument(Kernel, ArgIndex, (uptr)Ptr,
-                                            ValidateResult, PtrPair.second);
-                exit(1);
+        if (Options(logger).DetectKernelArguments) {
+            for (const auto &[ArgIndex, PtrPair] : KernelInfo->PointerArgs) {
+                auto Ptr = PtrPair.first;
+                if (Ptr == nullptr) {
+                    continue;
+                }
+                if (auto ValidateResult = ValidateUSMPointer(
+                        Context, DeviceInfo->Handle, (uptr)Ptr)) {
+                    ReportInvalidKernelArgument(Kernel, ArgIndex, (uptr)Ptr,
+                                                ValidateResult, PtrPair.second);
+                    exit(1);
+                }
             }
         }
 

source/loader/layers/sanitizer/asan_options.hpp

@@ -38,6 +38,7 @@ struct AsanOptions {
     uint32_t MaxQuarantineSizeMB = 0;
     bool DetectLocals = true;
     bool DetectPrivates = true;
+    bool DetectKernelArguments = true;
 
   private:
     AsanOptions(logger::Logger &logger) {
@@ -93,6 +94,7 @@ struct AsanOptions {
         SetBoolOption("debug", Debug);
         SetBoolOption("detect_locals", DetectLocals);
         SetBoolOption("detect_privates", DetectPrivates);
+        SetBoolOption("detect_kernel_arguments", DetectKernelArguments);
 
         auto KV = OptionsEnvMap->find("quarantine_size_mb");
         if (KV != OptionsEnvMap->end()) {

source/loader/layers/sanitizer/asan_validator.cpp

@@ -56,7 +56,7 @@ ValidateUSMResult ValidateUSMPointer(ur_context_handle_t Context,
                                        AllocInfo);
     }
 
-    if (!IsSameDevice(AllocInfo->Device, Device)) {
+    if (AllocInfo->Device && !IsSameDevice(AllocInfo->Device, Device)) {
         return ValidateUSMResult::fail(ValidateUSMResult::BAD_DEVICE,
                                        AllocInfo);
     }