Aart Bik ff82e2e7cf [traced-graph][sparse] propagate sparsity metadata into traced graph (#117907) ... Propagate sparsity metadata from sparse tensors of torch.sparse into the traced graph representation (with would be useful for a JIT backend that supports a "sparse compiler"). This is a first careful attempt, since the actual "meta" feature seem still incomplete for coo and completely lacking for csr/csc/bsr/bsc. For background see forum postings (with examples): https://discuss.pytorch.org/t/connecting-pytorch-sparse-tensors-with-mlir/195145 https://dev-discuss.pytorch.org/t/connecting-pytorch-sparse-tensors-with-mlir/1803 And feature request: https://github.com/pytorch/pytorch/issues/117188 Pull Request resolved: https://github.com/pytorch/pytorch/pull/117907 Approved by: https://github.com/pearu, https://github.com/ezyang		2024-05-23 22:46:46 +00:00
..
__init__.py
opinfo_schema.py	[export] add SchemaCheckMode testing for pre-dispatch export, OpInfo (#125481)	2024-05-14 21:07:21 +00:00
test_converter.py	TorchScript 2 ExportedProgram Converter (#126920)	2024-05-23 17:00:18 +00:00
test_db.py
test_experimental.py	Experimental prototype for converting torch.jit.trace modules to export (#124449)	2024-05-17 20:42:42 +00:00
test_export.py	[export] Fix unflatten with empty nn_module_stack (#126785)	2024-05-23 08:34:25 +00:00
test_export_nonstrict.py
test_functionalized_assertions.py
test_hop.py
test_lift_unlift.py	[export] handle constant aliasing for export (#125509)	2024-05-10 00:14:37 +00:00
test_pass_infra.py	switch tests from constrain_as* to torch._check* (#125253)	2024-05-01 21:01:27 +00:00
test_passes.py	[export] Add metadata for nodes insert_deferred_runtime_asserts (#125414)	2024-05-07 23:15:21 +00:00
test_retraceability.py
test_schema.py
test_serdes.py	[export] Delete predispatch tests (#126459)	2024-05-17 00:48:32 +00:00
test_serialize.py	[RELAND] Switch default behavoir of export IR to be predispatch (#125860)	2024-05-10 17:36:53 +00:00
test_sparse.py	[traced-graph][sparse] propagate sparsity metadata into traced graph (#117907)	2024-05-23 22:46:46 +00:00
test_tools.py	Tool for scouting exportability in one shot (#126471)	2024-05-18 00:10:46 +00:00
test_torchbind.py	[export] handle constant aliasing for export (#125509)	2024-05-10 00:14:37 +00:00
test_tree_utils.py
test_unflatten.py	[export] handle aliased/unused params for unflattening (#125758)	2024-05-14 23:00:46 +00:00
test_verifier.py
testing.py