Merge pull request #11579 from jacquesqiao/fix-concat-grad-kernel

fix concat grad kernel
7 years ago · 4116b55163
parent d020d7fd29 762160bd8c
commit 4116b55163
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/math/concat.cu
+++ b/paddle/fluid/operators/math/concat.cu
@ -209,7 +209,7 @@ class ConcatGradFunctor<platform::CUDADeviceContext, T> {

    outputs_cols[0] = 0;
    for (int i = 0; i < o_num; ++i) {
-      int t_col = outputs->at(i)->numel() / out_row;
+      int t_col = ref_inputs.at(i)->numel() / out_row;
      if (sameShape) {
        if (t_col != out0_col) sameShape = false;
      }