Fix LogAdd (#316)

Using 0 as the initial value, should not perform addition when both values are 0
k2-fsa · Sep 18, 2023 · 5ca0ff8 · 5ca0ff8
1 parent c471423
commit 5ca0ff8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sherpa-onnx/csrc/hypothesis.cc b/sherpa-onnx/csrc/hypothesis.cc
@@ -18,8 +18,10 @@ void Hypotheses::Add(Hypothesis hyp) {
   } else {
     it->second.log_prob = LogAdd<double>()(it->second.log_prob, hyp.log_prob);
 
-    it->second.lm_log_prob =
+    if (it->second.lm_log_prob != 0 && hyp.lm_log_prob != 0){
+      it->second.lm_log_prob =
         LogAdd<double>()(it->second.lm_log_prob, hyp.lm_log_prob);
+    }
   }
 }