Unverified Commit c544ced2 authored by Phuong Nguyen's avatar Phuong Nguyen Committed by GitHub
Browse files

[JAX] Relax tolerance for the test_multiprocessing_encoder.py with NVFP4 by 0.001 (#2375)



relax tol
Signed-off-by: default avatarPhuong Nguyen <phuonguyen@nvidia.com>
parent e4bfa628
...@@ -672,7 +672,7 @@ class TestEncoder(unittest.TestCase): ...@@ -672,7 +672,7 @@ class TestEncoder(unittest.TestCase):
def test_te_nvfp4(self): def test_te_nvfp4(self):
"""Test Transformer Engine with NVFP4""" """Test Transformer Engine with NVFP4"""
result = self.exec(True, "NVFP4BlockScaling") result = self.exec(True, "NVFP4BlockScaling")
assert result[0] < 0.451 and result[1] > 0.788 assert result[0] < 0.451 and result[1] > 0.787
@unittest.skipIf(not is_bf16_supported(), "Device compute capability 8.0+ is required for BF16") @unittest.skipIf(not is_bf16_supported(), "Device compute capability 8.0+ is required for BF16")
def test_te_bf16_shardy(self): def test_te_bf16_shardy(self):
...@@ -710,7 +710,7 @@ class TestEncoder(unittest.TestCase): ...@@ -710,7 +710,7 @@ class TestEncoder(unittest.TestCase):
def test_te_nvfp4_shardy(self): def test_te_nvfp4_shardy(self):
"""Test Transformer Engine with NVFP4""" """Test Transformer Engine with NVFP4"""
result = self.exec(True, "NVFP4BlockScaling", enable_shardy=True) result = self.exec(True, "NVFP4BlockScaling", enable_shardy=True)
assert result[0] < 0.451 and result[1] > 0.788 assert result[0] < 0.451 and result[1] > 0.787
if __name__ == "__main__": if __name__ == "__main__":
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment