Commit e80f260d authored by yuguo's avatar yuguo
Browse files

[DCU] fix rocblas backend

parent 3e001bbd
......@@ -1815,12 +1815,14 @@ void cublas_gemm(const Tensor *inputA, const Tensor *inputB, Tensor *outputD,
if (use_hipblaslt)
{
use_hipblaslt = false;
use_rocblas = true;
std::cout << "[NOTICE] hipBLASLt is not enabled, NVTE_USE_HIPBLASLT env is ignored\n";
}
#elif !defined(USE_ROCBLAS)
if (use_rocblas)
{
use_rocblas = false;
use_hipblaslt = true;
std::cout << "[NOTICE] rocBLAS is not enabled, NVTE_USE_ROCBLAS env is ignored\n";
}
#else
......@@ -1828,12 +1830,12 @@ void cublas_gemm(const Tensor *inputA, const Tensor *inputB, Tensor *outputD,
{
use_rocblas = false;
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)
{
use_rocblas = false;
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
......
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