torch-mlir/lib/Conversion/TorchToLinalg
Felix Schneider df6098e43d
[TorchToLinalg] Use `linalg.transpose` instead of `generic` when lowering `aten.T` (#3660)
The lowering pattern for `aten.T` uses transposition implemented via
`linalg.generic`. For downstream passes it is advantageous to use named
ops wherever possible, so this patch changes the lowering to use
`linalg.transpose` instead.
2024-09-07 08:09:10 +02:00
..
CMakeLists.txt Re-organize project structure to separate PyTorch dependencies from core project. (#2542) 2023-11-02 19:45:55 -07:00
DataMovement.cpp [TorchToLinalg] Use `linalg.transpose` instead of `generic` when lowering `aten.T` (#3660) 2024-09-07 08:09:10 +02:00
IndirectDataMovement.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
Linear.cpp [TorchToLinalg] Use `linalg.broadcast` instead of `generic` for conv bias (#3661) 2024-08-26 20:29:11 +02:00
Pooling.cpp [TorchToLinalg] Support lowering MaxPool3dWithIndices (#3652) 2024-08-27 14:14:25 -05:00
PopulatePatterns.h Re-enable custom op support 2022-08-16 22:49:08 +05:30
Random.cpp [TorchToLinalg] address a dtype mismatch in `aten.multinomial` lowering (#3630) 2024-08-20 15:14:48 -05:00
Reduction.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
TensorConstructors.cpp Adds misc fixes for some padding related issues (#3528) 2024-07-11 20:01:45 -05:00
TensorScalarInterop.cpp [NFC] Change to *cast instead of .*cast variants (#3405) 2024-05-30 23:45:13 -07:00
TorchToLinalg.cpp [TorchToLinalg][ONNX] Add Basic Determinant Support (#3481) 2024-06-25 13:34:19 -05:00
Uncategorized.cpp [MLIR][TORCH] Add E2E support for aten.polar op (#3671) 2024-09-03 10:51:03 +05:30
Utils.cpp Fix unused variable warning from assertion variable (#3512) 2024-06-28 12:20:29 -04:00