pytorch
f0b06c64 - Fix bugs in sparse compressed tensor shape and device inference (#85240)

Commit
2 years ago
Fix bugs in sparse compressed tensor shape and device inference (#85240) Fixes #84999 This PR - uses device option to set sparse compressed tensor instance device - enables shape and device inference tests that was disabled due to an oversight - fixes a bug in shape inference of hybrid tensors - fixes a bug in to_sparse_bsr of a cuda tensor - updates tests that catch the above bugs Pull Request resolved: https://github.com/pytorch/pytorch/pull/85240 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
Loading