Skip to content

Commit

Permalink
Add nullptr check in for statement of function RecodeBeamSearch::Cont…
Browse files Browse the repository at this point in the history
…inueContext

Signed-off-by: Stefan Weil <[email protected]>
  • Loading branch information
stweil committed May 23, 2024
1 parent cf7231f commit 5712e16
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion src/lstm/recodebeam.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -898,12 +898,13 @@ void RecodeBeamSearch::ContinueContext(
int length = LengthFromBeamsIndex(index);
bool use_dawgs = IsDawgFromBeamsIndex(index);
NodeContinuation prev_cont = ContinuationFromBeamsIndex(index);
for (int p = length - 1; p >= 0; --p, previous = previous->prev) {
for (int p = length - 1; p >= 0 && previous != nullptr; --p) {
while (previous->duplicate || previous->code == null_char_) {
previous = previous->prev;
}
prefix.Set(p, previous->code);
full_code.Set(p, previous->code);
previous = previous->prev;
}
if (prev != nullptr && !is_simple_text_) {
if (top_n_flags_[prev->code] == top_n_flag) {
Expand Down

0 comments on commit 5712e16

Please sign in to comment.