placeholder for setting proc table

xhcao · Sep 12, 2024 · dce0f18 · dce0f18
1 parent 8b61532
commit dce0f18
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 2 deletions.
diff --git a/onnxruntime/core/providers/webgpu/webgpu_provider_factory.cc b/onnxruntime/core/providers/webgpu/webgpu_provider_factory.cc
@@ -3,6 +3,8 @@
 
 #include <charconv>
 
+#include <dawn/dawn_proc.h>
+
 #include "core/framework/error_code_helper.h"
 #include "core/providers/webgpu/buffer_manager.h"
 #include "core/providers/webgpu/webgpu_execution_provider.h"
@@ -33,7 +35,19 @@ struct WebGpuProviderFactory : IExecutionProviderFactory {
 
 std::shared_ptr<IExecutionProviderFactory> WebGpuProviderFactoryCreator::Create(const ConfigOptions& config_options) {
   //
-  // STEP.1 - prepare WebGpuExecutionProviderInfo
+  // STEP.1 - set dawn proc table
+  //
+  std::string dawn_proc_table_str;
+  if (config_options.TryGetConfigEntry(kDawnProcTable, dawn_proc_table_str)) {
+    size_t dawn_proc_table = 0;
+    ORT_ENFORCE(std::errc{} ==
+                std::from_chars(dawn_proc_table_str.data(), dawn_proc_table_str.data() + dawn_proc_table_str.size(), dawn_proc_table).ec);
+    // TODO: do this for static link build
+    // dawnProcSetProcs(reinterpret_cast<const DawnProcTable*>(dawn_proc_table));
+  }
+
+  //
+  // STEP.2 - prepare WebGpuExecutionProviderInfo
   //
   WebGpuExecutionProviderInfo webgpu_ep_info{
       // preferred layout is NHWC by default
@@ -100,7 +114,7 @@ std::shared_ptr<IExecutionProviderFactory> WebGpuProviderFactoryCreator::Create(
   LOGS_DEFAULT(VERBOSE) << "WebGPU EP default buffer cache mode: " << webgpu_ep_info.default_buffer_cache_mode;
 
   //
-  // STEP.2 - prepare WebGpuContext
+  // STEP.3 - prepare WebGpuContext
   //
   int context_id = 0;
   std::string context_id_str;

diff --git a/onnxruntime/core/providers/webgpu/webgpu_provider_options.h b/onnxruntime/core/providers/webgpu/webgpu_provider_options.h
@@ -17,6 +17,8 @@ constexpr const char* kWebGpuInstance = "webgpuInstance";
 constexpr const char* kWebGpuAdapter = "webgpuAdapter";
 constexpr const char* kWebGpuDevice = "webgpuDevice";
 
+constexpr const char* kDawnProcTable = "dawnProcTable";
+
 constexpr const char* kStorageBufferCacheMode = "storageBufferCacheMode";
 constexpr const char* kUniformBufferCacheMode = "uniformBufferCacheMode";
 constexpr const char* kQueryResolveBufferCacheMode = "queryResolveBufferCacheMode";