diff options
| author | Jerry-Ge <jerry.ge@arm.com> | 2025-04-07 16:27:08 -0700 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-04-07 16:27:08 -0700 |
| commit | a38ad6e2a24b46eaea481c04de566dd2a45d667b (patch) | |
| tree | 429438de96d1b22e1f4b902a2e979c9dea364089 | |
| parent | ddb32a67c2e76bc0f69470a8d02a220e8083fa68 (diff) | |
| download | llvm-a38ad6e2a24b46eaea481c04de566dd2a45d667b.zip llvm-a38ad6e2a24b46eaea481c04de566dd2a45d667b.tar.gz llvm-a38ad6e2a24b46eaea481c04de566dd2a45d667b.tar.bz2 | |
[mlir][tosa] Check empty Concat input for inferReturnTypeComponents (#134728)
Signed-off-by: Jerry Ge <jerry.ge@arm.com>
| -rw-r--r-- | mlir/lib/Dialect/Tosa/IR/TosaOps.cpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp b/mlir/lib/Dialect/Tosa/IR/TosaOps.cpp index c8e9ad8..59946ca 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) { |