There was an error while loading. Please reload this page.
1 parent b7a4fec commit 94475d6Copy full SHA for 94475d6
src/cudamatrix/cu-device.cc
@@ -111,12 +111,14 @@ void CuDevice::Initialize() {
111
CUBLAS_SAFE_CALL(cublasCreate(&cublas_handle_));
112
CUBLAS_SAFE_CALL(cublasSetStream(cublas_handle_, cudaStreamPerThread));
113
114
+ #if CUDA_VERSION >= 9000
115
if (device_options_.use_tensor_cores) {
116
// Enable tensor cores in CUBLAS
117
// Note if the device does not support tensor cores this will fall back to normal math mode
118
CUBLAS_SAFE_CALL(cublasSetMathMode(cublas_handle_,
119
CUBLAS_TENSOR_OP_MATH));
120
}
121
+ #endif
122
123
// Initialize the cuSPARSE library
124
CUSPARSE_SAFE_CALL(cusparseCreate(&cusparse_handle_));
0 commit comments