torch-mlir

Commit Graph

Author	SHA1	Message	Date
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

Author

SHA1

Message

Date

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

1 Commits (bfb93cb99f259a614b4be7f6a4d04f6a07e4d395)