<table border="1" cellspacing="0" cellpadding="8">
    <tr>
        <th>Issue</th>
        <td>
            <a href=https://github.com/llvm/llvm-project/issues/68948>68948</a>
        </td>
    </tr>

    <tr>
        <th>Summary</th>
        <td>
            [mlir] --ownership-based-buffer-deallocation crashed with assertion failure "Value must already have been registered in the ownership map"
        </td>
    </tr>

    <tr>
      <th>Labels</th>
      <td>
            mlir
      </td>
    </tr>

    <tr>
      <th>Assignees</th>
      <td>
      </td>
    </tr>

    <tr>
      <th>Reporter</th>
      <td>
          wangzhiyingh
      </td>
    </tr>
</table>

<pre>
    git version: eb601430d3d7f45c30ef8d793a45cbcedf910577

system: `Ubuntu 20.04.6 LTS (Focal Fossa)`

reproduced with:: `mlir-opt --ownership-based-buffer-deallocation a.mlir`

a.mlir:
```
func.func private @func1() -> memref<?x?x?xf16> {
  %15 = tensor.empty() : tensor<15x2x5xf16>
  %cast = tensor.cast %15 : tensor<15x2x5xf16> to tensor<?x?x?xf16>
  %16 = bufferization.to_memref %cast : memref<?x?x?xf16>
  return %16 : memref<?x?x?xf16>
}
```

trace:
```
mlir-opt: /data/bin/llvm-project/mlir/lib/Dialect/Bufferization/IR/BufferDeallocationOpInterface.cpp:139: std::pair<mlir::Value, mlir::Value> mlir::bufferization::DeallocationState::getMemrefWithUniqueOwnership(mlir::OpBuilder&, mlir::Value, mlir::Block*): Assertion `iter != ownershipMap.end() && "Value must already have been registered in the ownership map"' 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/bin/llvm-project/build/bin/mlir-opt --ownership-based-buffer-deallocation a.mlir
 #0 0x0000557a1a69993f llvm::sys::PrintStackTrace(llvm::raw_ostream&, int) (/data/bin/llvm-project/build/bin/mlir-opt+0x15a693f)
 #1 0x0000557a1a6969d4 SignalHandler(int) Signals.cpp:0:0
 #2 0x00007ffa917f0420 __restore_rt (/lib/x86_64-linux-gnu/libpthread.so.0+0x14420)
 #3 0x00007ffa912db00b raise (/lib/x86_64-linux-gnu/libc.so.6+0x4300b)
 #4 0x00007ffa912ba859 abort (/lib/x86_64-linux-gnu/libc.so.6+0x22859)
 #5 0x00007ffa912ba729 (/lib/x86_64-linux-gnu/libc.so.6+0x22729)
 #6 0x00007ffa912cbfd6 (/lib/x86_64-linux-gnu/libc.so.6+0x33fd6)
 #7 0x0000557a1aaf4be5 mlir::bufferization::DeallocationState::getMemrefWithUniqueOwnership(mlir::OpBuilder&, mlir::Value, mlir::Block*) (/data/bin/llvm-project/build/bin/mlir-opt+0x1a01be5)
 #8 0x0000557a1abb5cff (anonymous namespace)::BufferDeallocation::materializeUniqueOwnership(mlir::OpBuilder&, mlir::Value, mlir::Block*) OwnershipBasedBufferDeallocation.cpp:0:0
 #9 0x0000557a1abb5e14 (anonymous namespace)::BufferDeallocation::materializeMemrefWithGuaranteedOwnership(mlir::OpBuilder&, mlir::Value, mlir::Block*) OwnershipBasedBufferDeallocation.cpp:0:0
