Skip to content

Commit aa77a9b

Browse files
authored
Add missing loop (#151)
1 parent f0f6a04 commit aa77a9b

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

src/Search/LanguageModelLookahead.cc

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1797,6 +1797,18 @@ bool LanguageModelLookahead::computeScoresSparse(LanguageModelLookahead::Context
17971797
waitingLookaheadNodesByDepth_[nodes_[nodeIdx].depth].push_back(std::make_pair(nodeIdx, (Score)0));
17981798
}
17991799

1800+
for (const Lm::WordScore& successor : successors) {
1801+
Score score = successor.score() * scale;
1802+
1803+
TokenNodeScore::const_iterator node = nodeForToken_.begin() + firstNodeForToken_[successor.token()],
1804+
nodeEnd = nodeForToken_.begin() + firstNodeForToken_[successor.token() + 1];
1805+
for (; node != nodeEnd; ++node) {
1806+
Score endScore = score + (*node).second;
1807+
LookaheadId nodeId = (*node).first;
1808+
waitingLookaheadNodesByDepth_[nodes_[nodeId].depth].push_back(std::make_pair(nodeId, endScore));
1809+
}
1810+
}
1811+
18001812
if (nodeRecombination_.empty()) {
18011813
nodeRecombination_.resize(nEntries_, 0);
18021814
}

0 commit comments

Comments
 (0)