pytorch
b4431989 - Fix sparse addmv ref impl for non-contig tensors (#97730)

Commit
1 year ago
Fix sparse addmv ref impl for non-contig tensors (#97730) Fix logic in `test_block_addmm` that tested op against itself rather than against dense implementation, by implementing `ref_addvm` function that converts tensor back to dense before multiplying it with vector. Fix reference implementation by passing stride for vector and result. (Not sure wether it will be more perf efficient to iterate over strided tensor or request a dense copy as MKL implementation does) Print more verbose error message if values differ. Fixes https://github.com/pytorch/pytorch/issues/97629 , https://github.com/pytorch/pytorch/issues/97589 , https://github.com/pytorch/pytorch/issues/97563 Pull Request resolved: https://github.com/pytorch/pytorch/pull/97730 Approved by: https://github.com/cpuhrsch
Author
Committer
Parents
Loading