[webgpu] Update types. (#1851)

FEATURE

Author: annxingyuan

tfjs-webgpu/package.json

@@ -42,9 +42,9 @@
   },
   "dependencies": {
     "@webgpu/shaderc": "0.0.6",
-    "@webgpu/types": "0.0.3"
+    "@webgpu/types": "0.0.6"
   },
   "peerDependencies": {
     "@tensorflow/tfjs-core": "1.2.1"
   }
-}
+}

tfjs-webgpu/src/backend_webgpu.ts

@@ -66,8 +66,8 @@ type TensorInfo = {
 
 interface DataId {}
 
-const DEFAULT_GPUBUFFER_USAGE = GPUBufferUsage.STORAGE |
-    GPUBufferUsage.TRANSFER_SRC | GPUBufferUsage.TRANSFER_DST;
+const DEFAULT_GPUBUFFER_USAGE =
+    GPUBufferUsage.STORAGE | GPUBufferUsage.COPY_SRC | GPUBufferUsage.COPY_DST;
 
 export class WebGPUBackend extends KernelBackend {
   device: GPUDevice;
@@ -191,7 +191,7 @@ export class WebGPUBackend extends KernelBackend {
   private async getBufferData(info: TensorInfo): Promise<ArrayBuffer> {
     const staging = this.acquireBuffer(
         info.bufferInfo.byteSize,
-        GPUBufferUsage.TRANSFER_DST | GPUBufferUsage.MAP_READ);
+        GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ);
     const encoder = this.device.createCommandEncoder({});
     encoder.copyBufferToBuffer(
         info.bufferInfo.buffer, 0, staging, 0, info.bufferInfo.byteSize);
@@ -204,7 +204,7 @@ export class WebGPUBackend extends KernelBackend {
     staging.unmap();
     this.releaseBuffer(
         staging, info.bufferInfo.byteSize,
-        GPUBufferUsage.TRANSFER_DST | GPUBufferUsage.MAP_READ);
+        GPUBufferUsage.COPY_DST | GPUBufferUsage.MAP_READ);
 
     return values;
   }
@@ -369,14 +369,14 @@ export class WebGPUBackend extends KernelBackend {
     }
     this.releaseBuffer(
         uniforms.resource.buffer, uniformData.byteLength,
-        GPUBufferUsage.TRANSFER_DST | GPUBufferUsage.UNIFORM);
+        GPUBufferUsage.COPY_DST | GPUBufferUsage.UNIFORM);
     return output as {} as K;
   }
 
   private makeUniforms(data: Uint32Array|
                        Int32Array): webgpu_program.BindingInfo {
     const dimensionsBuffer = this.acquireBuffer(
-        data.byteLength, GPUBufferUsage.TRANSFER_DST | GPUBufferUsage.UNIFORM);
+        data.byteLength, GPUBufferUsage.COPY_DST | GPUBufferUsage.UNIFORM);
     dimensionsBuffer.setSubData(0, data);
 
     return {

tfjs-webgpu/src/kernels/webgpu_program.ts

@@ -70,12 +70,12 @@ const makeBindGroupLayout =
     (device: GPUDevice, inputs: shader_preprocessor.InputInfo[], output: Tensor,
      uniforms?: BindingInfo): GPUBindGroupLayout => {
       const bindings = Array(1 + inputs.length).fill({
-        visibility: GPUShaderStageBit.COMPUTE,
+        visibility: GPUShaderStage.COMPUTE,
         type: 'storage-buffer' as GPUBindingType
       });
       if (uniforms) {
         bindings.push({
-          visibility: GPUShaderStageBit.COMPUTE,
+          visibility: GPUShaderStage.COMPUTE,
           type: 'uniform-buffer' as GPUBindingType
         });
       }

tfjs-webgpu/yarn.lock

@@ -50,10 +50,10 @@
   resolved "https://registry.yarnpkg.com/@webgpu/shaderc/-/shaderc-0.0.6.tgz#832df2684f3a5b15ad0924049aa55f82ea043020"
   integrity sha512-GTOPB0j47XCjcGoPGhO0cIwlIO6kGT0fkLZrkm69XSYfBuMFi/ozSsZF9VFBnQG5o2sANt/lH+KZ2+YqcV94qA==
 
-"@webgpu/types@0.0.3":
-  version "0.0.3"
-  resolved "https://registry.yarnpkg.com/@webgpu/types/-/types-0.0.3.tgz#3b7989fab7fbc50bfa201fa46708927518fa3ece"
-  integrity sha512-2LXPgHQwpjHQsHTuooOuL88vMkZaCgHaQ1EayZbDsQgMM/q78Me1QMSyZDs542DEiu6YcILKBbkFNruSmVGE7A==
+"@webgpu/types@0.0.6":
+  version "0.0.6"
+  resolved "https://registry.yarnpkg.com/@webgpu/types/-/types-0.0.6.tgz#c012d4e6a6a1571e0cb6d16e055cfc4791dee16b"
+  integrity sha512-KZMqX8VjcKXXlEB1Ws//Qh9P7boHAx+eMD1+ic47JLlPN3Iz5EXw0elGZk/pERlVVXGCNGYo3zhObWhgyxKBSw==
 
 abbrev@1:
   version "1.1.1"