fix(gpu): fix wrong assert in division

zama-ai · Feb 21, 2025 · 8fa310d · 8fa310d
1 parent 6b21bff
commit 8fa310d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/backends/tfhe-cuda-backend/cuda/src/integer/div_rem.cuh b/backends/tfhe-cuda-backend/cuda/src/integer/div_rem.cuh
@@ -575,8 +575,8 @@ __host__ void host_unsigned_integer_div_rem_kb(
       cuda_synchronize_stream(mem_ptr->sub_streams_3[j], gpu_indexes[j]);
     }
 
-    assert(first_trivial_block - 1 == cleaned_merged_interesting_remainder.len);
-    assert(first_trivial_block - 1 == new_remainder.len);
+    assert(first_trivial_block == cleaned_merged_interesting_remainder.len);
+    assert(first_trivial_block == new_remainder.len);
 
     remainder1.copy_from(cleaned_merged_interesting_remainder, 0,
                          first_trivial_block - 1, streams[0], gpu_indexes[0]);