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
@@ -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)