From dc64adb8e4e2ad64555fc7bbc82e867178ff6cff Mon Sep 17 00:00:00 2001 From: Nicolas Patry Date: Mon, 20 Nov 2023 14:17:07 +0100 Subject: [PATCH] Fixing cos_f16 test. --- candle-metal-kernels/src/lib.rs | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/candle-metal-kernels/src/lib.rs b/candle-metal-kernels/src/lib.rs index 6c2e5f2b..cff8e763 100644 --- a/candle-metal-kernels/src/lib.rs +++ b/candle-metal-kernels/src/lib.rs @@ -853,7 +853,7 @@ mod tests { #[test] fn cos_strided_random() { - let v: Vec<_> = (0..10_000).map(|i| rand::random::()).collect(); + let v: Vec<_> = (0..10_000).map(|_| rand::random::()).collect(); let shape = vec![5_000, 2]; let strides = vec![1, 5_000]; let offset = 0; @@ -1114,7 +1114,7 @@ mod tests { .collect(); let results = run(&v, unary::contiguous::cos::HALF); let expected: Vec = v.iter().map(|v| f16::from_f32(v.to_f32().cos())).collect(); - assert_eq!(approx_f16(results, 4), vec![0.54, -0.4165, -0.9902]); + assert_eq!(approx_f16(results, 4), vec![0.5405, -0.4163, -0.9902]); assert_eq!(approx_f16(expected, 4), vec![0.5405, -0.4163, -0.9902]); }