[DeviceSanitizer] Add assertion to possible nullptr and die when trying to create shadow with unsupported device (#19129)

Addresses #18333.

Author: yingcong-wu

unified-runtime/source/loader/layers/sanitizer/asan/asan_interceptor.cpp

@@ -671,6 +671,7 @@ KernelInfo &AsanInterceptor::getOrCreateKernelInfo(ur_kernel_handle_t Kernel) {
   // Create new KernelInfo
   auto Program = GetProgram(Kernel);
   auto PI = getProgramInfo(Program);
+  assert(PI != nullptr && "unregistered program!");
   bool IsInstrumented = PI->isKernelInstrumented(Kernel);
 
   std::scoped_lock<ur_shared_mutex> Guard(m_KernelMapMutex);

unified-runtime/source/loader/layers/sanitizer/msan/msan_interceptor.cpp

@@ -229,6 +229,7 @@ ur_result_t MsanInterceptor::registerSpirKernels(ur_program_handle_t Program) {
     }
 
     auto PI = getProgramInfo(Program);
+    assert(PI != nullptr && "unregistered program!");
     for (const auto &SKI : SKInfo) {
       if (SKI.Size == 0) {
         continue;
@@ -404,6 +405,7 @@ KernelInfo &MsanInterceptor::getOrCreateKernelInfo(ur_kernel_handle_t Kernel) {
 
   // Create new KernelInfo
   auto PI = getProgramInfo(GetProgram(Kernel));
+  assert(PI != nullptr && "unregistered program!");
   auto KI = std::make_unique<KernelInfo>(Kernel);
 
   KI->IsInstrumented = PI->isKernelInstrumented(Kernel);

unified-runtime/source/loader/layers/sanitizer/msan/msan_shadow.cpp

@@ -75,10 +75,10 @@ GetMsanShadowMemory(ur_context_handle_t Context, ur_device_handle_t Device,
     static std::shared_ptr<MsanShadowMemory> ShadowDG2 =
         std::make_shared<MsanShadowMemoryDG2>(Context, Device);
     return ShadowDG2;
-  } else {
-    UR_LOG_L(getContext()->logger, ERR, "Unsupport device type");
-    return nullptr;
   }
+
+  die("GetMsanShadowMemory: Unsupport device type");
+  return nullptr;
 }
 
 ur_result_t MsanShadowMemoryCPU::Setup() {

unified-runtime/source/loader/layers/sanitizer/tsan/tsan_shadow.cpp

@@ -29,10 +29,10 @@ std::shared_ptr<ShadowMemory> GetShadowMemory(ur_context_handle_t Context,
     return ShadowCPU;
   } else if (Type == DeviceType::GPU_PVC) {
     return std::make_shared<ShadowMemoryPVC>(Context, Device);
-  } else {
-    UR_LOG_L(getContext()->logger, ERR, "Unsupport device type");
-    return nullptr;
   }
+
+  die("GetShadowMemory: Unsupport device type");
+  return nullptr;
 }
 
 ur_result_t ShadowMemoryCPU::Setup() {