Missing passes in ONNX to LinAlg lowering pass makes llama2 torch MLIR to LinAlg lowering fail

[kumardeepakamd]

Vivek please add details

[vivekkhandelwal1]

I have added the lowering pipeline here: llvm/torch-mlir#2844

[vivekkhandelwal1]

I'm getting the following crash:

torch-mlir-opt: /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/Support/Casting.h:566: decltype(auto) llvm::cast(const From &) [To = mlir::DenseElementsAttr, From = mlir::Attribute]: Assertion `isa<To>(Val) && "cast<Ty>() argument of incompatible type!"' failed.
PLEASE submit a bug report to https://github.com/llvm/llvm-project/issues/ and include the crash backtrace.
Stack dump:
0.      Program arguments: /home/azureuser/work/torch-mlir-vivek/build/bin/torch-mlir-opt -onnx-backend-to-linalg-on-tensors-backend-pipeline llama-7b-stripped.torch-onnx.mlir
 #0 0x000055dde77edb6d llvm::sys::PrintStackTrace(llvm::raw_ostream&, int) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/lib/Support/Unix/Signals.inc:723:11
 #1 0x000055dde77ee05b PrintStackTraceSignalHandler(void*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/lib/Support/Unix/Signals.inc:798:1
 #2 0x000055dde77ec086 llvm::sys::RunSignalHandlers() /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/lib/Support/Signals.cpp:105:5
 #3 0x000055dde77ee875 SignalHandler(int) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/lib/Support/Unix/Signals.inc:413:1
 #4 0x00007fbfb7442520 (/lib/x86_64-linux-gnu/libc.so.6+0x42520)
 #5 0x00007fbfb74969fc pthread_kill (/lib/x86_64-linux-gnu/libc.so.6+0x969fc)
 #6 0x00007fbfb7442476 gsignal (/lib/x86_64-linux-gnu/libc.so.6+0x42476)
 #7 0x00007fbfb74287f3 abort (/lib/x86_64-linux-gnu/libc.so.6+0x287f3)
 #8 0x00007fbfb742871b (/lib/x86_64-linux-gnu/libc.so.6+0x2871b)
 #9 0x00007fbfb7439e96 (/lib/x86_64-linux-gnu/libc.so.6+0x39e96)
#10 0x000055dde403c558 decltype(auto) llvm::cast<mlir::DenseElementsAttr, mlir::Attribute>(mlir::Attribute const&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/Support/Casting.h:567:43
#11 0x000055dde403c355 mlir::DenseElementsAttr mlir::Attribute::cast<mlir::DenseElementsAttr>() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/../mlir/include/mlir/IR/Attributes.h:190:10
#12 0x000055dde400fa0e getScalarIntValue(mlir::Value, mlir::Location, mlir::PatternRewriter&) /home/azureuser/work/torch-mlir-vivek/lib/Dialect/Torch/IR/TorchOps.cpp:152:21
#13 0x000055dde400f15b rewrite0DBinaryTensorOp(mlir::Operation*, mlir::PatternRewriter&) /home/azureuser/work/torch-mlir-vivek/lib/Dialect/Torch/IR/TorchOps.cpp:1029:14
#14 0x000055dde401bfa2 mlir::torch::Torch::AtenMulTensorOp::getCanonicalizationPatterns(mlir::RewritePatternSet&, mlir::MLIRContext*)::$_17::operator()(mlir::torch::Torch::AtenMulTensorOp, mlir::PatternRewriter&) const /home/azureuser/work/torch-mlir-vivek/lib/Dialect/Torch/IR/TorchOps.cpp:1165:12
#15 0x000055dde401bf5d mlir::torch::Torch::AtenMulTensorOp::getCanonicalizationPatterns(mlir::RewritePatternSet&, mlir::MLIRContext*)::$_17::__invoke(mlir::torch::Torch::AtenMulTensorOp, mlir::PatternRewriter&) /home/azureuser/work/torch-mlir-vivek/lib/Dialect/Torch/IR/TorchOps.cpp:1163:17
#16 0x000055dde405b4ae mlir::RewritePatternSet& mlir::RewritePatternSet::add<mlir::torch::Torch::AtenMulTensorOp>(mlir::LogicalResult (*)(mlir::torch::Torch::AtenMulTensorOp, mlir::PatternRewriter&), mlir::PatternBenefit, llvm::ArrayRef<llvm::StringRef>)::FnPattern::matchAndRewrite(mlir::torch::Torch::AtenMulTensorOp, mlir::PatternRewriter&) const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/../mlir/include/mlir/IR/PatternMatch.h:851:16
#17 0x000055dde405b40b mlir::detail::OpOrInterfaceRewritePatternBase<mlir::torch::Torch::AtenMulTensorOp>::matchAndRewrite(mlir::Operation*, mlir::PatternRewriter&) const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/../mlir/include/mlir/IR/PatternMatch.h:330:12
#18 0x000055dde6073b36 mlir::PatternApplicator::matchAndRewrite(mlir::Operation*, mlir::PatternRewriter&, llvm::function_ref<bool (mlir::Pattern const&)>, llvm::function_ref<void (mlir::Pattern const&)>, llvm::function_ref<mlir::LogicalResult (mlir::Pattern const&)>)::$_1::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Rewrite/PatternApplicator.cpp:208:31
#19 0x000055dde6073975 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<mlir::LogicalResult (mlir::Pattern const&)>)::$_1>(long) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:45:5
#20 0x000055dde39848b9 llvm::function_ref<void ()>::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:68:5
#21 0x000055dde607540d void mlir::MLIRContext::executeAction<mlir::ApplyPatternAction, mlir::Pattern const&>(llvm::function_ref<void ()>, llvm::ArrayRef<mlir::IRUnit>, mlir::Pattern const&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/IR/MLIRContext.h:276:3
#22 0x000055dde6072335 mlir::PatternApplicator::matchAndRewrite(mlir::Operation*, mlir::PatternRewriter&, llvm::function_ref<bool (mlir::Pattern const&)>, llvm::function_ref<void (mlir::Pattern const&)>, llvm::function_ref<mlir::LogicalResult (mlir::Pattern const&)>) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Rewrite/PatternApplicator.cpp:229:9
#23 0x000055dde602e94e (anonymous namespace)::GreedyPatternRewriteDriver::processWorklist() /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp:586:17
#24 0x000055dde602dc31 (anonymous namespace)::RegionPatternRewriteDriver::simplify(bool*) &&::$_8::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp:820:30
#25 0x000055dde602dc05 void llvm::function_ref<void ()>::callback_fn<(anonymous namespace)::RegionPatternRewriteDriver::simplify(bool*) &&::$_8>(long) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:45:5
#26 0x000055dde39848b9 llvm::function_ref<void ()>::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:68:5
#27 0x000055dde602d2ed void mlir::MLIRContext::executeAction<(anonymous namespace)::GreedyPatternRewriteIteration, long&>(llvm::function_ref<void ()>, llvm::ArrayRef<mlir::IRUnit>, long&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/IR/MLIRContext.h:276:3
#28 0x000055dde602b69d (anonymous namespace)::RegionPatternRewriteDriver::simplify(bool*) && /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp:828:3
#29 0x000055dde602b35f mlir::applyPatternsAndFoldGreedily(mlir::Region&, mlir::FrozenRewritePatternSet const&, mlir::GreedyRewriteConfig, bool*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Transforms/Utils/GreedyPatternRewriteDriver.cpp:860:47
#30 0x000055dde3924ed5 mlir::applyPatternsAndFoldGreedily(mlir::Operation*, mlir::FrozenRewritePatternSet const&, mlir::GreedyRewriteConfig, bool*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/Transforms/GreedyPatternRewriteDriver.h:148:9
#31 0x000055dde391ceda (anonymous namespace)::Canonicalizer::runOnOperation() /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Transforms/Canonicalizer.cpp:63:9
#32 0x000055dde6131bdb mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int)::$_7::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:0:17
#33 0x000055dde6131b75 void llvm::function_ref<void ()>::callback_fn<mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int)::$_7>(long) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:45:5
#34 0x000055dde39848b9 llvm::function_ref<void ()>::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:68:5
#35 0x000055dde6134c3d void mlir::MLIRContext::executeAction<mlir::PassExecutionAction, mlir::Pass&>(llvm::function_ref<void ()>, llvm::ArrayRef<mlir::IRUnit>, mlir::Pass&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/IR/MLIRContext.h:276:3
#36 0x000055dde612d0a3 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:525:17
#37 0x000055dde612d654 mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:585:16
#38 0x000055dde6132fe8 mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::$_15::operator()(mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo&) const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:805:36
#39 0x000055dde6132c59 mlir::LogicalResult mlir::failableParallelForEach<__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)::$_15&>(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)::$_15&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/IR/Threading.h:46:18
#40 0x000055dde612e9a3 mlir::LogicalResult mlir::failableParallelForEach<std::vector<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo, std::allocator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo>>&, mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::$_15&>(mlir::MLIRContext*, std::vector<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo, std::allocator<mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::OpPMInfo>>&, mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool)::$_15&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/IR/Threading.h:92:10
#41 0x000055dde612e247 mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:815:14
#42 0x000055dde612dd67 mlir::detail::OpToOpPassAdaptor::runOnOperation(bool) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:706:5
#43 0x000055dde6131bc6 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int)::$_7::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:517:11
#44 0x000055dde6131b75 void llvm::function_ref<void ()>::callback_fn<mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int)::$_7>(long) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:45:5
#45 0x000055dde39848b9 llvm::function_ref<void ()>::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:68:5
#46 0x000055dde6134c3d void mlir::MLIRContext::executeAction<mlir::PassExecutionAction, mlir::Pass&>(llvm::function_ref<void ()>, llvm::ArrayRef<mlir::IRUnit>, mlir::Pass&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/IR/MLIRContext.h:276:3
#47 0x000055dde612d0a3 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:525:17
#48 0x000055dde612d654 mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:585:16
#49 0x000055dde6131b51 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int)::$_6::operator()(mlir::OpPassManager&, mlir::Operation*) const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:502:12
#50 0x000055dde61318d2 mlir::LogicalResult llvm::function_ref<mlir::LogicalResult (mlir::OpPassManager&, mlir::Operation*)>::callback_fn<mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int)::$_6>(long, mlir::OpPassManager&, mlir::Operation*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:45:12
#51 0x000055dde3961da1 llvm::function_ref<mlir::LogicalResult (mlir::OpPassManager&, mlir::Operation*)>::operator()(mlir::OpPassManager&, mlir::Operation*) const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:68:12
#52 0x000055dde3961ba5 mlir::Pass::runPipeline(mlir::OpPassManager&, mlir::Operation*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/Pass/Pass.h:196:12
#53 0x000055dde450c8ea (anonymous namespace)::LowerToBackendContractPass::runOnOperation() /home/azureuser/work/torch-mlir-vivek/lib/Dialect/Torch/Transforms/LowerToBackendContract.cpp:304:18
#54 0x000055dde6131bdb mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int)::$_7::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:0:17
#55 0x000055dde6131b75 void llvm::function_ref<void ()>::callback_fn<mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int)::$_7>(long) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:45:5
#56 0x000055dde39848b9 llvm::function_ref<void ()>::operator()() const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:68:5
#57 0x000055dde6134c3d void mlir::MLIRContext::executeAction<mlir::PassExecutionAction, mlir::Pass&>(llvm::function_ref<void ()>, llvm::ArrayRef<mlir::IRUnit>, mlir::Pass&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/include/mlir/IR/MLIRContext.h:276:3
#58 0x000055dde612d0a3 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:525:17
#59 0x000055dde612d654 mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:585:16
#60 0x000055dde612f13c mlir::PassManager::runPasses(mlir::Operation*, mlir::AnalysisManager) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:896:10
#61 0x000055dde612f05f mlir::PassManager::run(mlir::Operation*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Pass/Pass.cpp:876:60
#62 0x000055dde3902642 performActions(llvm::raw_ostream&, std::shared_ptr<llvm::SourceMgr> const&, mlir::MLIRContext*, mlir::MlirOptMainConfig const&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Tools/mlir-opt/MlirOptMain.cpp:394:17
#63 0x000055dde3902279 processBuffer(llvm::raw_ostream&, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, mlir::MlirOptMainConfig const&, mlir::DialectRegistry&, llvm::ThreadPool*) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Tools/mlir-opt/MlirOptMain.cpp:459:12
#64 0x000055dde390206b mlir::MlirOptMain(llvm::raw_ostream&, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, mlir::DialectRegistry&, mlir::MlirOptMainConfig const&)::$_3::operator()(std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&) const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Tools/mlir-opt/MlirOptMain.cpp:530:12
#65 0x000055dde3901fed 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&)::$_3>(long, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:45:12
#66 0x000055dde76bc9f9 llvm::function_ref<mlir::LogicalResult (std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&)>::operator()(std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, llvm::raw_ostream&) const /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/llvm/include/llvm/ADT/STLFunctionalExtras.h:68:12
#67 0x000055dde76bbfd5 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) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Support/ToolUtilities.cpp:28:12
#68 0x000055dde38febd9 mlir::MlirOptMain(llvm::raw_ostream&, std::unique_ptr<llvm::MemoryBuffer, std::default_delete<llvm::MemoryBuffer>>, mlir::DialectRegistry&, mlir::MlirOptMainConfig const&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Tools/mlir-opt/MlirOptMain.cpp:533:10
#69 0x000055dde38fee7d mlir::MlirOptMain(int, char**, llvm::StringRef, llvm::StringRef, mlir::DialectRegistry&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Tools/mlir-opt/MlirOptMain.cpp:568:14
#70 0x000055dde38ff058 mlir::MlirOptMain(int, char**, llvm::StringRef, mlir::DialectRegistry&) /home/azureuser/work/torch-mlir-vivek/externals/llvm-project/mlir/lib/Tools/mlir-opt/MlirOptMain.cpp:584:10
#71 0x000055dde38f9e25 main /home/azureuser/work/torch-mlir-vivek/tools/torch-mlir-opt/torch-mlir-opt.cpp:43:33
#72 0x00007fbfb7429d90 (/lib/x86_64-linux-gnu/libc.so.6+0x29d90)
#73 0x00007fbfb7429e40 __libc_start_main (/lib/x86_64-linux-gnu/libc.so.6+0x29e40)
#74 0x000055dde38f9cb5 _start (/home/azureuser/work/torch-mlir-vivek/build/bin/torch-mlir-opt+0x23cacb5)
Aborted (core dumped)

Working on it.