From 507d2a3ce76db516aefd4a201159d857f58c69c6 Mon Sep 17 00:00:00 2001 From: Thomas Viehmann Date: Wed, 22 Jan 2025 22:17:58 +0100 Subject: [PATCH] more flakiness (#1683) --- thunder/tests/opinfos.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/thunder/tests/opinfos.py b/thunder/tests/opinfos.py index 6c41bd6234..d3a7d29860 100644 --- a/thunder/tests/opinfos.py +++ b/thunder/tests/opinfos.py @@ -7864,12 +7864,13 @@ def rms_norm_error_generator(op, device, **kwargs): ), # See issue - https://github.com/Lightning-AI/lightning-thunder/issues/1395 DecorateInfo( - custom_comparator(partial(assert_close, atol=2e-3, rtol=2e-3)), + custom_comparator(partial(assert_close, atol=1e-2, rtol=1e-2)), dtypes=(datatypes.float16,), devicetypes=(devices.DeviceType.CUDA,), ), DecorateInfo( pytest.mark.skip(reason="Flaky. See https://github.com/Lightning-AI/lightning-thunder/issues/1678"), + "test_core_vs_torch_consistency", dtypes=(datatypes.bfloat16,), ), ),