diff options
author | Hieu Hoang <hieuhoang@gmail.com> | 2018-01-16 19:24:11 +0300 |
---|---|---|
committer | Hieu Hoang <hieuhoang@gmail.com> | 2018-01-16 19:24:11 +0300 |
commit | ed9ca006fd2dd4427b9b4bb884e9bc50846f6c8b (patch) | |
tree | b4b0fd6a21f9286fd26b03f928fd35a8db0b7b14 | |
parent | 7b4ef6322ef925737d4eb7e257ec93ac77eaa518 (diff) |
disable tensorcore support
-rw-r--r-- | src/amun/gpu/mblas/handles.cu | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/amun/gpu/mblas/handles.cu b/src/amun/gpu/mblas/handles.cu index 13c99f8c..baa96639 100644 --- a/src/amun/gpu/mblas/handles.cu +++ b/src/amun/gpu/mblas/handles.cu @@ -28,13 +28,13 @@ CublasHandler::CublasHandler() } #if CUDA_VERSION >= 9000 - ///* + /* stat = cublasSetMathMode(handle_, CUBLAS_TENSOR_OP_MATH); if (stat != CUBLAS_STATUS_SUCCESS) { printf ("cublasSetMathMode failed\n"); abort(); } - //*/ + */ #endif stat = cublasSetStream(handle_, CudaStreamHandler::GetStream()); |