<table border="1" cellspacing="0" cellpadding="8">
<tr>
<th>Issue</th>
<td>
<a href=https://github.com/llvm/llvm-project/issues/108151>108151</a>
</td>
</tr>
<tr>
<th>Summary</th>
<td>
[mlir] [tosa] -tosa-to-tensor crashes in TosaToTensor.cpp:195: SmallVector<ReassociationExprs> (anonymous namespace)::createReassociationMapForCollapse(OpBuilder &, Type, Type): Assertion `currSrcDim == srcShape.size() && currDstDim == dstShape.size()' failed.
</td>
</tr>
<tr>
<th>Labels</th>
<td>
</td>
</tr>
<tr>
<th>Assignees</th>
<td>
</td>
</tr>
<tr>
<th>Reporter</th>
<td>
axeabc
</td>
</tr>
</table>
<pre>
git version: 761bf333e378b52614c
system: `Ubuntu 18.04.6 LTS`
reproduce with: `mlir-opt -tosa-to-tensor a.mlir`
a.mlir:
```
module {
func.func @test_reshape_3d(%arg1: tensor<2x3x4xf32>) -> (tensor<2x?x4xf32>, tensor<2x?x4xf32>) {
%c0 = arith.constant 0 : index
%0 = tosa.reshape %arg1 {new_shape = array<i64: 2, 4, 4>} : (tensor<2x3x4xf32>) -> tensor<2x?x4xf32>
%1 = tosa.reshape %arg1 {new_shape = array<i64: 2, 1, 4>} : (tensor<2x3x4xf32>) -> tensor<2x?x4xf32>
return %0, %1 : tensor<2x?x4xf32>, tensor<2x?x4xf32>
}
}
```
stack trace:
```
mlir-opt: /data/szy/MLIR/llvm-release/llvm-project/mlir/lib/Conversion/TosaToTensor/TosaToTensor.cpp:195: SmallVector<ReassociationExprs> (anonymous namespace)::createReassociationMapForCollapse(OpBuilder &, Type, Type): Assertion `currSrcDim == srcShape.size() && currDstDim == dstShape.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: /data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt -tosa-to-tensor a.mlir
#0 0x000055b5adeb3128 llvm::sys::PrintStackTrace(llvm::raw_ostream&, int) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x10d7128)
#1 0x000055b5adeb0c3e llvm::sys::RunSignalHandlers() (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x10d4c3e)
#2 0x000055b5adeb3abd SignalHandler(int) Signals.cpp:0:0
#3 0x00007f7864715420 __restore_rt (/lib/x86_64-linux-gnu/libpthread.so.0+0x14420)
#4 0x00007f7863d5200b raise /build/glibc-LcI20x/glibc-2.31/signal/../sysdeps/unix/sysv/linux/raise.c:51:1
#5 0x00007f7863d31859 abort /build/glibc-LcI20x/glibc-2.31/stdlib/abort.c:81:7
#6 0x00007f7863d31729 get_sysdep_segment_value /build/glibc-LcI20x/glibc-2.31/intl/loadmsgcat.c:509:8
#7 0x00007f7863d31729 _nl_load_domain /build/glibc-LcI20x/glibc-2.31/intl/loadmsgcat.c:970:34
#8 0x00007f7863d42fd6 (/lib/x86_64-linux-gnu/libc.so.6+0x33fd6)
#9 0x000055b5b09c4e77 (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x3be8e77)
#10 0x000055b5b09c3fb7 (anonymous namespace)::ReshapeConverter::matchAndRewrite(mlir::tosa::ReshapeOp, mlir::tosa::ReshapeOpAdaptor, mlir::ConversionPatternRewriter&) const TosaToTensor.cpp:0:0
#11 0x000055b5b09c3350 mlir::OpConversionPattern<mlir::tosa::ReshapeOp>::matchAndRewrite(mlir::Operation*, llvm::ArrayRef<mlir::Value>, mlir::ConversionPatternRewriter&) const (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x3be7350)
#12 0x000055b5b1003b11 mlir::ConversionPattern::matchAndRewrite(mlir::Operation*, mlir::PatternRewriter&) const (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x4227b11)
#13 0x000055b5b398ec91 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
#14 0x000055b5b398b94b 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&)>) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x6baf94b)
#15 0x000055b5b1004b53 (anonymous namespace)::OperationLegalizer::legalize(mlir::Operation*, mlir::ConversionPatternRewriter&) DialectConversion.cpp:0:0
#16 0x000055b5b1003bb7 mlir::OperationConverter::convert(mlir::ConversionPatternRewriter&, mlir::Operation*) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x4227bb7)
#17 0x000055b5b1004d7f mlir::OperationConverter::convertOperations(llvm::ArrayRef<mlir::Operation*>) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x4228d7f)
#18 0x000055b5b100c96b mlir::applyPartialConversion(mlir::Operation*, mlir::ConversionTarget const&, mlir::FrozenRewritePatternSet const&, mlir::ConversionConfig) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x423096b)
#19 0x000055b5b09bf6c8 (anonymous namespace)::TosaToTensor::runOnOperation() TosaToTensorPass.cpp:0:0
#20 0x000055b5b0fa52d6 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x41c92d6)
#21 0x000055b5b0fa5c40 mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x41c9c40)
#22 0x000055b5b0fa8282 mlir::PassManager::run(mlir::Operation*) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x41cc282)
#23 0x000055b5b0fa0ab1 performActions(llvm::raw_ostream&, std::shared_ptr<llvm::SourceMgr> const&, mlir::MLIRContext*, mlir::MlirOptMainConfig const&) MlirOptMain.cpp:0:0
#24 0x000055b5b0fa070b 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
#25 0x000055b5b104d3a5 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) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x42713a5)
#26 0x000055b5b0f9bb35 mlir::MlirOptMain(llvm::raw_ostream&, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, mlir::DialectRegistry&, mlir::MlirOptMainConfig const&) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x41bfb35)
#27 0x000055b5b0f9bddf mlir::MlirOptMain(int, char**, llvm::StringRef, llvm::StringRef, mlir::DialectRegistry&) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x41bfddf)
#28 0x000055b5b0f9c10e mlir::MlirOptMain(int, char**, llvm::StringRef, mlir::DialectRegistry&) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x41c010e)
#29 0x000055b5ade93d67 main (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x10b7d67)
#30 0x00007f7863d33083 __libc_start_main /build/glibc-LcI20x/glibc-2.31/csu/../csu/libc-start.c:342:3
#31 0x000055b5ade938ee _start (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x10b78ee)
```
</pre>
<img width="1px" height="1px" alt="" src="http://email.email.llvm.org/o/eJzsWslu47rSfhplQ8SgSI2LLNxJjL-BDhIk-c_W4CSZ99CULkl12_30F5RkW5LtJD34HNwBSGxTLLHqq_pYLIoi1spSC3ETxJ-C-O6KNG5VmRuyEYSyK1rx7U0pHfgqjJWVDvAcpElIC4yxwGlGY5SEEQvgXQDn3afdWifWXjBI4P_TRrsGhNkMRrMEfHl9CRI4lDaiNhVvmADfpFv1d62VNNdV7cC1qyy5dtW1E9pWBpCZ75oM0V_EfdP3tn-ga68r3igBgvRT1wagaDSb-Q8QRNAJ65ZG2BWpxRLzAGUBiokpQ29LpzbAt2iDN9GmwCjA9wHKwXWA70GAsoFAgBcDkVtwtisf2gJAgGIGQYDvADHSrWas0tYR7YC_OAdSc7EZSXfC3jOz3m7Qm-wH1uLbsr_YDmnINsC3Mon8YMgbFnUf-D5I71oVYxwngJ6DMrQq_EWrwt9vlRGuMbp1mR-8N3J-PjLng9aPGaR3Pcn2PyZss46wP4EzhIkDI0_zsid5h3XBiSMBWtjv2wAtHr58fg7QQqmv62sjlCBW7Jq1qf4hmAvQomU9WihJA7S4rfRuiqLFa2XJa_XaYRk3Z6yuAzwP89jrfVkTpf4QzLWYnwWxtmKSOFnp-01tbM9yoiu9XVeNBZqsha09OJR7fHjOjCBOjG59IPWiMreVUqT2hmeP9adGKi4MCFDi3fy6rcXh248E5tYK4-_2CYA1xrwYdifXni-eMtawF8-fmZXfRTtJ826wBHjhO-sGwty6qXCAUlAQqQSfde5_-nI_f7kHtqFr6QABtCmBEXVlHHAVWDlXWw8QLQK0KKVbNXTGqnUfheNgSGsbYQO0AERzIDVTDRfArQRghtgVoIT92bKiV__S8oQ363pPEzgLYP5kqtKQNSCmbNZCO_vT9KDe5f5b6p4sb6XUjuABwhDADYQQxjGNCRcUhygDLeY23HZrux9PRmrXonhtyY6yg5Ah35aVdUaQdR9wqV0Xsew3YgnQJ7gJIU_DLsR7DOEEA2RYnMLw3OgXWWqi_o9oroSxe2JdwsyIYTEyE01dTSgHI4MClPWe6y7bfvrC9n8_EO4HSos0S6I0jCMEwdKvaq4yYmlcj6hLFZssWSbRtZK62VyXuuk6arcygvCZrWawsziKEBzZGw3VYB4jCCkwRFqf6vfgSyUpu_7CPiO42TfRDIfely2KAC1mM9_aWi5qP2kaLTfdha-tNbrxzXboGQvwPParcXiwJB5bgsMszgGhVYv0Y5Y43nmjvatVknkl6UFJMlWSohyUwi07u5dWlH6GLr8S1XzYA1I7j19VhK9tyUinOoY-B2YH3ekp3Uutlv7GJa_WROpfU5mnnkQ4OujMxjojVPDkA8RhnjJJSxmMC56MKJMPKE5hziKRppeYXpiKTKTpXrdPAnCiGxc0fW81e-5Kl241dcJ0V9fEsdVc82fxzUjnc92u4sRzn05H9z7WPuG9KTDnpPZL7kjusII_EeeE0b0206bQHLR1ITixlg-SgYcdTmHjGA7UPNZHigJ8-zYefP--Hx5rYdrVP0BzD-yQbue-2nsWxUjNH37S9EXXj_rgMvxJcQxH_BmmZxpCiGkYvmXrj_vo0PMXw40QSmkYjuDiIVycZ4LlIfhaST4Ipd80efuXpg1n29vXVzuOMKKUL3aWxZhXPcJ5XSvJSFtw_l533Z63k1aVAqOh-wE6D3fe7sn4LthfGuTQ8aUqJSPqWdhGuY-N2-eoAEVL2F7IVKVLT5Qj555ODdEkxDSP6LFX_xeiXwjRJWZrQkmRR3Q0W-NJcopojN9b3PbB-iJKouT33fKm-uYHI_tekr6TRAnmDmKnyZgcpVeaghP6J2sx65ojW9-26PbUqC2qi0Srza10XIqk02jxtPg41n23He2yTq6rI4AXY2SEUMbTYoQxm2BkeTJMLqSu1faJGCeJGj6s-EHKvRJTCjeYd0Oxham-i13weyq8nJU-DHpb6UKWF_MVhnkynr2TspgWCcvem72jhzrdNrvRj3rgtHbvOhR7IvbEjtHvPMe1cUFixJOBa7hwRKpdVF6rx9oPtStbd8onGdHao7idj-hcE7W10j4QTUrRlsJ-CfDfjW6fSPMLPjSIQpajwVbFuyScuoRF8Idd8iRroaSeplIvukf6Zkr6KR-N111rP2vrTPv4yG8yjhbmoUCnuevobX8iRmj3WRfVbuJcLFWGLGfRqOpGaBKGDGVoYv3OHyd5-Fdk-JAxlKGR2XhiNiQ0BLUwRWXWc3acvI8fkVnH-8dTK2IEX9bOjKqRl6oxTDyUJsD3ZzKaB3ZbaSc27ijsD0qax9o9ENmnu2HtAga9pxNGNMWXQgrOlUo_U1vt4Tda_rMRR_AfxLoy209NUXQzYS_PRUEa5ZZcKOHE-Xv8BmVS-U2D8P4uZuCnj4bz4ngO1vWV17MopXVme8yPtzhwZnPxVyJ5IzLvM3RSEUcck3iA3dZKurnmT6ZiwtqduX8L7_4NZ8WbEsO-F2ekLn1Zeu7yhaqsNMQkHuXkZJyzckpxfHo2_KfN5cuserSgeOzhdOphzouzHu5LFbYiXVEy_1HivOeei8HmfLTVQdkENguh-C2w_xZ8DIZQjPDl44OpHPMkBf2RwwVOxmjKk9F-eX8IuTv-wDDDYLlUkrKldcS45Y8cgDDb7I6cup9tZztOexSCI-Q_D-rDKf5MCNApvpQHMjEIwf5NgSt-g3mOc3IlbsIUJRDnCIdXq5uEZJCIGCYsQylNuUAJFlmRF0xARpm4kjcIogjmYQhTBHE4ozHNoyITIWIozwkPIijWRKqZt3NWmfKqPcK-CWEWxuGVIlQou3spyNy0aGhT2iCCSlpnD_c56VT7-lDL3vgOBPGn9hQhvjs6bW4Pw4UFUp86yfhvfivhqjHq5udfO-jD9vUG_SsAAP__PwyiXg">