| Issue |
148372
|
| Summary |
[MLIR] Crash in `-sparsifier="vl=8"` pass when running ElementwiseOpFusion on dynamic-rank tensors with assertion message '__n < this->size()'
|
| Labels |
mlir
|
| Assignees |
|
| Reporter |
sweead
|
test commit: [f2e244f](https://github.com/llvm/llvm-project/commit/f2e244f443c99f814216bef37a82f1b8d3e5975d)
step to reproduce:
```
mlir-opt -sparsifier="vl=8" test.mlir
```
minimal test case:
```
#map = affine_map<(d0, d1) -> (d0, d1)>
module {
func.func @main() {
%c9 = arith.constant 9 : index
%c16 = arith.constant 16 : index
%c18 = arith.constant 18 : index
%c21 = arith.constant 21 : index
%c23 = arith.constant 23 : index
%c27 = arith.constant 27 : index
%c28 = arith.constant 28 : index
%c30 = arith.constant 30 : index
%0 = tensor.empty() : tensor<17x17xi1>
%1 = tensor.empty(%c30, %c9) : tensor<?x?xi1>
%alloc = memref.alloc(%c9, %c9) : memref<?x?x1xi32>
%collapsed = tensor.collapse_shape %1 [[0, 1]] : tensor<?x?xi1> into tensor<?xi1>
%2 = tensor.empty(%c21) : tensor<?xi1>
%mapped = linalg.map ins(%collapsed : tensor<?xi1>) outs(%2 : tensor<?xi1>)
(%in: i1) {
%3 = index.mul %c23, %c30
%4 = index.shru %c30, %c27
%false = index.bool.constant false
%alloc_0 = memref.alloc(%c16, %3) : memref<?x?xf16>
linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]} ins(%alloc_0, %alloc_0 : memref<?x?xf16>, memref<?x?xf16>) outs(%alloc_0 : memref<?x?xf16>) {
^bb0(%in_1: f16, %in_2: f16, %out: f16):
%alloc_3 = memref.alloc(%4, %3) : memref<?x?x1x7xi32>
linalg.broadcast ins(%alloc : memref<?x?x1xi32>) outs(%alloc_3 : memref<?x?x1x7xi32>) dimensions = [3]
linalg.yield %in_2 : f16
}
linalg.yield %false : i1
}
return
}
}
```
Crash backtrace:
```
/usr/lib/gcc/x86_64-linux-gnu/11/../../../../include/c++/11/bits/stl_vector.h:1045: reference std::vector<mlir::BlockArgument>::operator[](size_type) [_Tp = mlir::BlockArgument, _Alloc = std::allocator<mlir::BlockArgument>]: Assertion '__n < this->size()' failed.
PLEASE submit a bug report to https://github.com/llvm/llvm-project/issues/ and include the crash backtrace.
Stack dump:
0. Program arguments: mlir-opt -sparsifier=vl=8 test.mlir
#0 0x000055a21cf7a028 llvm::sys::PrintStackTrace(llvm::raw_ostream&, int) (/home/workdir/llvm-project/build/bin/mlir-opt+0x1421028)
#1 0x000055a21cf776a5 llvm::sys::RunSignalHandlers() (/home/workdir/llvm-project/build/bin/mlir-opt+0x141e6a5)
#2 0x000055a21cf7b141 SignalHandler(int, siginfo_t*, void*) Signals.cpp:0:0
#3 0x00007f41ccaad520 (/lib/x86_64-linux-gnu/libc.so.6+0x42520)
#4 0x00007f41ccb019fc pthread_kill (/lib/x86_64-linux-gnu/libc.so.6+0x969fc)
#5 0x00007f41ccaad476 gsignal (/lib/x86_64-linux-gnu/libc.so.6+0x42476)
#6 0x00007f41cca937f3 abort (/lib/x86_64-linux-gnu/libc.so.6+0x287f3)
#7 0x000055a21cf668bf (/home/workdir/llvm-project/build/bin/mlir-opt+0x140d8bf)
#8 0x000055a21dbf1662 mlir::linalg::detail::LinalgOpTrait<mlir::linalg::GenericOp>::payloadUsesValueFromOperand(mlir::OpOperand*) (/home/workdir/llvm-project/build/bin/mlir-opt+0x2098662)
#9 0x000055a21df1f898 (anonymous namespace)::RemoveOutsDependency::matchAndRewrite(mlir::linalg::GenericOp, mlir::PatternRewriter&) const ElementwiseOpFusion.cpp:0:0
#10 0x000055a222d69132 void llvm::function_ref<void ()>::callback_fn<mlir::PatternApplicator::matchAndRewrite(mlir::Operation*, mlir::PatternRewriter&, llvm::function_ref<bool (mlir::Pattern const&)>, llvm::function_ref<void (mlir::Pattern const&)>, llvm::function_ref<llvm::LogicalResult (mlir::Pattern const&)>)::$_0>(long) PatternApplicator.cpp:0:0
#11 0x000055a222d65e2f mlir::PatternApplicator::matchAndRewrite(mlir::Operation*, mlir::PatternRewriter&, llvm::function_ref<bool (mlir::Pattern const&)>, llvm::function_ref<void (mlir::Pattern const&)>, llvm::function_ref<llvm::LogicalResult (mlir::Pattern const&)>) (/home/workdir/llvm-project/build/bin/mlir-opt+0x720ce2f)
#12 0x000055a2204d327a (anonymous namespace)::GreedyPatternRewriteDriver::processWorklist() GreedyPatternRewriteDriver.cpp:0:0
#13 0x000055a2204cfad3 mlir::applyPatternsGreedily(mlir::Region&, mlir::FrozenRewritePatternSet const&, mlir::GreedyRewriteConfig, bool*) (/home/workdir/llvm-project/build/bin/mlir-opt+0x4976ad3)
#14 0x000055a21df14365 (anonymous namespace)::LinalgElementwiseOpFusionPass::runOnOperation() ElementwiseOpFusion.cpp:0:0
#15 0x000055a220451a93 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48f8a93)
#16 0x000055a220452332 mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48f9332)
#17 0x000055a2204587be auto void mlir::parallelForEach<__gnu_cxx::__normal_iterator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo*, std::vector<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo, std::allocator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo>>>, mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::$_0>(mlir::MLIRContext*, __gnu_cxx::__normal_iterator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo*, std::vector<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo, std::allocator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo>>>, __gnu_cxx::__normal_iterator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo*, std::vector<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo, std::allocator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo>>>, mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::$_0&&)::'lambda'(__gnu_cxx::__normal_iterator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo*, std::vector<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo, std::allocator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo>>>&&)::operator()<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo&>(__gnu_cxx::__normal_iterator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo*, std::vector<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo, std::allocator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo>>>&&) const Pass.cpp:0:0
#18 0x000055a220453b8b mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48fab8b)
#19 0x000055a220451bec mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48f8bec)
#20 0x000055a220452332 mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48f9332)
#21 0x000055a220454b0e mlir::PassManager::run(mlir::Operation*) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48fbb0e)
#22 0x000055a22044ca7b performActions(llvm::raw_ostream&, std::shared_ptr<llvm::SourceMgr> const&, mlir::MLIRContext*, mlir::MlirOptMainConfig const&) MlirOptMain.cpp:0:0
#23 0x000055a22044c6d4 llvm::LogicalResult llvm::function_ref<llvm::LogicalResult (std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&)>::callback_fn<mlir::MlirOptMain(llvm::raw_ostream&, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, mlir::DialectRegistry&, mlir::MlirOptMainConfig const&)::$_0>(long, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&) MlirOptMain.cpp:0:0
#24 0x000055a2204fc425 mlir::splitAndProcessBuffer(std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::function_ref<llvm::LogicalResult (std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&)>, llvm::raw_ostream&, llvm::StringRef, llvm::StringRef) (/home/workdir/llvm-project/build/bin/mlir-opt+0x49a3425)
#25 0x000055a2204458c2 mlir::MlirOptMain(llvm::raw_ostream&, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, mlir::DialectRegistry&, mlir::MlirOptMainConfig const&) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48ec8c2)
#26 0x000055a220445b78 mlir::MlirOptMain(int, char**, llvm::StringRef, llvm::StringRef, mlir::DialectRegistry&) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48ecb78)
#27 0x000055a220445d92 mlir::MlirOptMain(int, char**, llvm::StringRef, mlir::DialectRegistry&) (/home/workdir/llvm-project/build/bin/mlir-opt+0x48ecd92)
#28 0x000055a21cf58a67 main (/home/workdir/llvm-project/build/bin/mlir-opt+0x13ffa67)
#29 0x00007f41cca94d90 (/lib/x86_64-linux-gnu/libc.so.6+0x29d90)
#30 0x00007f41cca94e40 __libc_start_main (/lib/x86_64-linux-gnu/libc.so.6+0x29e40)
#31 0x000055a21cf58505 _start (/home/workdir/llvm-project/build/bin/mlir-opt+0x13ff505)
Aborted (core dumped)
```
![]()
_______________________________________________
llvm-bugs mailing list
llvm-bugs@lists.llvm.org
https://lists.llvm.org/cgi-bin/mailman/listinfo/llvm-bugs
