pytorch
e6101f55 - fixes lda condition for blas functions, fixes bug with beta=0 in addmv slow path (#44681)

Commit
4 years ago
fixes lda condition for blas functions, fixes bug with beta=0 in addmv slow path (#44681) Summary: per title. If `beta=0` and slow path was taken, `nan` and `inf` in the result were not masked as is the case with other linear algebra functions. Similarly, since `mv` is implemented as `addmv` with `beta=0`, wrong results were sometimes produced for `mv` slow path. Pull Request resolved: https://github.com/pytorch/pytorch/pull/44681 Reviewed By: mruberry Differential Revision: D23708653 Pulled By: ngimel fbshipit-source-id: e2d5d3e6f69b194eb29b327e1c6f70035f3b231c
Author
Natalia Gimelshein
Parents
Loading