-
Notifications
You must be signed in to change notification settings - Fork 68
[release/2.6] Change gfx110x BLAS preferred backend #2053
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: release/2.6
Are you sure you want to change the base?
Conversation
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as FAILURE |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit is in progress |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as FAILURE |
For upstream release/2.7 we applied this patch which adds a Default blas backend which then becomes cublas or cublaslt. pytorch#150212 For release/2.6, it should be as straightforward as this diff: diff --git a/aten/src/ATen/Context.cpp b/aten/src/ATen/Context.cpp
index a0e3b3d638..fbdbe767e3 100644
--- a/aten/src/ATen/Context.cpp
+++ b/aten/src/ATen/Context.cpp
@@ -320,7 +320,7 @@ at::BlasBackend Context::blasPreferredBackend() {
static const std::vector<std::string> archs = {
"gfx90a", "gfx942"
#if ROCM_VERSION >= 60300
- , "gfx1100", "gfx1101", "gfx1200", "gfx1201"
+ , "gfx1200", "gfx1201"
#endif
#if ROCM_VERSION >= 60500
, "gfx950" |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as FAILURE |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as FAILURE |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as FAILURE |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as ABORTED |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as FAILURE |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as FAILURE |
Jenkins build for d9b0d061725412951c47dce318dbf10d4823a297 commit finished as FAILURE |
Change gfx110x BLAS preferred backend from hipBLASLt to rocBLAS.