Support heur_b mode

tensorflow · Nov 30, 2021 · 174231c · 174231c
1 parent 229efc0
commit 174231c
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