pytorch
b407d98d - [cuBLAS] Fix default cuBLAS workspace size and parsing for multiple workspaces (#89027)

Commit
1 year ago
[cuBLAS] Fix default cuBLAS workspace size and parsing for multiple workspaces (#89027) Follow-up of #86167 ; The number of pools was mistakenly ignored and the default workspace size appears to be too small to match selected cuBLAS kernels before the explicit allocation change. CC @ptrblck @ngimel Pull Request resolved: https://github.com/pytorch/pytorch/pull/89027 Approved by: https://github.com/ngimel
Author
eqy eqy
Committer
Parents
Loading