fix cudnn enforce

54e9fd3f · typhoonzero · 3ade95d0 · 54e9fd3f
隐藏空白更改
内联并排

Showing with 6 addition and 7 deletion

paddle/fluid/platform/cudnn_helper.h paddle/fluid/platform/cudnn_helper.h +6 -7

未找到文件。
--- a/paddle/fluid/platform/cudnn_helper.h
+++ b/paddle/fluid/platform/cudnn_helper.h
@@ -59,13 +59,12 @@ inline const char* cudnnGetErrorString(cudnnStatus_t status) {
 #define CUDNN_VERSION_MIN(major, minor, patch) \
  (CUDNN_VERSION >= ((major)*1000 + (minor)*100 + (patch)))
-#define CUDNN_ENFORCE(condition)                                  \
+#define CUDNN_ENFORCE(condition)                                     \
-  do {                                                            \
+  do {                                                               \
-    cudnnStatus_t status = condition;                             \
+    cudnnStatus_t status = condition;                                \
-    if (status != CUDNN_STATUS_SUCCESS) {                         \
+    if (UNLIKELY(status != CUDNN_STATUS_SUCCESS)) {                  \
-      VLOG(1) << ::paddle::platform::cudnnGetErrorString(status); \
+      PADDLE_THROW(::paddle::platform::cudnnGetErrorString(status)); \
-      PADDLE_THROW("cuDNN call failed");                          \
+    }                                                                \
-    }                                                             \
  } while (false)
 enum class DataLayout {  // Not use