small fix

7 years ago · b7801b9fcb
parent 70e7122785
commit b7801b9fcb
1 changed files with 2 additions and 2 deletions
--- a/paddle/fluid/operators/softmax_op.cc
+++ b/paddle/fluid/operators/softmax_op.cc
@ -44,7 +44,7 @@ class SoftmaxOp : public framework::OperatorWithKernel {
    framework::LibraryType library_{framework::LibraryType::kPlain};
 #ifdef PADDLE_WITH_CUDA
    if (platform::CanCUDNNBeUsed(ctx)) {
-      library = framework::LibraryType::kCUDNN;
+      library_ = framework::LibraryType::kCUDNN;
    }
 #endif
    std::string data_format = ctx.Attr<std::string>("data_format");
@ -118,7 +118,7 @@ class SoftmaxOpGrad : public framework::OperatorWithKernel {
    framework::LibraryType library_{framework::LibraryType::kPlain};
 #ifdef PADDLE_WITH_CUDA
    if (platform::CanCUDNNBeUsed(ctx)) {
-      library = framework::LibraryType::kCUDNN;
+      library_ = framework::LibraryType::kCUDNN;
    }
 #endif
    std::string data_format = ctx.Attr<std::string>("data_format");