torch-mlir/lib/Dialect/Torch
Aart Bik 46a25d7241
[torch-mlir][sparse] preserve sparsity during lowering torch to linalg (#2809)
This preserves sparsity at the most obvious places of lowering TORCH
tensors to MLIR RankedTensorType tensors. Other places are marked for
audit. With some initial lowering tests.
2024-01-26 10:54:59 -08:00
..
IR [torch-mlir][sparse] preserve sparsity during lowering torch to linalg (#2809) 2024-01-26 10:54:59 -08:00
Transforms [torch] `torch.dequantize` for per channel tensors to` linalg` (#2769) 2024-01-25 16:40:21 -08:00
Utils [TorchToLinalg] Add lowering for torch.aten.diagonal (#2632) 2024-01-22 12:47:13 -05:00
CMakeLists.txt E2e support for aten.softmax.int and aten.embedding 2021-10-18 17:57:45 -04:00