pytorch
affa3f84 - Sparse CSR CPU: add `torch.addmm` (#65606)

Commit
3 years ago
Sparse CSR CPU: add `torch.addmm` (#65606) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/65606 This PR adds `torch.addmm(c, a, b, alpha=1.0, beta=0.0, out=out)` variant with `a, b, c, out` all being sparse CSR tensors on CPU. cc nikitaved pearu cpuhrsch IvanYashchuk Test Plan: Imported from OSS Reviewed By: mrshenli Differential Revision: D32366236 Pulled By: cpuhrsch fbshipit-source-id: e910bcc96eee99d624b80ee881df3887ab3ba5ac
Author
Parents
Loading