diff --git a/tensorflow/stream_executor/cuda/cuda_dnn.cc b/tensorflow/stream_executor/cuda/cuda_dnn.cc index e4e9914adf0..6c227622072 100644 --- a/tensorflow/stream_executor/cuda/cuda_dnn.cc +++ b/tensorflow/stream_executor/cuda/cuda_dnn.cc @@ -1123,8 +1123,13 @@ class CudnnRnnDescriptor : public dnn::RnnDescriptor { "Algo requests disallowed tensor op evaluation."); } +#if CUDNN_VERSION >= 8000 + cudnnMathType_t math_type = + use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_FMA_MATH; +#else cudnnMathType_t math_type = use_tensor_ops ? CUDNN_TENSOR_OP_MATH : CUDNN_DEFAULT_MATH; +#endif #if CUDNN_VERSION >= 8000 cudnnRNNBiasMode_t bias_mode = CUDNN_RNN_DOUBLE_BIAS;