arnaucube/poulpy
1
0
Fork 0
mirror of https://github.com/arnaucube/poulpy.git synced 2026-02-10 05:06:44 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix multithread bdd

Browse Source
This commit is contained in:
Pro7ech
2025-11-12 20:25:19 +01:00
parent 10d1780aaa
commit b034ea8253
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
poulpy-schemes/src/tfhe/bdd_arithmetic/eval.rs
View File

@@ -145,14 +145,15 @@ where
let chunk_size: usize = circuit.output_size().div_ceil(threads); let chunk_size: usize = circuit.output_size().div_ceil(threads);
thread::scope(|scope| { thread::scope(|scope| {
for (scratch_thread, out_chunk) in scratches for (thread_idx, (scratch_thread, out_chunk)) in scratches
.iter_mut() .iter_mut()
.zip(out[..circuit.output_size()].chunks_mut(chunk_size)) .zip(out[..circuit.output_size()].chunks_mut(chunk_size))
.enumerate()
{ {
// Capture chunk + thread scratch by move // Capture chunk + thread scratch by move
scope.spawn(move || { scope.spawn(move || {
for (idx, out_i) in out_chunk.iter_mut().enumerate() { for (idx, out_i) in out_chunk.iter_mut().enumerate() {
let (nodes, state_size) = circuit.get_circuit(idx); let (nodes, state_size) = circuit.get_circuit(thread_idx * chunk_size + idx);
if state_size == 0 { if state_size == 0 {
out_i.data_mut().zero(); out_i.data_mut().zero();