pytorch_sparse
pytorch_sparse copied to clipboard
port `spmm_reduce` from `pytorch_sparse.matmul` to `pytorch.torch.matmul`
trafficstars
POC for spmm_sum: https://github.com/pytorch/pytorch/pull/83727 Links to other PRs: pyg - https://github.com/pyg-team/pytorch_geometric/pull/5498 pytorch_sparse - this pytorch - https://github.com/pytorch/pytorch/pull/85466