diff options
| author | Hieu Hoang <hieuhoang@gmail.com> | 2018-01-22 16:44:45 +0300 |
|---|---|---|
| committer | Hieu Hoang <hieuhoang@gmail.com> | 2018-01-22 16:44:45 +0300 |
| commit | a27f16972b18b9f06dac9b501160192e4f319e32 (patch) | |
| tree | 63833a2c25db61393848d97f3e2b0e3373ba1853 | |
| parent | af6427767218a77658841f56365ab1c958c0547e (diff) | |
correct active sentences vector size
| -rw-r--r-- | src/amun/gpu/decoder/encoder_decoder.cu | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/amun/gpu/decoder/encoder_decoder.cu b/src/amun/gpu/decoder/encoder_decoder.cu index ef18a868..1375b374 100644 --- a/src/amun/gpu/decoder/encoder_decoder.cu +++ b/src/amun/gpu/decoder/encoder_decoder.cu @@ -188,7 +188,7 @@ void EncoderDecoder::DecodeAsyncInternal() InitBatch(histories, sentenceLengths, sourceContext, SCU, *state); //cerr << "prevHyps2=" << prevHyps.size() << endl; - std::vector<unsigned> activeCount(histories.size(), 0); + std::vector<unsigned> activeCount(histories.size() + 1, 0); unsigned step = 0; while (histories.NumActive()) { |