From 19975e969d7b65c74b978a7afdce5a5bd6bc9bfa Mon Sep 17 00:00:00 2001
From: Guillaume Lagrange <lagrange.guillaume.1@gmail.com>
Date: Mon, 9 Dec 2024 09:00:07 -0500
Subject: [PATCH] Fix cast float test precision (#2586)

---
 crates/burn-tensor/src/tests/ops/cast.rs | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/crates/burn-tensor/src/tests/ops/cast.rs b/crates/burn-tensor/src/tests/ops/cast.rs
index bc45518c38..3f760d6766 100644
--- a/crates/burn-tensor/src/tests/ops/cast.rs
+++ b/crates/burn-tensor/src/tests/ops/cast.rs
@@ -46,6 +46,7 @@ mod tests {
         let output = tensor.cast(DType::F32);
 
         assert_eq!(output.dtype(), DType::F32);
-        output.into_data().assert_approx_eq(&data, 5);
+        // Use precision 2 for parametrized tests in f16 and bf16
+        output.into_data().assert_approx_eq(&data, 2);
     }
 }