Merge pull request #53253 from kaixih:two_level_heur_b

PiperOrigin-RevId: 420147695 Change-Id: I5a3bbb3ed8557b33af75bb5ab794a40237e872d5
tensorflow · Jan 6, 2022 · eced720 · eced720
2 parents 7390025 + 174231c
commit eced720
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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