Add nullptr check in for statement of function RecodeBeamSearch::Cont…

…inueContext Signed-off-by: Stefan Weil <sw@weilnetz.de>
Shibrain · May 23, 2024 · 5712e16 · 5712e16
1 parent cf7231f
commit 5712e16
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/lstm/recodebeam.cpp b/src/lstm/recodebeam.cpp
@@ -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) {