This is an automated email from the ASF dual-hosted git repository.
tqchen pushed a commit to branch unity
in repository https://gitbox.apache.org/repos/asf/tvm.git
The following commit(s) were added to refs/heads/unity by this push:
new a2f55a8812 [WEBGPU] Update to latest compilationHints API (#16197)
a2f55a8812 is described below
commit a2f55a8812f907f2c0486645f7d8cb0a6414d80d
Author: Tianqi Chen <[email protected]>
AuthorDate: Sat Dec 2 21:09:41 2023 -0500
[WEBGPU] Update to latest compilationHints API (#16197)
This PR updates the webgpu runtime to use the latest
compilationHints API.
---
web/Makefile | 2 +-
web/package-lock.json | 12 ++++++------
web/package.json | 2 +-
web/src/webgpu.ts | 7 ++++---
4 files changed, 12 insertions(+), 11 deletions(-)
diff --git a/web/Makefile b/web/Makefile
index 79f71bcb54..8fccf9636c 100644
--- 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
index 3783b0697a..37bac4493f 100644
--- a/web/package-lock.json
+++ b/web/package-lock.json
@@ -1,12 +1,12 @@
{
"name": "tvmjs",
- "version": "0.14.0-dev0",
+ "version": "0.15.0-dev0",
"lockfileVersion": 3,
"requires": true,
"packages": {
"": {
"name": "tvmjs",
- "version": "0.14.0-dev0",
+ "version": "0.15.0-dev0",
"license": "Apache-2.0",
"devDependencies": {
"@rollup/plugin-commonjs": "^20.0.0",
@@ -14,7 +14,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",
@@ -1766,9 +1766,9 @@
}
},
"node_modules/@webgpu/types": {
- "version": "0.1.24",
- "resolved":
"https://registry.npmjs.org/@webgpu/types/-/types-0.1.24.tgz";,
- "integrity":
"sha512-Mkz+SVJwHApTg6nCzqIuHDt3HsGRcCvHJNkWT2PgZTTC2Gy+LXvN4+7x6YvduAcx3F/pEDWW5OfAHs6VSo6J4Q==",
+ "version": "0.1.40",
+ "resolved":
"https://registry.npmjs.org/@webgpu/types/-/types-0.1.40.tgz";,
+ "integrity":
"sha512-/BBkHLS6/eQjyWhY2H7Dx5DHcVrS2ICj9owvSRdgtQT6KcafLZA86tPze0xAOsd4FbsYKCUBUQyNi87q7gV7kw==",
"dev": true
},
"node_modules/abab": {
diff --git a/web/package.json b/web/package.json
index 23855700e3..95a5c8efd7 100644
--- 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
index fb4d868f45..95dc7af9fc 100644
--- 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) {
