Skip to content

Commit a5b5bd7

Browse files
authored
Get rid of unneeded version checks (#2765)
1 parent 5645ddc commit a5b5bd7

File tree

1 file changed

+3
-16
lines changed

1 file changed

+3
-16
lines changed

lib/cublas/CUBLAS.jl

Lines changed: 3 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -46,25 +46,12 @@ function math_mode!(handle, mode)
4646

4747
flags |= if mode == CUDA.PEDANTIC_MATH
4848
# prevent use of tensor cores
49-
if version() < v"11"
50-
CUBLAS_DEFAULT_MATH
51-
else
52-
CUBLAS_PEDANTIC_MATH
53-
end
49+
CUBLAS_PEDANTIC_MATH
5450
elseif mode == CUDA.DEFAULT_MATH
55-
# use tensor cores, but don't reduce precision
56-
if version() < v"11"
57-
CUBLAS_TENSOR_OP_MATH
58-
else
59-
CUBLAS_DEFAULT_MATH
60-
end
51+
CUBLAS_DEFAULT_MATH
6152
elseif mode == CUDA.FAST_MATH
6253
# we'll additionally select a compute-mode with reduced precision whenever possible
63-
if version() < v"11"
64-
CUBLAS_TENSOR_OP_MATH
65-
else
66-
CUBLAS_TF32_TENSOR_OP_MATH
67-
end
54+
CUBLAS_TF32_TENSOR_OP_MATH
6855
end
6956

7057
cublasSetMathMode(handle, cublasMath_t(flags))

0 commit comments

Comments
 (0)