<table border="1" cellspacing="0" cellpadding="8">
<tr>
<th>Issue</th>
<td>
<a href=https://github.com/llvm/llvm-project/issues/120884>120884</a>
</td>
</tr>
<tr>
<th>Summary</th>
<td>
[mlir] -sparsification-and-bufferization crashes
</td>
</tr>
<tr>
<th>Labels</th>
<td>
mlir
</td>
</tr>
<tr>
<th>Assignees</th>
<td>
</td>
</tr>
<tr>
<th>Reporter</th>
<td>
wwy6191
</td>
</tr>
</table>
<pre>
git version: bc29fc937c6cb4a210f80c93c79fc6ed97c801f8
system: `Ubuntu 18.04.6 LTS`
reproduce with: `mlir-opt -sparsification-and-bufferization a.mlir`
a.mlir:
```
#map = affine_map<()[s0,s1,s2] -> ((s0 * s1) * s1)>
module {
func.func @test_reshape_3d_same_d2d_explicit_empty(%arg0: tensor<3x?x?xf32>) -> tensor<?x?x?xf32> {
%c2 = arith.constant 2 : index
%c1 = arith.constant 1 : index
%c3 = arith.constant 3 : index
%collapsed = tensor.collapse_shape %arg0 [[0, 1, 2]] : tensor<3x?x?xf32> into tensor<?xf32>
%dim = tensor.dim %arg0, %c1 : tensor<3x?x?xf32>
%dim_0 = tensor.dim %collapsed, %c2 : tensor<?xf32>
%dim_1 = tensor.dim %collapsed, %c3 : tensor<?xf32>
%0 = affine.apply #map()[%dim, %dim_0, %dim_1]
%1 = arith.divui %0, %c1 : index
%expanded = tensor.expand_shape %collapsed [[0, 1, 2]] output_shape [%1, 3, %0] : tensor<?xf32> into tensor<?x3x?xf32>
%cast = tensor.cast %expanded : tensor<?x3x?xf32> to tensor<?x?x?xf32>
return %cast : tensor<?x?x?xf32>
}
}
```
stack trace:
```
mlir-opt: /data/szy/MLIR/llvm-release/llvm-project/mlir/lib/Dialect/MemRef/IR/MemRefOps.cpp:2315: static void mlir::memref::ExpandShapeOp::build(OpBuilder &, OperationState &, Type, Value, ArrayRef<ReassociationIndices>): Assertion `succeeded(outputShape) && "unable to infer output shape"' 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 -sparsification-and-bufferization /data/szy/MLIR/seed/seed14/tmp.hNYivZemUi.mlir
#0 0x0000556e46049418 llvm::sys::PrintStackTrace(llvm::raw_ostream&, int) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x11ee418)
#1 0x0000556e46046f2e llvm::sys::RunSignalHandlers() (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x11ebf2e)
#2 0x0000556e46049dad SignalHandler(int) Signals.cpp:0:0
#3 0x00007f218845a420 __restore_rt (/lib/x86_64-linux-gnu/libpthread.so.0+0x14420)
#4 0x00007f2187a9700b raise /build/glibc-LcI20x/glibc-2.31/signal/../sysdeps/unix/sysv/linux/raise.c:51:1
#5 0x00007f2187a76859 abort /build/glibc-LcI20x/glibc-2.31/stdlib/abort.c:81:7
#6 0x00007f2187a76729 get_sysdep_segment_value /build/glibc-LcI20x/glibc-2.31/intl/loadmsgcat.c:509:8
#7 0x00007f2187a76729 _nl_load_domain /build/glibc-LcI20x/glibc-2.31/intl/loadmsgcat.c:970:34
#8 0x00007f2187a87fd6 (/lib/x86_64-linux-gnu/libc.so.6+0x33fd6)
#9 0x0000556e478b5681 (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x2a5a681)
#10 0x0000556e48590b6b mlir::memref::ExpandShapeOp mlir::OpBuilder::create<mlir::memref::ExpandShapeOp, llvm::ArrayRef<long>, mlir::Value&, llvm::SmallVector<llvm::SmallVector<long, 2u>, 4u>>(mlir::Location, llvm::ArrayRef<long>&&, mlir::Value&, llvm::SmallVector<llvm::SmallVector<long, 2u>, 4u>&&) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x3735b6b)
#11 0x0000556e485902ea mlir::bufferization::detail::BufferizableOpInterfaceInterfaceTraits::FallbackModel<mlir::tensor::(anonymous namespace)::ExpandShapeOpInterface>::bufferize(mlir::bufferization::detail::BufferizableOpInterfaceInterfaceTraits::Concept const*, mlir::Operation*, mlir::RewriterBase&, mlir::bufferization::BufferizationOptions const&) BufferizableOpInterfaceImpl.cpp:0:0
#12 0x0000556e465a1340 mlir::bufferization::bufferizeOp(mlir::Operation*, mlir::bufferization::BufferizationOptions const&, mlir::bufferization::BufferizationStatistics*) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x1746340)
#13 0x0000556e465dc1e4 mlir::bufferization::bufferizeModuleOp(mlir::ModuleOp, mlir::bufferization::OneShotBufferizationOptions const&, mlir::bufferization::BufferizationStatistics*) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x17811e4)
#14 0x0000556e4805e588 mlir::sparse_tensor::SparsificationAndBufferizationPass::runDenseBufferization() (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x3203588)
#15 0x0000556e4805e0dc mlir::sparse_tensor::SparsificationAndBufferizationPass::runOnOperation() (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x32030dc)
#16 0x0000556e48fa9d66 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x414ed66)
#17 0x0000556e48faa690 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+0x414f690)
#18 0x0000556e48facc62 mlir::PassManager::run(mlir::Operation*) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x4151c62)
#19 0x0000556e48fa53da performActions(llvm::raw_ostream&, std::shared_ptr<llvm::SourceMgr> const&, mlir::MLIRContext*, mlir::MlirOptMainConfig const&) MlirOptMain.cpp:0:0
#20 0x0000556e48fa502d 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
#21 0x0000556e49050985 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+0x41f5985)
#22 0x0000556e48f9f015 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+0x4144015)
#23 0x0000556e48f9f2bf mlir::MlirOptMain(int, char**, llvm::StringRef, llvm::StringRef, mlir::DialectRegistry&) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x41442bf)
#24 0x0000556e48f9f5ee mlir::MlirOptMain(int, char**, llvm::StringRef, mlir::DialectRegistry&) (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x41445ee)
#25 0x0000556e46029f07 main (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x11cef07)
#26 0x00007f2187a78083 __libc_start_main /build/glibc-LcI20x/glibc-2.31/csu/../csu/libc-start.c:342:3
#27 0x0000556e46029a7e _start (/data/szy/MLIR/llvm-release/llvm-project/build/bin/mlir-opt+0x11cea7e)
```
</pre>
<img width="1" height="1" alt="" src="http://email.email.llvm.org/o/eJzcWl9v2zgS_zTMCxGDIvX3IQ9KWuMKNJei6S5w9yJQ1MjmrUTpSCqN99MfSMm25Dhtuldjb69oFIoiZ-Y38-NwRIUbIzcK4AZFtyh6d8UHu-30zdevuzjIgquyq3Y3G2nxE2gjO4VYjktBs1pkLBGxKENOA1KnRGRMJFktYqiyRKQkqFNEckRyszMWWjcPxeSXclB2wEG6IuEqxh-_PKKYjOM09LqrBgH4q7TbaXzbSH3d9RZjfG16ro2speBWduqaq-q6HOoatPzd92C-csMPAhHJpx6WY-w6YjL-9zeUtbzHiL3DvK6lgqLlPWJ3iKaIZii6NQTROxO4C0XRO3yN2Hvsn6aGYERz7B5mxxZi7xHJ264aGsAouUUkx7gelFi5C0YhsWBsocFseQ8FqwrDWygqWhXw3DdSSFtA29udVxJxvSHOcgvKdBqxO_aM2Nr_1Iw6ZTQbjTqM2D8_jjnYgTGikaAjXi3tdiU6ZSxXFrvOHEtVwfNsaHBuaHB2KDs3lJ0d2jUN7w1UfsZo92rfWXi_4Ak69ny8dUHALgjYBcHF4Zs-wVLZbumQyVkHGyrZzrX728nb9O4A_Vt-n4sqyBlhB5h7iXQpcSlrEhS8RRD7lqDRKDLj9Ir3fbPDI9cPzB4VTiI9hlk7cG4-CJuzoJJPg_Qalo46xBjRCJ57rqplfMe-Y3RnJHglxN1g-8HuZ3iD_XM2KSYvePBNBrCzwRPc2AUN_f0CQv66HHyq5CxHNNhBq5m2_LuTUOLcP11PcpaxXPyGreYCEMvPDNinS5896briliO6Nr_vEF3ff_zwGdF10zy11xoa4Ab2t73u_gXCIrr26ZKuG1kiun4neTN230P7GWpE117EePfQm5Xoe8RyyoLIaTSWWynwUycrPOVdxPIWWg312H7vffvo4vrQj13lIJsK0fShv3Ut0BjR2MX5oQft8_qj5Rb2vV92Pbjfv_Jm8I1ca75zxrG7z8CN6YT0sz6oSgowY5501uXGgPb7BIqJGYQAqNzCSke2eaPGfB4jGmNE6aB42YCLtFQ16ImW2IwjKaIJrrlsoFohkn_6-D5_fI_NULbSYo7LYYM19J22TsDW2t44wHSN6Hoj7XYoV6Jrpwi8DIQ0ZgCD6BpzVWGpRDNUgO0WsNDcbHHJxW-eCE73o-dFNbT9SAuywv7fJ91tNG8x15uhBWXNH-bFFKV1KdXEEr8rf39PfkWbAed6_ysIEV3btl9t__4P-fRPaH-R467tMwojmDwTQkgUxRDGJMzCIMXeZ54-ZmfGxictlfWO-OLXB02PgzT_WnTGauDtRCOp7Bjr9Ce6A9Fb8hwEAGHgM-0IIDgBENcUzgH4PKhHuVG8-RtXVQPajPn6QjaWNYWjjfTUyRWv8MIaRNPJZ2P3fu0T_zNKYZOUpKZBmoYRDynBhSt4bKeh0HbCMmaX5zQu4vC6kWp4vt6oYXzQ260GXq1MtyKjrWFIydHScK4j4VlCSIk1l8ZliAPmTSNLcf1RfKDk-XBLVyxwLvT2I7perdzdzlTQu4U2KPk8djx5U9Tgbr3olUAsjwLE8mAyI1qakcRplGFedh7j28yw1egHP8trSJ2GZNIQn2pIaIY3YIvR4sLAxi3p4sklwrcqlco65E3Hq9ZsBB_1RsQlyHRSnJxTXKimcLOKqmu5VP-dvixxrGHhpDBdKkyTuorfwBThOBJ7jjBWV_GRI9mczUlaRnEaXGIZUR7xOA1GxW6lL3JVGmWkjMu37ISzMYdtcLwVGrgFxO7esp_Su1leme2LTac2fh-8mymaNtB4Oeux5U3zKwjrS5RX-51AV68Nk9jQN1w7PWr42I27wvftiic7LmrdpOUiCZUlLCrj8siE4JQJFPgM3mKPHLsqsFw2Y_t2_7hs4KH_oCzomgs4NL5oLu20bax507hi4L6roFkQZV9p5r7uSLnq1K7tBoMVb8H0fpPMzvDooMbHdG4uLOL7E0HcdUpAb7F_e0Q0X5LhUAm-ePIZvmppQd_6gJ1w6Ix9t_Ouh95dzV6p58ZrRrd9c7LhuSgvt82IBywk3zbg4Em3XNO3QPxxFD8y29XW0lgpjFd8mWojCWMWkuPiYEu3VSKA8I1uu_enKyfOO3Z-B_qDgsdtZ_9y_kuDAMKj_8JFciERRGk6s9nX5FDMl__jokzPVbUA8YmbaR3qQb0DZWDx-HJ1KKOERWl6RBadIiOV-GnIHtRslV0SEqnEEVK8gFTzrIrjGaR5xnzov3QPvbM5r3hv9xD1oBZ096DeniNzxZudkeaeK75xhfwdLruucb8H5c9-qwu-DoVBCFUcH_2RnPiDxxn5YX98kj00UsFJDnVDDzDjn-6g-SSn6oMyVvsX606_kLocMMsfe9s_cQ3KflB1N9v2LhWDOs5mCTg9iYEQMT0xfe-Mswxc-vJCNkeBiOnR5uzE5ohVHPeg6063ufA5_Dvv_MZWUxrZcg1V0duTKrIbtID7jUbs_SvbgUN11ykLzy-rlPtG6ofe3nOp7jpVy82isJg9fVlIUHIKjtBqVvd-7DZS8OYzmKGxs_56UB55ocd6-pUJiKYH7IOS_x7gBfZ7aDu9G3PnwlcV1HxobFFBA_5F5JU5U_k_r9ZPI5Ad6kkx1axFrRYV68xJb43lxfEcrZuOQz_DRhqrdy_J8S0C7MvwsCDje9L-LeV_IjLfoefibSYjEcnSaLEtN9LmqvqkOwHG7G39U0j3F1wS3xyxeP21WqqNP4h_pftCqbiOsjQ6pGJKl9kqq0kQnV8H_2-r-EK7c0iCmXvZqXtpWb_q3qkwEVs-liD5j1Lme765GGZa1kfM4SnmCOCnYP6TwEUAR3DR8nSdZjVJ8HSMeoGzfQE1SY7aTw-SU5IyXBSNLEVhLNe2-JETXWGG_dH52PQPvRx_tstC6q6T7uQUOU8Aj1ovhZ0ne88fPoxeVTesyljGr-AmSFiYxgEl5Gp7Q1hEoIxFzIETTkTGoixiWZQw4JwSeiVvKKFhQCklWcQoWaUsgTggWQI1iSoOKCTQctmsnJGrTm-u_Ie7m4CSNA2vGl5CY_yf1lA6flulKHp3pW88qHLYGBSSRhprjhKstI3_cxw_IXr3hq9s_psgmKtBNzd__FPjZPTTDf1PAAAA__8GkBeo">