torch-mlir/projects/pt1/e2e_testing
Giacomo Serafini 46a5772d92
[TorchToLinalg] Add `aten.fft_rfft` and lowering (#3857)
- Add `AtenFftRfftOp` to Torch dialect.
- Add conversion of `AtenFftRfftOp` to Linalg, using a `linalg.matmul`
per output component (real and imaginary). Computing the DFT is
_O(n^2)_.
- Add decomposition of `AtenFftRfftOp` into Torch-level ops (same
paradigm as above).
- Add unit and end-to-end tests.
2024-11-27 10:24:36 -06:00
..
main.py [TOSA] Add Torch to Tosa Legalization for torch.tril (#3678) 2024-09-05 11:27:29 -07:00
xfail_sets.py [TorchToLinalg] Add `aten.fft_rfft` and lowering (#3857) 2024-11-27 10:24:36 -06:00