diff --git a/tensorflow/stream_executor/cuda/cuda_dnn.cc b/tensorflow/stream_executor/cuda/cuda_dnn.cc index 623c48eda66665..fb4c5523f22864 100644 --- a/tensorflow/stream_executor/cuda/cuda_dnn.cc +++ b/tensorflow/stream_executor/cuda/cuda_dnn.cc @@ -3327,7 +3327,11 @@ dnn::DataType GetConvAccumulatorType(dnn::DataType data_type) { #if CUDNN_VERSION >= 8100 && TF_ENABLE_CUDNN_FRONTEND cudnnBackendHeurMode_t GetCudnnFrontendHeurMode() { +#if CUDNN_VERSION >= 8300 + return CUDNN_HEUR_MODE_B; +#else return CUDNN_HEUR_MODE_INSTANT; +#endif // CUDNN_VERSION >= 8300 } #endif // CUDNN_VERSION >= 8100 && TF_ENABLE_CUDNN_FRONTEND