test: Fix parallel test in simd mode

aseyboldt · aseyboldt · commit ba8044921059 · 2024-11-28T16:22:57.000+01:00
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());
                 }
 

Original file line number	Diff line number	Diff line change
`@@ -979,7 +979,7 @@ pub mod test_logps {`
`979`	`979`	`for (p, g) in pos.chunks_exact(4).zip(grad.chunks_exact_mut(4)) {`
`980`	`980`	`let p = f64x4::from_slice(p);`
`981`	`981`	`let val = mu_splat - p;`
`982`		`- logp = val * val * f64x4::splat(0.5);`
	`982`	`+ logp -= val * val * f64x4::splat(0.5);`
`983`	`983`	`g.copy_from_slice(&val.to_array());`
`984`	`984`	`}`
`985`	`985`