Merge pull request #3044 from hedaoyuan/fix_675

Fix hl_sequence_avg_forward.

Merge pull request #3044 from hedaoyuan/fix_675
Fix hl_sequence_avg_forward.
da4ce232 · hedaoyuan · GitHub · 0c2790f7 · 41e1484e · da4ce232
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/cuda/src/hl_cuda_sequence.cu paddle/cuda/src/hl_cuda_sequence.cu +1 -1

未找到文件。
--- 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;
  }
 }