mirror of https://github.com/llvm/torch-mlir
11cd7cd9e7
While playing with TorchDynamo on ResNet18. I notice following issues: - `prims.convert_element_type` can’t be canonicalized even if the input and the output share the same type - `aten.max_pool2d_with_indices` is always used instead of `aten.max_pool2d`, even if the second returned output (indices) has no user This PR fixes above issues by adding a folder to the PrimsConvertElementTypeOp and a canonicalizer to the AtenMaxPool2dWithIndicesOp Lit test: `cmake --build build --target check-torch-mlir-all` --------- Co-authored-by: Ze Zhang <ze.zhang@getcruise.com> |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| TorchDialect.cpp | ||
| TorchOps.cpp | ||
| TorchOpsODSGenerated.cpp | ||
| TorchTypes.cpp | ||
| UtilsForODSGenerated.cpp | ||
| UtilsForODSGenerated.h | ||