未验证 提交 354643d8 编写于 作者: W whs 提交者: GitHub

Add warning for cudnn warpctc kernel in CUDA9\CUDA10. (#18046)

test=develop
上级 e81756f1
...@@ -54,6 +54,15 @@ class WarpCTCOp : public framework::OperatorWithKernel { ...@@ -54,6 +54,15 @@ class WarpCTCOp : public framework::OperatorWithKernel {
framework::LibraryType library_{framework::LibraryType::kPlain}; framework::LibraryType library_{framework::LibraryType::kPlain};
#ifdef PADDLE_WITH_CUDA #ifdef PADDLE_WITH_CUDA
if (platform::CanCUDNNBeUsed(ctx)) { 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; library_ = framework::LibraryType::kCUDNN;
} }
#endif #endif
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册