[mlir][tosa] Check empty Concat input for inferReturnTypeComponents

[Jerry-Ge]

No description provided.

[llvmbot]

@llvm/pr-subscribers-mlir

Author: Jerry-Ge (Jerry-Ge)

Changes

---

Full diff: https://github.com/llvm/llvm-project/pull/134728.diff

1 Files Affected:

• (modified) mlir/lib/Dialect/Tosa/IR/TosaOps.cpp (+4)

diff --git a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
index c8e9ad8bd3346..889fe7a6bdb25 100644
--- a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
+++ b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
@@ -1066,6 +1066,10 @@ LogicalResult tosa::ConcatOp::inferReturnTypeComponents(
 
     hasRankedInput = true;
   }
+  
+  if (adaptor.getInput1().empty())
+    return failure();
+
   Type inputType =
       llvm::cast<TensorType>(adaptor.getInput1().getType()[0]).getElementType();
   if (!hasRankedInput) {

[llvmbot]

@llvm/pr-subscribers-mlir-tosa

Author: Jerry-Ge (Jerry-Ge)

Changes

---

Full diff: https://github.com/llvm/llvm-project/pull/134728.diff

1 Files Affected:

• (modified) mlir/lib/Dialect/Tosa/IR/TosaOps.cpp (+4)

diff --git a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
index c8e9ad8bd3346..889fe7a6bdb25 100644
--- a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
+++ b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp
@@ -1066,6 +1066,10 @@ LogicalResult tosa::ConcatOp::inferReturnTypeComponents(
 
     hasRankedInput = true;
   }
+  
+  if (adaptor.getInput1().empty())
+    return failure();
+
   Type inputType =
       llvm::cast<TensorType>(adaptor.getInput1().getType()[0]).getElementType();
   if (!hasRankedInput) {

[github-actions]

✅ With the latest revision this PR passed the C/C++ code formatter.