pytorch
721a8ca6 - Sparse CSR: enable autograd for torch.sparse.addmm and torch.sparse.mm

Comment changes are shownComment changes are hidden
Commit
3 years ago
Sparse CSR: enable autograd for torch.sparse.addmm and torch.sparse.mm This PR updates the derivative rule for `torch.sparse.addmm` to be working with CSR sparse matrix. Notably `torch.sparse.sampled_addmm` is used in the backward function. Pull Request resolved: https://github.com/pytorch/pytorch/pull/76591 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
  • aten/src/ATen/native
    • File
      native_functions.yaml
  • test
    • forward_backward_compatibility
      • File
        check_forward_backward_compatibility.py
    • File
      test_sparse_csr.py
  • tools/autograd
    • File
      derivatives.yaml
  • torch/csrc/autograd
    • File
      FunctionsManual.cpp
    • File
      FunctionsManual.h