File tree Expand file tree Collapse file tree 1 file changed +3
-16
lines changed Expand file tree Collapse file tree 1 file changed +3
-16
lines changed Original file line number Diff line number Diff 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))
You can’t perform that action at this time.
0 commit comments