From 354643d8d9d8418c4426f30d035200e7b18f6c40 Mon Sep 17 00:00:00 2001
From: whs <wanghaoshuang@baidu.com>
Date: Fri, 14 Jun 2019 12:45:11 +0800
Subject: [PATCH] Add warning for cudnn warpctc kernel in CUDA9\CUDA10.
 (#18046)

test=develop
---
 paddle/fluid/operators/warpctc_op.cc | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/paddle/fluid/operators/warpctc_op.cc b/paddle/fluid/operators/warpctc_op.cc
index 217d400bb3c..deb5681f210 100644
--- a/paddle/fluid/operators/warpctc_op.cc
+++ b/paddle/fluid/operators/warpctc_op.cc
@@ -54,6 +54,15 @@ class WarpCTCOp : public framework::OperatorWithKernel {
     framework::LibraryType library_{framework::LibraryType::kPlain};
 #ifdef PADDLE_WITH_CUDA
     if (platform::CanCUDNNBeUsed(ctx)) {
+#if CUDA_VERSION >= 9000
+      LOG(WARNING)
+          << "The cudnnCTCLoss of CUDNN7 have some diff between "
+             "CUDA9/CUDA10 and CUDA8. You can close use_cudnn option to "
+             "use "
+             "baidu-research/warp-ctc(https://github.com/baidu-research/"
+             "warp-ctc)";
+#endif
+
       library_ = framework::LibraryType::kCUDNN;
     }
 #endif
-- 
GitLab