torch-mlir/test/python/fx_importer
Aart Bik 105aad6f57
[torch-mlir] provide FX traced graph importer for sparse tensors (#2817)
Note that we are waiting for actual FX traced graph support for sparse
tensors. For details see

https://github.com/pytorch/pytorch/issues/117188

Until then, however, we provide this clever importer that builds the FX
traced graph for for the dense case and then puts a sparse annotation
back on the parameters.

With import test.
2024-01-30 21:22:12 -08:00
..
basic_test.py [fx] Upstream the turbine FxImporter to torch-mlir. (#2681) 2023-12-21 08:40:10 -08:00
sparse_test.py [torch-mlir] provide FX traced graph importer for sparse tensors (#2817) 2024-01-30 21:22:12 -08:00