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
1616c32a · Yiqun Liu · GitHub · 79892545 · 1616c32a
隐藏空白更改
内联并排

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 {