torch-mlir/lib/Conversion
Ramiro Leal-Cavazos f85f5799e4
Fix creation of empty tensor in decomposition for randn ops (#2043)
The current decomposition for `aten.randn.generator` does not specify
the `dtype` argument of the empty tensors created to store the random
values. This leads to invalid IR when the output type of the `randn`
op is not the default PyTorch dtype.
2023-04-19 08:25:39 -07:00
..
TorchConversionToMLProgram Fix a few missing dependencies. (#2014) 2023-04-11 11:18:49 -07:00
TorchToArith [MLIR][TORCH] Extend implementation of aten._index_put_impl op. 2023-04-05 14:04:30 +05:30
TorchToLinalg Fix creation of empty tensor in decomposition for randn ops (#2043) 2023-04-19 08:25:39 -07:00
TorchToSCF build: update llvm tag to 798fa4b4 (#1684) 2022-12-07 12:20:41 -08:00
TorchToStablehlo [MLIR][TORCH] Patch up Ops and their lowerings to deal with +ve `dim` 2023-04-14 13:12:56 +05:30
TorchToTMTensor [MLIR][TORCH] Add e2e support for aten.sort op 2023-04-13 12:59:43 +05:30
TorchToTosa [TOSA] Add torch.prim.NumToTensor.Scalar float support (#1802) 2023-04-18 13:36:57 -07:00
Utils [MLIR][TORCH] Extend implementation of aten._index_put_impl op. 2023-04-05 14:04:30 +05:30
CMakeLists.txt mhlo: migrate conversion to stablehlo (#1840) 2023-02-02 07:29:47 -06:00
PassDetail.h Minor fixes for `ConvertTorchConversionToMLProgram`. (#1991) 2023-04-04 09:09:58 -07:00
Passes.cpp mhlo: migrate conversion to stablehlo (#1840) 2023-02-02 07:29:47 -06:00