Skip to content

Commit

Permalink
more flakiness (#1683)
Browse files Browse the repository at this point in the history
  • Loading branch information
t-vi authored and riccardofelluga committed Jan 27, 2025
1 parent e54469b commit 507d2a3
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion thunder/tests/opinfos.py
Original file line number Diff line number Diff line change
Expand Up @@ -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,),
),
),
Expand Down

0 comments on commit 507d2a3

Please sign in to comment.