use cached rank

8 years ago · d607f0b703
parent 958511160b
commit d607f0b703
1 changed files with 1 additions and 1 deletions
--- a/paddle/framework/ddim.cc
+++ b/paddle/framework/ddim.cc
@ -226,7 +226,7 @@ Eigen::DSizes<Eigen::DenseIndex, NDIMS> ToEigenDSizes(DDim dims) const {
  int rank = paddle::framework::arity(dims);
  PADDLE_ENFORCE(rank == NDIMS, "DDim and NDIMS must be same")
  Eigen::DSizes<Eigen::DenseIndex, NDIMS> dsizes;
-  for (int d = 0; d < paddle::framework::arity(dims); d++) {
+  for (int d = 0; d < rank; d++) {
    dsizes[d] = dims[d];
  }
  return dsizes;