Skip to content

Commit

Permalink
Fix hl_sequence_avg_forward.
Browse files Browse the repository at this point in the history
  • Loading branch information
hedaoyuan committed Jul 25, 2017
1 parent 0c2790f commit 41e1484
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion paddle/cuda/src/hl_cuda_sequence.cu
Original file line number Diff line number Diff line change
Expand Up @@ -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;
}
}

Expand Down

0 comments on commit 41e1484

Please sign in to comment.