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

test=develop
6 years ago · 1616c32acf
parent 798925453e
commit 1616c32acf
1 changed files with 3 additions and 1 deletions
--- 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 {