diff --git a/src/cudamatrix/cu-device.cc b/src/cudamatrix/cu-device.cc index 140275d3b6e..85c2492c074 100644 --- a/src/cudamatrix/cu-device.cc +++ b/src/cudamatrix/cu-device.cc @@ -111,12 +111,14 @@ void CuDevice::Initialize() { CUBLAS_SAFE_CALL(cublasCreate(&cublas_handle_)); CUBLAS_SAFE_CALL(cublasSetStream(cublas_handle_, cudaStreamPerThread)); + #if CUDA_VERSION >= 9000 if (device_options_.use_tensor_cores) { // Enable tensor cores in CUBLAS // Note if the device does not support tensor cores this will fall back to normal math mode CUBLAS_SAFE_CALL(cublasSetMathMode(cublas_handle_, CUBLAS_TENSOR_OP_MATH)); } + #endif // Initialize the cuSPARSE library CUSPARSE_SAFE_CALL(cusparseCreate(&cusparse_handle_));