[WEBGPU] Update to latest compilationHints API (#16197)

This PR updates the webgpu runtime to use the latest compilationHints API.
apache · Dec 3, 2023 · a2f55a8 · a2f55a8
1 parent 9e4e17c
commit a2f55a8
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 11 deletions.
diff --git a/web/Makefile b/web/Makefile
@@ -49,6 +49,6 @@ src/tvmjs_runtime_wasi.js: dist/wasm/tvmjs_runtime.wasm emcc/decorate_as_wasi.py
 	python3 emcc/decorate_as_wasi.py dist/wasm/tvmjs_runtime.js $@ es
 
 clean:
-	@rm -rf dist/wasm
+	@rm -rf dist/wasm lib src/tvmjs_runtime_wasi.js
 
 -include dist/wasm/*.d
diff --git a/web/package-lock.json b/web/package-lock.json
diff --git a/web/package.json b/web/package.json
@@ -25,7 +25,7 @@
     "@types/node": "^20.4.5",
     "@typescript-eslint/eslint-plugin": "^5.59.6",
     "@typescript-eslint/parser": "^5.59.6",
-    "@webgpu/types": "^0.1.24",
+    "@webgpu/types": "^0.1.40",
     "eslint": "^8.41.0",
     "jest": "^26.0.1",
     "rollup": "^2.56.2",

diff --git a/web/src/webgpu.ts b/web/src/webgpu.ts
@@ -686,11 +686,12 @@ export class WebGPUContext {
 
     const shaderModule = this.device.createShaderModule({
       code: code,
-      hints: {
-        main: {
+      compilationHints: [
+        {
+          entryPoint: "main",
           layout: pipelineLayout
         }
-      }
+      ]
     });
 
     if (asyncMode) {