Skip to content

Commit c5b0392

Browse files
authored
Change deprecated cublasGemmAlgo_t type (#1938)
1 parent aa7f562 commit c5b0392

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/cuda/primitives.cu

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -503,7 +503,7 @@ namespace ctranslate2 {
503503
beta_ptr,
504504
c, CUDA_R_16F, ldc,
505505
compute_type,
506-
CUBLAS_GEMM_DEFAULT_TENSOR_OP));
506+
CUBLAS_GEMM_DEFAULT));
507507
}
508508

509509
template<>
@@ -528,7 +528,7 @@ namespace ctranslate2 {
528528
&beta,
529529
c, CUDA_R_16BF, ldc,
530530
CUDA_R_32F,
531-
CUBLAS_GEMM_DEFAULT_TENSOR_OP));
531+
CUBLAS_GEMM_DEFAULT));
532532
}
533533

534534
template<>
@@ -556,7 +556,7 @@ namespace ctranslate2 {
556556
&beta_i,
557557
c, CUDA_R_32I, ldc,
558558
CUDA_R_32I,
559-
CUBLAS_GEMM_DEFAULT_TENSOR_OP));
559+
CUBLAS_GEMM_DEFAULT));
560560
}
561561

562562
template<>
@@ -617,7 +617,7 @@ namespace ctranslate2 {
617617
c, CUDA_R_16F, ldc, stridec,
618618
batch_size,
619619
compute_type,
620-
CUBLAS_GEMM_DEFAULT_TENSOR_OP));
620+
CUBLAS_GEMM_DEFAULT));
621621
}
622622

623623
template<>
@@ -642,7 +642,7 @@ namespace ctranslate2 {
642642
c, CUDA_R_16BF, ldc, stridec,
643643
batch_size,
644644
CUDA_R_32F,
645-
CUBLAS_GEMM_DEFAULT_TENSOR_OP));
645+
CUBLAS_GEMM_DEFAULT));
646646
}
647647

648648
template <typename T>

0 commit comments

Comments
 (0)