pytorch/benchmarks/sparse/__init__.py at e98e7fe42811b656bd272e92044df67ccb2d3880 - saymrwulf/pytorch - Forgejo: Beyond coding. We forge.

saymrwulf/pytorch

mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00

Sameer Deshmukh 5fb1142702 Add CSR (compressed sparse row) layout for sparse tensors (#50937 )

Summary:
Implement compressed sparse row format. Derived from the GCS implementation at https://github.com/pytorch/pytorch/pull/44190

Pull Request resolved: https://github.com/pytorch/pytorch/pull/50937

Reviewed By: mrshenli

Differential Revision: D27439865

Pulled By: ezyang

fbshipit-source-id: 3ba3dcb9679505b980ff6a5f513e913bbae2fb1d

2021-04-12 10:09:12 -07:00

3 lines

37 B

Python

Raw Blame History


	`if __name__ == "__main__":`
	`pass`