torch-mlir/lib/Dialect/TorchConversion
Yuanqiang Liu 88533b1968
[Stablehlo] fix aten.arange's lowering to stablehlo (#3138)
* promote to f64 to do division, avoid division on i64 (floor div)
* refactor torch-to-stablehlo-pipeline
2024-04-11 15:55:56 +08:00
..
IR [MLIR][Torch] Canonicalize torch.from_i1 and torch.to_i1 (#3067) 2024-04-01 14:25:51 -07:00
Transforms [Stablehlo] fix aten.arange's lowering to stablehlo (#3138) 2024-04-11 15:55:56 +08:00
CMakeLists.txt Move external/torch-mlir to the root of the repo. 2021-09-27 17:11:08 -07:00