polish codes

02c10ca9 · alinag · 244ba709 · 02c10ca9
隐藏空白更改
内联并排

Showing with 9 addition and 9 deletion

paddle/fluid/platform/dynload/cudnn.h paddle/fluid/platform/dynload/cudnn.h +9 -9

未找到文件。
--- a/paddle/fluid/platform/dynload/cudnn.h
+++ b/paddle/fluid/platform/dynload/cudnn.h
@@ -181,16 +181,16 @@ CUDNN_DNN_ROUTINE_EACH_R7(DECLARE_DYNAMIC_LOAD_CUDNN_WRAP)
 #endif

 #if CUDNN_VERSION >= 7201
-#define CUDNN_DNN_ROUTINE_EACH_AFTER_R72(__macro) \
-  __macro(cudnnCreateRNNDataDescriptor);          \
-  __macro(cudnnDestroyRNNDataDescriptor);         \
-  __macro(cudnnSetRNNDataDescriptor);             \
-  __macro(cudnnRNNForwardTrainingEx);             \
-  __macro(cudnnSetRNNPaddingMode);                \
-  __macro(cudnnRNNBackwardDataEx);                \
-  __macro(cudnnRNNBackwardWeightsEx);             \
+#define CUDNN_DNN_ROUTINE_EACH_AFTER_TWO_R7(__macro) \
+  __macro(cudnnCreateRNNDataDescriptor);             \
+  __macro(cudnnDestroyRNNDataDescriptor);            \
+  __macro(cudnnSetRNNDataDescriptor);                \
+  __macro(cudnnSetRNNPaddingMode);                   \
+  __macro(cudnnRNNForwardTrainingEx);                \
+  __macro(cudnnRNNBackwardDataEx);                   \
+  __macro(cudnnRNNBackwardWeightsEx);                \
  __macro(cudnnRNNForwardInferenceEx);
-CUDNN_DNN_ROUTINE_EACH_AFTER_R72(DECLARE_DYNAMIC_LOAD_CUDNN_WRAP)
+CUDNN_DNN_ROUTINE_EACH_AFTER_TWO_R7(DECLARE_DYNAMIC_LOAD_CUDNN_WRAP)
 #endif

 #if CUDNN_VERSION >= 7401