torch-mlir

Commit Graph

Author	SHA1	Message	Date
Aart Bik	a033bbfe6c	[torch-mlir][sparse] recognize to_dense primitive (#3308 ) also maps simply to sparse_tensor.convert the sparsity types do the rest!	2024-05-08 22:50:17 -07:00
Aart Bik	4361178caa	[torch-mlir][sparse] recognize sparse tensor conversion (#3226 ) Sparse tensor conversions are represented by special aten operators. This PR ensures the conversions are recognized (instead of failing the full torch aten lowering to linalg).	2024-04-26 02:32:07 +08:00
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

Author

SHA1

Message

Date

Aart Bik

a033bbfe6c

[torch-mlir][sparse] recognize to_dense primitive (#3308 )

also maps simply to sparse_tensor.convert
the sparsity types do the rest!

2024-05-08 22:50:17 -07:00

Aart Bik

4361178caa

[torch-mlir][sparse] recognize sparse tensor conversion (#3226 )

Sparse tensor conversions are represented by special aten operators.
This PR ensures the conversions are recognized (instead of failing the
full torch aten lowering to linalg).

2024-04-26 02:32:07 +08:00

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

3 Commits (431d98b405900f2cb2cc816b9c742f292ff5f4e6)