mirror of https://github.com/llvm/torch-mlir
163fa57cde
In the `pyhpc_turbulent_kinetic_energy` TorchBench benchmark, the shape calculation occurs inside loops, but because `DropShapeCalculationsPass` does not explicitly mark the Torch dialect as legal, the pass execution fails. This patch adds Torch to the list of legal dialects, and adds a test to validate the translation. |
||
|---|---|---|
| .. | ||
| AdjustCallingConventions.cpp | ||
| CMakeLists.txt | ||
| DecomposeComplexOps.cpp | ||
| DropShapeCalculations.cpp | ||
| GlobalizeObjectGraph.cpp | ||
| InlineGlobalSlots.cpp | ||
| MaximizeValueSemantics.cpp | ||
| PassDetail.h | ||
| Passes.cpp | ||
| PrepareForGlobalizeObjectGraph.cpp | ||
| ReduceOpVariants.cpp | ||
| RefinePublicReturn.cpp | ||
| RefineTypes.cpp | ||
| ReifyShapeCalculations.cpp | ||
| ShapeLibrary.cpp | ||
| SimplifyShapeCalculations.cpp | ||