pytorch
3488a85a - Sparse CSR CUDA: fix input checks for `addmm` and `mm` (#66485)

Commit
3 years ago
Sparse CSR CUDA: fix input checks for `addmm` and `mm` (#66485) Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/66485 The errors for incorrectly sized inputs should match the dense variants of functions. Moved addmm_out_sparse_csr_dense_cuda from SparseCsrTensorMath.cu and removed unnecessary device check. cc nikitaved pearu cpuhrsch IvanYashchuk Test Plan: Imported from OSS Reviewed By: jbschlosser Differential Revision: D31764036 Pulled By: cpuhrsch fbshipit-source-id: 76900fe9e4a49474695a01f34bad41cb3422321c
Author
Parents
Loading