Skip to content
This repository was archived by the owner on Mar 12, 2021. It is now read-only.

Commit 4838670

Browse files
committed
Fix CUBLAS math mode under fast math (#510).
1 parent 1921827 commit 4838670

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/blas/CUBLAS.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,7 +43,7 @@ function handle()
4343
# enable tensor math mode if our device supports it, and fast math is enabled
4444
dev = CUDAdrv.device()
4545
if Base.JLOptions().fast_math == 1 && CUDAdrv.capability(dev) >= v"7.0" && version() >= v"9"
46-
cublasSetMathMode(CUBLAS_TENSOR_OP_MATH, handle)
46+
cublasSetMathMode(handle, CUBLAS_TENSOR_OP_MATH)
4747
end
4848

4949
handle

0 commit comments

Comments
 (0)