Fix goldens in //third_party/py/aqt/jax/v2/numerics:fp8_numerics_test

PiperOrigin-RevId: 696616461
google · Nov 14, 2024 · eeb1c7e · eeb1c7e
1 parent 6659f1e
commit eeb1c7e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aqt/jax/v2/numerics/fp8_numerics_test.py b/aqt/jax/v2/numerics/fp8_numerics_test.py
@@ -146,7 +146,7 @@ def test_mnist_training(self, fwd_bits: str):
             "TPU v2": [3.2267177104949951171875],
             "TPU v3": [3.2267177104949951171875],
             "TPU v4": [3.2266914844512939453125],
-            "TPU v5 lite": [3.2268390655517578125],
+            "TPU v5 lite": [3.1812143325805664062500],
         },
     }