Add the include of cudnn.h to enable the use of CUDNN_VERSION. (#15961)

test=develop

Add the include of cudnn.h to enable the use of CUDNN_VERSION. (#15961)
test=develop
2bdf4464 · Yiqun Liu · ceci3 · b94307a9 · 2bdf4464
显示空白变更内容
内联并排

Showing with 3 addition and 1 deletion

paddle/fluid/inference/api/paddle_pass_builder.cc paddle/fluid/inference/api/paddle_pass_builder.cc +3 -1

未找到文件。
--- a/paddle/fluid/inference/api/paddle_pass_builder.cc
+++ b/paddle/fluid/inference/api/paddle_pass_builder.cc
@@ -13,7 +13,9 @@
 // limitations under the License.

 #include "paddle/fluid/inference/api/paddle_pass_builder.h"
-
+#ifdef PADDLE_WITH_CUDA
+#include <cudnn.h>
+#endif
 #include <glog/logging.h>

 namespace paddle {