fix CI errors

Author: qjia7

onnxruntime/core/providers/webgpu/webgpu_context.cc

@@ -995,6 +995,11 @@ WebGpuContext& WebGpuContextFactory::GetContext(int context_id) {
   return *it->second.context;
 }
 
+bool WebGpuContextFactory::HasContext(int context_id) {
+  std::lock_guard<std::mutex> lock(mutex_);
+  return contexts_.find(context_id) != contexts_.end();
+}
+
 void WebGpuContextFactory::ReleaseContext(int context_id) {
   std::lock_guard<std::mutex> lock(mutex_);
 

onnxruntime/core/providers/webgpu/webgpu_context.h

@@ -64,6 +64,7 @@ class WebGpuContextFactory {
 
   static WebGpuContext& CreateContext(const WebGpuContextConfig& config);
   static WebGpuContext& GetContext(int context_id);
+  static bool HasContext(int context_id);
 
   static void ReleaseContext(int context_id);
 

onnxruntime/core/providers/webgpu/webgpu_provider_factory.cc

@@ -384,19 +384,14 @@ struct WebGpuDataTransferImpl : OrtDataTransferImpl {
 
 OrtDataTransferImpl* OrtWebGpuCreateDataTransfer(int context_id) {
   webgpu::WebGpuContext* context_ptr = nullptr;
-  try {
+  if (webgpu::WebGpuContextFactory::HasContext(context_id)) {
     context_ptr = &webgpu::WebGpuContextFactory::GetContext(context_id);
-  } catch (...) {
-    // Context doesn't exist, create a default one using shared helper
+  } else {
     WebGpuContextParams params = GetDefaultWebGpuContextParams(context_id);
     context_ptr = &webgpu::WebGpuContextFactory::CreateContext(params.context_config);
     context_ptr->Initialize(params.buffer_cache_config, params.backend_type, params.enable_pix_capture);
   }
-  if (context_ptr) {
-    return new WebGpuDataTransferImpl(*OrtApis::GetApi(ORT_API_VERSION), context_ptr->BufferManager());
-  }
-
-  return nullptr;
+  return new WebGpuDataTransferImpl(*OrtApis::GetApi(ORT_API_VERSION), context_ptr->BufferManager());
 }
 
 }  // namespace onnxruntime