From 65713bc93140d63cfec3ffeec17d7f603fb74a71 Mon Sep 17 00:00:00 2001
From: Feiteng <lifeiteng0422@gmail.com>
Date: Wed, 27 Nov 2024 18:34:15 +0800
Subject: [PATCH] Fix PropagateBackward CHECK (#1312)

---
 k2/csrc/intersect_dense_pruned.cu | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/k2/csrc/intersect_dense_pruned.cu b/k2/csrc/intersect_dense_pruned.cu
index 7c1fe1d30..c5befcc1c 100644
--- a/k2/csrc/intersect_dense_pruned.cu
+++ b/k2/csrc/intersect_dense_pruned.cu
@@ -1301,7 +1301,7 @@ class MultiGraphDenseIntersectPruned {
             cur_states_data[arcs_rowids2[arcs_idx012]].forward_loglike);
 
         // should be <= 0.0, mathematically.
-        K2_CHECK_LT(backward_loglike, -src_state_forward_loglike + 2.0);
+        K2_CHECK_LE(backward_loglike, -src_state_forward_loglike + 2.0);
         if (backward_loglike + src_state_forward_loglike >= -output_beam) {
           keep_this_arc = 1;
         } else {