mit-b0 pytorch model support

[AmosLewis]

https://github.com/nod-ai/e2eshark-reports/blob/main/2024-06-24/onnx_reports/statusreport.md input ir: mit-b0.default.pytorch.linalg.elide.mlir

iree version: candidate-20240624.934

/home/chi/src/iree-build/tools/iree-compile --iree-input-demote-i64-to-i32 --iree-hal-target-backends=llvm-cpu mit-b0.default.pytorch.linalg.elide.mlir > mit-b0.default.vmfb 2>iree-compile.log

iree-compile: iree/third_party/llvm-project/mlir/include/mlir/Dialect/Utils/IndexingUtils.h:204: SmallVector<T> mlir::applyPermutation(ArrayRef<T>, ArrayRef<int64_t>) [T = llvm::SmallVector<long, 2>]: Assertion `input.size() == permutation.size() && "expected input rank to equal permutation rank"' failed.
Please report issues to https://github.com/iree-org/iree/issues and include the crash backtrace.
Stack dump:
0.  Program arguments: /home/chi/src/iree-build/tools/iree-compile --iree-input-demote-i64-to-i32 --iree-hal-target-backends=llvm-cpu mit-b0.default.pytorch.linalg.mlir
Stack dump without symbol names (ensure you have llvm-symbolizer in your PATH or set the environment var `LLVM_SYMBOLIZER_PATH` to point to it):
0  libIREECompiler.so 0x000073a30d3f033d llvm::sys::PrintStackTrace(llvm::raw_ostream&, int) + 61
1  libIREECompiler.so 0x000073a30d3f082b
2  libIREECompiler.so 0x000073a30d3ee896 llvm::sys::RunSignalHandlers() + 134
3  libIREECompiler.so 0x000073a30d3f0fe5
4  libc.so.6          0x000073a301442520
5  libc.so.6          0x000073a3014969fc pthread_kill + 300
6  libc.so.6          0x000073a301442476 raise + 22
7  libc.so.6          0x000073a3014287f3 abort + 211
8  libc.so.6          0x000073a30142871b
9  libc.so.6          0x000073a301439e96
10 libIREECompiler.so 0x000073a310368e98
11 libIREECompiler.so 0x000073a310368dd9
12 libIREECompiler.so 0x000073a310368979
13 libIREECompiler.so 0x000073a3135599c5
14 libIREECompiler.so 0x000073a3135594e6
15 libIREECompiler.so 0x000073a313559057
16 libIREECompiler.so 0x000073a313558fef
17 libIREECompiler.so 0x000073a313558b45
18 libIREECompiler.so 0x000073a313558a91
19 libIREECompiler.so 0x000073a3103b87eb
20 libIREECompiler.so 0x000073a3156cf52e
21 libIREECompiler.so 0x000073a3156cf385
22 libIREECompiler.so 0x000073a30d3074a9
23 libIREECompiler.so 0x000073a3156d0d55
24 libIREECompiler.so 0x000073a3156cde27 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&)>) + 1463
25 libIREECompiler.so 0x000073a3156898db
26 libIREECompiler.so 0x000073a315688b61
27 libIREECompiler.so 0x000073a315688b35
28 libIREECompiler.so 0x000073a30d3074a9
29 libIREECompiler.so 0x000073a315688285
30 libIREECompiler.so 0x000073a31568670e
31 libIREECompiler.so 0x000073a3156863f7 mlir::applyPatternsAndFoldGreedily(mlir::Region&, mlir::FrozenRewritePatternSet const&, mlir::GreedyRewriteConfig, bool*) + 231
32 libIREECompiler.so 0x000073a30d24a105
33 libIREECompiler.so 0x000073a31031f23b
34 libIREECompiler.so 0x000073a30d80335b
35 libIREECompiler.so 0x000073a30d8032f5
36 libIREECompiler.so 0x000073a30d3074a9
37 libIREECompiler.so 0x000073a30d806175
38 libIREECompiler.so 0x000073a30d7feab3 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) + 851
39 libIREECompiler.so 0x000073a30d7ff034 mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) + 388
40 libIREECompiler.so 0x000073a30d8045e5
41 libIREECompiler.so 0x000073a30d804269
42 libIREECompiler.so 0x000073a30d8002eb
43 libIREECompiler.so 0x000073a30d7ffbfa mlir::detail::OpToOpPassAdaptor::runOnOperationAsyncImpl(bool) + 1194
44 libIREECompiler.so 0x000073a30d7ff727 mlir::detail::OpToOpPassAdaptor::runOnOperation(bool) + 71
45 libIREECompiler.so 0x000073a30d803346
46 libIREECompiler.so 0x000073a30d8032f5
47 libIREECompiler.so 0x000073a30d3074a9
48 libIREECompiler.so 0x000073a30d806175
49 libIREECompiler.so 0x000073a30d7feab3 mlir::detail::OpToOpPassAdaptor::run(mlir::Pass*, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int) + 851
50 libIREECompiler.so 0x000073a30d7ff034 mlir::detail::OpToOpPassAdaptor::runPipeline(mlir::OpPassManager&, mlir::Operation*, mlir::AnalysisManager, bool, unsigned int, mlir::PassInstrumentor*, mlir::PassInstrumentation::PipelineParentInfo const*) + 388
51 libIREECompiler.so 0x000073a30d800a78 mlir::PassManager::runPasses(mlir::Operation*, mlir::AnalysisManager) + 104
52 libIREECompiler.so 0x000073a30d8009a2 mlir::PassManager::run(mlir::Operation*) + 1138
53 libIREECompiler.so 0x000073a30d251cba
54 libIREECompiler.so 0x000073a30d251593 ireeCompilerInvocationPipeline + 35
55 libIREECompiler.so 0x000073a30d78a88e
56 libIREECompiler.so 0x000073a30d789d1e
57 libIREECompiler.so 0x000073a30d29baab ireeCompilerRunMain + 27
58 iree-compile       0x0000559f7d7347a2
59 libc.so.6          0x000073a301429d90
60 libc.so.6          0x000073a301429e40 __libc_start_main + 128
61 iree-compile       0x0000559f7d7346b5
Aborted (core dumped)

[AmosLewis]

pytorch/models/mobilebert-uncased same issues mobilebert-uncased.default.pytorch.linalg.elide.mlir