Skip to content

GitLab

"...git@developer.sourcefind.cn:gaoqiong/pybind11.git" did not exist on "b3d9c3543dcbebde30185c163208b8a00793ae45"
Commit e80f260d authored by yuguo's avatar yuguo
Browse files

[DCU] fix rocblas backend

parent 3e001bbd
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
transformer_engine/common/gemm/rocm_gemm.cu
View file @ e80f260d
...@@ -1815,12 +1815,14 @@ void cublas_gemm(const Tensor *inputA, const Tensor *inputB, Tensor *outputD, ...@@ -1815,12 +1815,14 @@ void cublas_gemm(const Tensor *inputA, const Tensor *inputB, Tensor *outputD,
if (use_hipblaslt) if (use_hipblaslt)
{ {
use_hipblaslt = false; use_hipblaslt = false;
use_rocblas = true;
std::cout << "[NOTICE] hipBLASLt is not enabled, NVTE_USE_HIPBLASLT env is ignored\n"; std::cout << "[NOTICE] hipBLASLt is not enabled, NVTE_USE_HIPBLASLT env is ignored\n";
} }
#elif !defined(USE_ROCBLAS) #elif !defined(USE_ROCBLAS)
if (use_rocblas) if (use_rocblas)
{ {
use_rocblas = false; use_rocblas = false;
use_hipblaslt = true;
std::cout << "[NOTICE] rocBLAS is not enabled, NVTE_USE_ROCBLAS env is ignored\n"; std::cout << "[NOTICE] rocBLAS is not enabled, NVTE_USE_ROCBLAS env is ignored\n";
} }
#else #else
...@@ -1828,12 +1830,12 @@ void cublas_gemm(const Tensor *inputA, const Tensor *inputB, Tensor *outputD, ...@@ -1828,12 +1830,12 @@ void cublas_gemm(const Tensor *inputA, const Tensor *inputB, Tensor *outputD,
{ {
use_rocblas = false; use_rocblas = false;
use_hipblaslt = true; use_hipblaslt = true;
std::cout << "[NOTICE] Two GEMM backend are enabled, hipBLASLt will be used\n"; // std::cout << "[NOTICE] Two GEMM backend are enabled, hipBLASLt will be used\n";
} else if (!use_hipblaslt && !use_rocblas) } else if (!use_hipblaslt && !use_rocblas)
{ {
use_rocblas = false; use_rocblas = false;
use_hipblaslt = true; use_hipblaslt = true;
std::cout << "[NOTICE] Two GEMM backend are disabled, hipBLASLt will be used\n"; // std::cout << "[NOTICE] Two GEMM backend are disabled, hipBLASLt will be used\n";
} }
#endif #endif
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment