Skip to content

Commit df5d19a

Browse files
authored
temoprarily disable the code of use kCUDNN, test=develop (#16205)
* temoprarily disable the code of use kCUDNN, test=develop * add TODO comment, test=develop
1 parent 09e05a1 commit df5d19a

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

paddle/fluid/operators/activation_op.cc

Lines changed: 10 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -76,12 +76,16 @@ framework::OpKernelType GetKernelType(const framework::ExecutionContext& ctx,
7676
const std::string& name) {
7777
framework::LibraryType library{framework::LibraryType::kPlain};
7878
framework::DataLayout layout = framework::DataLayout::kAnyLayout;
79-
#ifdef PADDLE_WITH_CUDA
80-
auto it1 = oper.Attrs().find("use_cudnn");
81-
if (it1 != oper.Attrs().end() && platform::CanCUDNNBeUsed(ctx)) {
82-
library = framework::LibraryType::kCUDNN;
83-
}
84-
#endif
79+
// FIXME(liuwei1031) temporarily disable the code to unblock users
80+
// TODO(liuwei1031) figure out the reason behind
81+
// https://github.com/PaddlePaddle/Paddle/issues/16096
82+
// and re-enable this in the future
83+
// #ifdef PADDLE_WITH_CUDA
84+
// auto it1 = oper.Attrs().find("use_cudnn");
85+
// if (it1 != oper.Attrs().end() && platform::CanCUDNNBeUsed(ctx)) {
86+
// library = framework::LibraryType::kCUDNN;
87+
// }
88+
// #endif
8589
#ifdef PADDLE_WITH_MKLDNN
8690
auto it = oper.Attrs().find("use_mkldnn");
8791
if (library == framework::LibraryType::kPlain && it != oper.Attrs().end() &&

0 commit comments

Comments
 (0)