fix cudnn enforce

7 years ago · 54e9fd3f61
parent 3ade95d0db
commit 54e9fd3f61
1 changed files with 6 additions and 7 deletions
--- a/paddle/fluid/platform/cudnn_helper.h
+++ b/paddle/fluid/platform/cudnn_helper.h
@ -62,9 +62,8 @@ inline const char* cudnnGetErrorString(cudnnStatus_t status) {
 #define CUDNN_ENFORCE(condition)                                     \
  do {                                                               \
    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)