address Yang's comments

microsoft · Jan 22, 2024 · 0bf0b64 · 0bf0b64
1 parent 3ac29fe
commit 0bf0b64
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/js/web/lib/wasm/binding/ort-wasm.d.ts b/js/web/lib/wasm/binding/ort-wasm.d.ts
@@ -183,7 +183,7 @@ export interface OrtWasmModule extends EmscriptenModule {
       (gpuBuffer: GPUBuffer, size: number,
        type: Tensor.GpuBufferDataTypes) => () => Promise<Tensor.DataTypeMap[Tensor.GpuBufferDataTypes]>;
   /**
-   *  [exported from js_internal_api.js] specify a session.run is triggered.
+   *  [exported from js_internal_api.js] Called when InferenceSession.run started.
    */
   jsepOnRunStart: () => void;
   // #endregion

diff --git a/js/web/lib/wasm/jsep/backend-webgpu.ts b/js/web/lib/wasm/jsep/backend-webgpu.ts
@@ -207,6 +207,9 @@ export class WebGpuBackend {
     };
 
     Object.defineProperty(this.env.webgpu, 'device', {value: this.device});
+
+    // init queryType, which is necessary for InferenceSession.create
+    this.setQueryType();
   }
 
   dispose(): void {