diff --git a/paddle/framework/ddim.cc b/paddle/framework/ddim.cc index f3dd396613b0f0726d0613da1bf9f91be29aa495..eb1a18ee4075b07d6f8ae2a90a451f6d458c8bf4 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 init_list) { *this = make_ddim(init_list); diff --git a/paddle/operators/softmax_op.cu b/paddle/operators/softmax_op.cu index 59f32b35cfd4708965176dc2feb51f397519038c..903eef1b62231d65e2f9ec7a1f57fca0f4c4605c 100644 --- a/paddle/operators/softmax_op.cu +++ b/paddle/operators/softmax_op.cu @@ -1,5 +1,5 @@ #include -#include +#include REGISTER_OP_GPU_KERNEL( softmax, paddle::operators::SoftmaxKernel);