Fix hl_sequence_avg_forward.

8 years ago · 41e1484eb2
parent 0c2790f7e5
commit 41e1484eb2
1 changed files with 1 additions and 1 deletions
--- a/paddle/cuda/src/hl_cuda_sequence.cu
+++ b/paddle/cuda/src/hl_cuda_sequence.cu
@ -330,7 +330,7 @@ __global__ void KeSequenceAvgForward(real* dst,
    }
    sum = mode == 1 ? sum :
        (mode == 0 ? sum / seqLength : sum * my_rsqrt((real)seqLength));
-    dst[gid] = sum;
+    dst[gid] += sum;
  }
 }