Replaced global_id.x with global_idx in gemm.ts.

microsoft · Dec 15, 2023 · 238aa47 · 238aa47
1 parent 5df50d9
commit 238aa47
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/gemm.ts b/js/web/lib/wasm/jsep/webgpu/ops/gemm.ts
@@ -97,8 +97,8 @@ const createGemmProgramInfo = (inputs: readonly TensorView[], attributes: GemmAt
   ${shaderHelper.mainStart()}
     ${shaderHelper.guardAgainstOutOfBoundsWorkgroupSizes(outputSize)}
 
-    let m = global_id.x / N;
-    let n = global_id.x % N;
+    let m = global_idx / N;
+    let n = global_idx % N;
 
     var value = ${dataType}(0);
     for (var k: u32 = 0u; k<${K}u; k++) {
@@ -107,7 +107,7 @@ const createGemmProgramInfo = (inputs: readonly TensorView[], attributes: GemmAt
 
     ${calculateAlpha}
     ${calculateC}
-    output[global_id.x] = value;
+    output[global_idx] = value;
 
   }`;
   return {