torch-mlir/lib/Dialect/Torch
Rob Suderman 9ab93436c4
[torch] Support diagonal `einsum.Diagonal` (#3618)
The einsum lowering was missing the behavior for duplicate indices in
the equation. This amounts to a diagonalization along duplicate pairs of
indices in the equation.
2024-08-13 09:38:43 -07:00
..
IR [torch] Unpacking sometimes misses shape inference (#3609) 2024-08-08 16:17:31 -07:00
Transforms [torch] Support diagonal `einsum.Diagonal` (#3618) 2024-08-13 09:38:43 -07:00
Utils [TorchToLinalg] Fix Quantized Convolution Accumulator Type (#3459) 2024-06-20 13:54:20 -07:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00