frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 05:34:18 +08:00
Code Issues Packages Projects Releases Wiki Activity

[ROCm] change preferred blas lib defaults (#150212)

Browse Source 
Fixes #148883
Fixes #150155

Also adds at::BlasBackend:Default. Instinct cards prefer hipBLASLt, everything else prefers rocBLAS.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150212
Approved by: https://github.com/jeffdaily
This commit is contained in:
Nichols A. Romero
2025-03-29 03:33:04 +00:00
committed by PyTorch MergeBot
parent 29b3fdab01
commit 7a470c9320
6 changed files with 43 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
torch/csrc/Module.cpp
View File

@ -2243,6 +2243,7 @@ Call this whenever a new thread is created in order to propagate values from
});
py::enum_<at::BlasBackend>(py_module, "_BlasBackend")
.value("Default", at::BlasBackend::Default)
.value("Cublas", at::BlasBackend::Cublas)
.value("Cublaslt", at::BlasBackend::Cublaslt)
.value("Ck", at::BlasBackend::Ck);