#10 0x0000557a1abba68b (anonymous namespace)::BufferDeallocation::deallocate(mlir::Block*) OwnershipBasedBufferDeallocation.cpp:0:0
#11 0x0000557a1abbac00 mlir::WalkResult llvm::function_ref<mlir::WalkResult (mlir::Block*)>::callback_fn<(anonymous namespace)::BufferDeallocation::deallocate(mlir::FunctionOpInterface)::'lambda'(mlir::Block*)>(long, mlir::Block*) OwnershipBasedBufferDeallocation.cpp:0:0
#12 0x0000557a1abbbcce mlir::WalkResult mlir::detail::walk<mlir::ForwardDominanceIterator<false>>(mlir::Operation*, llvm::function_ref<mlir::WalkResult (mlir::Block*)>, mlir::WalkOrder) (/data/bin/llvm-project/build/bin/mlir-opt+0x1ac8cce)
#13 0x0000557a1abbc1cb mlir::bufferization::deallocateBuffersOwnershipBased(mlir::FunctionOpInterface, bool) (/data/bin/llvm-project/build/bin/mlir-opt+0x1ac91cb)
#14 0x0000557a1abbcfa3 _ZN4llvm12function_refIFN4mlir10WalkResultEPNS1_9OperationEEE11callback_fnIZNS1_6detail4walkILNS1_9WalkOrderE1ENS1_15ForwardIteratorEZN12_GLOBAL__N_136OwnershipBasedBufferDeallocationPass14runOnOperationEvEUlNS1_4func6FuncOpEE_SF_S2_EENSt9enable_ifIXaantsrSt11disjunctionIJSt7is_sameIT2_S4_ESJ_ISK_PNS1_6RegionEESJ_ISK_PNS1_5BlockEEEE5valuesrSJ_IT3_S2_E5valueEST_E4typeES4_OT1_EUlS4_E_EES2_lS4_ OwnershipBasedBufferDeallocation.cpp:0:0
#15 0x0000557a1a7d9da2 mlir::WalkResult mlir::detail::walk<mlir::ForwardIterator>(mlir::Operation*, llvm::function_ref<mlir::WalkResult (mlir::Operation*)>, mlir::WalkOrder) (/data/bin/llvm-project/build/bin/mlir-opt+0x16e6da2)
#16 0x0000557a1abb3672 (anonymous namespace)::OwnershipBasedBufferDeallocationPass::runOnOperation() OwnershipBasedBufferDeallocation.cpp:0:0
#17 0x0000557a1d15115e mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) (/data/bin/llvm-project/build/bin/mlir-opt+0x405e15e)
#18 0x0000557a1d15163a mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) (/data/bin/llvm-project/build/bin/mlir-opt+0x405e63a)
#19 0x0000557a1d151e64 mlir::PassManager::run(mlir::Operation*) (/data/bin/llvm-project/build/bin/mlir-opt+0x405ee64)
#20 0x0000557a1d14272b performActions(llvm::raw_ostream&, std::shared_ptr<llvm::SourceMgr> const&, mlir::MLIRContext*, mlir::MlirOptMainConfig const&) MlirOptMain.cpp:0:0
#21 0x0000557a1d143be4 processBuffer(llvm::raw_ostream&, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, mlir::MlirOptMainConfig const&, mlir::DialectRegistry&, llvm::ThreadPool*) MlirOptMain.cpp:0:0
#22 0x0000557a1d143d94 mlir::LogicalResult llvm::function_ref<mlir::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&)::'lambda'(std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&)>(long, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&) MlirOptMain.cpp:0:0
#23 0x0000557a1d2406a4 mlir::splitAndProcessBuffer(std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::function_ref<mlir::LogicalResult (std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&)>, llvm::raw_ostream&, bool, bool) (/data/bin/llvm-project/build/bin/mlir-opt+0x414d6a4)
#24 0x0000557a1d13d067 mlir::MlirOptMain(llvm::raw_ostream&, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, mlir::DialectRegistry&, mlir::MlirOptMainConfig const&) (/data/bin/llvm-project/build/bin/mlir-opt+0x404a067)
#25 0x0000557a1d1440bb mlir::MlirOptMain(int, char**, llvm::StringRef, mlir::DialectRegistry&) (/data/bin/llvm-project/build/bin/mlir-opt+0x40510bb)
#26 0x0000557a1a5d24ab main (/data/bin/llvm-project/build/bin/mlir-opt+0x14df4ab)
#27 0x00007ffa912bc083 __libc_start_main (/lib/x86_64-linux-gnu/libc.so.6+0x24083)
#28 0x0000557a1a66a2ee _start (/data/bin/llvm-project/build/bin/mlir-opt+0x15772ee)
Aborted (core dumped)
```
</pre>
<img width="1px" height="1px" alt="" src="http://email.email.llvm.org/o/eJzcWl1z27iS_TX0C8oqEPx-8INsU7OatWNX5OxszQurSTQlTCiQC4CJPb9-CyQlkbKdaGLPndybilUiQBz06T4NogmB1mItES-c4NIJrs-gNZtaXXwFuf5zI56EXG_O8po_XayFIV9QaVFLx5sTzEPq-h7lHo9KPyg8imXMo8QDPyjyAnmZuDSIIodeO3Tef-onbXBrRzsh_ZS30rSE0Rn1ZyG5eVgRh8WLuoCKLGqtwWGJE9LxeIWNqnlbICdfhdk43nzA2lZCndeNIefn9VeJSm9Ec56DRn6et2WJ6pwjVFVdgBG1JDCzA47Ah0ZvuLS9_f_usmxlMbMfpFHiCxgkjk_tteuw2GEJOXe8lGxxq7B0vCvHWzzu_ko3tH1OdNlDEeKwwA2I410Tg1LXaobbxjwNQJZS3-x4V27wyB6DAWI0vABtxgD99QD72nhi6kPPcwvH1oUdeO878WfntZmps57fyID5tyjvABWaVsk97gljnOj6xSj0n0ZBga8FaieFThlswcGAwxa5kA5bVNWX7Xmj6j-wMA5bdPFmi0rkDltcC6j65ssxa4ctlh_3jdcjFd01S2lQlVDgrGgax5u7XmJn1Yb3ymzA6ulqJytv_j9QteiwK3LcZKWzb5p4vW8az7syYLBvXqO57Vz5mzCbT1L8X4t3O_k7LD5A3jWXrag4KoeFL80_abqs6uKzw-Y2_7w5mWuNqksbJ6TCoCIOc6069pl2C80MJd_p104REoexDptsW20IVAqBP5ENfEGSI0qicC20QYWcCEnMBg94ZAuNw5jDIlKCqJDP-tDe36TzVUp0m2-FIUDydk0UNrUyVtkbYxptGbCFwxZrYTZtPivq7RD35-EXWreoHbYgIK0RRdVy7CwpFOgNyaH43EltmH5loPhMeLtt9tqjM9L9u1f1WsGWgFq3W5RGn6C-3EZk3_ljK1ifYA7zKKGPlFIaBBG4ECZJ4pWko92FVD_p_su9EtJ0RB66JGLx4SYFX7NaG4WwHWQipOkjGv8QF4dd0kc3gDDxSiumvbXukbVhwn2yEmsJ1X-B5JXVaTxM3jfrIcVo97cHYgNQVJaQuFFJfUZJlinUplaYKTPY3qf4YxxmoX9eCdk-nq9l23c0ZmO1OdP1jPYW-z6jE3u9yTSM55TmRIHQeAJ-YZHDDtn3KM0nyP4UOYc4SAjk9UmWj5AZi4NkghwcI0f7SJ6MGbEpZjjFLPKSh38N0_NKHk4wo4kWoPRzDH7OxfCNeQDUzTGYcI8n3PM8KEr7cI1B1vJpW7eaSNiibrpMTQaDnj2I-vYtGFQCKvEn_i3k93CXdlF6bsXLCZocM0TXfweGh0D_0oICaRD5P87XLmz0iC-Ecf6jfPcLPk4Yvd1I99jIgtKRF36D6vNH1G1lRk8Qu9W1iFm_cXvx7tfMtNu6rq2AqrJP1ayUdvP3rm5ZDAaOtmV7KIdFFWxzDg6LvmUki6tart9bE-zI3XlR4MvuPjRyNCCq_vtXqD5PXL6o1VdQ_LreCgmywKVBBabb1ZdQaez8nR6lgb2l28_OLb_3C-zEWXbMneqy7c3LZREXfQj3jvSOHFm4Rf7tR8VBK32w9DSC3xfQFcnrunoPNolb5BM2_jGbEjyS_f7Bt7AuG8dlufjgWzyXHoKS3n9YuVmyD2yapq47SrDl77Y_7IXkWw0tb7oR-xilbmob3GDQ005G6e8fXJb9cnN3Ob_Jsg-Z64XfE_49aO36qpV38mDQl_RTZSfwLZXQuveuSdNstchWLEvTDyuToIS8wkyUy_8FkEarlXFdLvQfA_nlrysTCZ1p2OLygWUrP0tXv2bL1X9nHf3wI6477uPGoFNomqZp8MWu8VrZ3gevm7ZvSlcPWeqbpwbTlZ_dPbhZ-qmy4Fmarlhmv_5gsgeTqEY84cDeI9kPOf63JfYU6m9N7hBDDmySDuFROnhhxL735DxFlkNtM5HmUKn-WIQnG1buBq4b4CvBvGse6rvGWjHn0HTRG4yZOL4zsw_f64E99MwlVE9a6FuQsLbR2C1SV6SV3cs8_i6Vm08DdIPpEhwfkw89-Mvk70WDlZB4pD57655T-O7eGA-yUy2lNqqr1mv1DHV6w-iJsrP9HhRKs5RlTYpaavMeJYJ1eOjBxOHJscMx9I_s3DF_UVtHSf12AzH0xwYyOjXQZxHLSYOqrNV23i1A-jtvGfbvy_QGFPKsMXYncxixqltV4O3arn07Zx_J4_Zm-fGqlgYfzbNQ3lZC3TXmFoS8qmUp1iOMhIx6X0x35h7x83L0SaPqArXul4xT6bVdWfaM3i1ua_W0gxrdz7GEtjIZxwptkfvamGG3dyrn8X3DS8-P3bs49TT0H-Z56F6N3Hf5ND_JXezYXTwZ6_WmXosCqtPLi-kAh8X_One-HtTXKpoXA_ATyuPlsJ-YMi-WVD9NVA4l3E9h0vcTZlLUcObTEMYJo5tKmLnk90crzj_B7t8wSb95x2Gn8C5Fnu_6PITp09GfLocep2H0cqL9py0Tb91p-EDDaOLL4OjR4tM8f9WXw5av2EC_uTsqjlZGCbn-iOX3Gb99z-TSfFL6s2mtE3DmQ062IOQbyyqflz5Mp4qOjgEKGnskyyqRF5k2oEw2mvfUgwGfxt5klumL7DAEhkh6_LeeHkURw0MNMs9rZZBb0KJW2J3GIT_YsjsLPuMXHk-8BM7wwg2TyA39KArPNheR62NAacJjVpQ0oYlLQ4xzDhgnfhLgmbhglHkudT3KvMQLZ5RTjBI3yP2Ye7HHHJ_iFkQ1syxmtVqfdQeJF2Gc-PFZBTlWuvsxBWP9ATNzguszddGRztu1dnxaCW30AcAIU3U_wOgGBNcnHgJ2x5TDbyEI7I9pSxBVq_AdTmDPWlVd_Pi5aueR_w8AAP__NJewrA">