From b7801b9fcbe0a2c3a1b8a92c1925def166a13e25 Mon Sep 17 00:00:00 2001 From: Kexin Zhao Date: Tue, 20 Mar 2018 17:33:21 -0700 Subject: [PATCH] small fix --- paddle/fluid/operators/softmax_op.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/operators/softmax_op.cc b/paddle/fluid/operators/softmax_op.cc index 3e5457bddc4..2506ffe48e2 100644 --- 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("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("data_format"); -- GitLab