torch-mlir/python/torch_mlir
Aart Bik 4147b280ce
[torch-mlir][sparse] add block sparsity to mlir lowering (#2942)
Also note that we are in the process of proposing SparseTensorMetadata
to PyTorch FX graph export (see
https://github.com/pytorch/pytorch/pull/117907). This will hopefully
eventually replace the current data structures in torch-mlir.
2024-02-23 11:57:20 -08:00
..
_mlir_libs [ci] Upgrade to new runners and disable unsupported jobs. (#2818) 2024-01-27 18:35:45 -08:00
dialects Upstream the ONNX importer. (#2636) 2023-12-12 19:02:51 -08:00
extras [torch-mlir][sparse] add block sparsity to mlir lowering (#2942) 2024-02-23 11:57:20 -08:00
tools/import_onnx [onnx] Add testing using the `onnx` compilation using torch tests (#2795) 2024-02-15 10:17:13 -08:00
fx.py [fx] Support mutation in ExportedProgram. (#2916) 2024-02-16 09:46:30 -08:00