From c617826dce0687eb5137ec07fce6e97a679af0d6 Mon Sep 17 00:00:00 2001 From: Xu Xing Date: Fri, 3 Nov 2023 13:51:44 +0800 Subject: [PATCH] Fix comment --- js/web/lib/wasm/jsep/webgpu/ops/concat.ts | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/js/web/lib/wasm/jsep/webgpu/ops/concat.ts b/js/web/lib/wasm/jsep/webgpu/ops/concat.ts index fd3f12c31de8..9f0f7ae92d66 100644 --- 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) {