torch-mlir/lib/Dialect
Ashay Rane 163fa57cde
torch: allow torch dialect ops after running drop-shape pass (#979)
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.
2022-06-25 07:27:47 -07:00
..
Torch torch: allow torch dialect ops after running drop-shape pass (#979) 2022-06-25 07:27:47 -07:00
TorchConversion [MLIR][TORCH] Add folder for torch_c.from_i64 & torch_c.to_i64 (#933) 2022-06-24 09:34:39 +08:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00