test: Fix parallel test in simd mode

pymc-devs · Nov 28, 2024 · ba80449 · ba80449
1 parent 298121d
commit ba80449
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/sampler.rs b/src/sampler.rs
@@ -979,7 +979,7 @@ pub mod test_logps {
                 for (p, g) in pos.chunks_exact(4).zip(grad.chunks_exact_mut(4)) {
                     let p = f64x4::from_slice(p);
                     let val = mu_splat - p;
-                    logp = val * val * f64x4::splat(0.5);
+                    logp -= val * val * f64x4::splat(0.5);
                     g.copy_from_slice(&val.to_array());
                 }