From 5847b96a61fec031555f245702f604e928f8ad4e Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Mon, 17 Jul 2017 18:30:40 +0800
Subject: [PATCH] Follow comments, correct implement `DDim::size()`

* Also fix unit test
---
 paddle/framework/ddim.cc       | 2 +-
 paddle/operators/softmax_op.cu | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/paddle/framework/ddim.cc b/paddle/framework/ddim.cc
index f3dd396613..eb1a18ee40 100644
--- a/paddle/framework/ddim.cc
+++ b/paddle/framework/ddim.cc
@@ -278,7 +278,7 @@ std::ostream& operator<<(std::ostream& os, const DDim& ddim) {
   return os;
 }
 
-ssize_t DDim::size() const { return product(*this); }
+ssize_t DDim::size() const { return arity(*this); }
 
 DDim::DDim(std::initializer_list<int> init_list) {
   *this = make_ddim(init_list);
diff --git a/paddle/operators/softmax_op.cu b/paddle/operators/softmax_op.cu
index 59f32b35cf..903eef1b62 100644
--- a/paddle/operators/softmax_op.cu
+++ b/paddle/operators/softmax_op.cu
@@ -1,5 +1,5 @@
 #include <paddle/framework/op_registry.h>
-#include <paddle/operators/sigmoid_op.h>
+#include <paddle/operators/softmax_op.h>
 
 REGISTER_OP_GPU_KERNEL(
     softmax, paddle::operators::SoftmaxKernel<paddle::platform::GPUPlace>);
-- 
GitLab