Aart Bik a8490a0762 [traced-graph][sparse] propagate sparsity in fx graph (#131920) ... This PR proceeds with implementing the feature request #117188 by generalizing more cases that already work with COO to work with the compressed sparse formats as well. Feature request: https://github.com/pytorch/pytorch/issues/117188 Rebranch of older PRs (for history): https://github.com/pytorch/pytorch/pull/131474 https://github.com/pytorch/pytorch/pull/128549 Pull Request resolved: https://github.com/pytorch/pytorch/pull/131920 Approved by: https://github.com/ezyang		2024-08-05 15:49:53 +00:00
..
__init__.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
_fake_tensor_utils.py	[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)	2024-07-31 10:42:50 +00:00
fake_impls.py	Add pinned memory support to sparse COO/CSR/CSC/BSR/BSC tensors (#129645)	2024-08-02 08:55:55 +00:00
fake_tensor.py	[traced-graph][sparse] propagate sparsity in fx graph (#131920)	2024-08-05 15:49:53 +00:00
fake_utils.py	Skip frame if torch dispatch mode enabled (#131828)	2024-08-01 19:06:20 +00:00
functional_tensor.py	[traced-graph][sparse] propagate sparsity in fx graph (#131920)	2024-08-05 15:49:53 +00:00
meta_utils.py	[traced-graph][sparse] propagate sparsity in fx graph (#131920)	2024-08-05 15:49:53 +00:00
schema_check_mode.py	Add None return type to init (#132335)	2024-08-01 15:26:45 +00:00