concat: fix 1 input

microsoft · Sep 13, 2022 · 67ea4cb · 67ea4cb
1 parent 21b5dfe
commit 67ea4cb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/js/web/lib/onnxjs/backends/webgpu/ops/concat.ts b/js/web/lib/onnxjs/backends/webgpu/ops/concat.ts
@@ -130,7 +130,9 @@ const readBufferDataImpl = (indicesHelper: readonly IndicesHelper[], tensorRank:
   const codeLines: string[] = [];
   for (let i = 0; i < numberOfTensors; ++i) {
     const returnSnippet = `return input${i}[${indicesHelper[i].i2oExpression('indices', true)}];`;
-    if (i === 0) {
+    if (numberOfTensors === 1) {
+      codeLines.push(returnSnippet);
+    } else if (i === 0) {
       codeLines.push(`if (textureIndex == ${i}u) { ${returnSnippet} }`);
     } else if (i === numberOfTensors - 1) {
       codeLines.push(`else { ${returnSnippet} }`);