Fix comment

axinging · Nov 3, 2023 · c617826 · c617826
1 parent a0aa145
commit c617826
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/js/web/lib/wasm/jsep/webgpu/ops/concat.ts b/js/web/lib/wasm/jsep/webgpu/ops/concat.ts
@@ -102,7 +102,7 @@ const createConcatProgramInfo = (inputs: readonly TensorView[], axis: number): P
     enableInputShapesUniforms.push(enableShapesUniforms(inputs[i].dims.length));
     inputShapeOrRanks.push(enableInputShapesUniforms[i] ? inputs[i].dims.length : inputs[i].dims);
     inputVars[i] = inputVariable(`input${i}`, dataType, inputShapeOrRanks[i]);
-    inputDependencies.push('rank');
+    inputDependencies.push(enableInputShapesUniforms[i] ? 'rank' : 'dims');
   }
   programUniforms.push({type: 'uint32', data: sizeInConcatAxis});
   for (let i = 0; i < inputs.length; ++i) {