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)
46
46
47
47
flags |= if mode == CUDA. PEDANTIC_MATH
48
48
# 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
54
50
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
61
52
elseif mode == CUDA. FAST_MATH
62
53
# 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
68
55
end
69
56
70
57
cublasSetMathMode (handle, cublasMath_t (flags))
You can’t perform that action at this time.
0 commit comments