Skip to content

Commit edf10bb

Browse files
authored
Update the error message for cublas version check (NVIDIA#2843)
update the error message for cublas version check Signed-off-by: Xin Yao <xiny@nvidia.com>
1 parent fdf9fb1 commit edf10bb

1 file changed

Lines changed: 5 additions & 5 deletions

File tree

transformer_engine/common/gemm/cublaslt_grouped_gemm.cu

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1363,7 +1363,7 @@ void nvte_grouped_gemm(const NVTEGroupedTensor A, int transa, const NVTEGroupedT
13631363
NVTETensor workspace_cublas, NVTEGroupedMatmulConfig config,
13641364
cudaStream_t stream) {
13651365
NVTE_ERROR("nvte_grouped_gemm requires cuBLAS 13.3+, but compile-time cuBLAS version is ",
1366-
CUBLAS_VERSION, ". Please upgrade to CUDA 13.3 or newer.");
1366+
CUBLAS_VERSION, ". Please upgrade to cuBLAS 13.3 (shipped with CUDA 13.2) or newer.");
13671367
}
13681368

13691369
void nvte_grouped_gemm_with_discrete_inputA(const NVTETensor *A_list, size_t num_a_tensors,
@@ -1375,7 +1375,7 @@ void nvte_grouped_gemm_with_discrete_inputA(const NVTETensor *A_list, size_t num
13751375
NVTE_ERROR(
13761376
"nvte_grouped_gemm_with_discrete_inputA requires cuBLAS 13.3+, but compile-time "
13771377
"cuBLAS version is ",
1378-
CUBLAS_VERSION, ". Please upgrade to CUDA 13.3 or newer.");
1378+
CUBLAS_VERSION, ". Please upgrade to cuBLAS 13.3 (shipped with CUDA 13.2) or newer.");
13791379
}
13801380

13811381
void nvte_grouped_gemm_with_discrete_out(const NVTEGroupedTensor A, int transa,
@@ -1388,20 +1388,20 @@ void nvte_grouped_gemm_with_discrete_out(const NVTEGroupedTensor A, int transa,
13881388
NVTE_ERROR(
13891389
"nvte_grouped_gemm_with_discrete_out requires cuBLAS 13.3+, but compile-time "
13901390
"cuBLAS version is ",
1391-
CUBLAS_VERSION, ". Please upgrade to CUDA 13.3 or newer.");
1391+
CUBLAS_VERSION, ". Please upgrade to cuBLAS 13.3 (shipped with CUDA 13.2) or newer.");
13921392
}
13931393

13941394
void nvte_grouped_bias_add(const NVTEGroupedTensor output, const NVTEGroupedTensor bias,
13951395
cudaStream_t stream) {
13961396
NVTE_ERROR("nvte_grouped_bias_add requires cuBLAS 13.3+, but compile-time cuBLAS version is ",
1397-
CUBLAS_VERSION, ". Please upgrade to CUDA 13.3 or newer.");
1397+
CUBLAS_VERSION, ". Please upgrade to cuBLAS 13.3 (shipped with CUDA 13.2) or newer.");
13981398
}
13991399

14001400
size_t nvte_get_grouped_gemm_setup_workspace_size(size_t num_tensors) {
14011401
NVTE_ERROR(
14021402
"nvte_get_grouped_gemm_setup_workspace_size requires cuBLAS 13.3+, but compile-time cuBLAS "
14031403
"version is ",
1404-
CUBLAS_VERSION, ". Please upgrade to CUDA 13.3 or newer.");
1404+
CUBLAS_VERSION, ". Please upgrade to cuBLAS 13.3 (shipped with CUDA 13.2) or newer.");
14051405
return 0;
14061406
}
14071407

0 commit comments

Comments
 (0)