From 28cf1e2c57188b116b467ef14beb7225192e8188 Mon Sep 17 00:00:00 2001 From: flybird11111 <1829166702@qq.com> Date: Wed, 9 Apr 2025 15:20:14 +0800 Subject: [PATCH] fix --- tests/test_fp8/test_fp8_allgather.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/test_fp8/test_fp8_allgather.py b/tests/test_fp8/test_fp8_allgather.py index 91e66e83c..df54c252f 100644 --- a/tests/test_fp8/test_fp8_allgather.py +++ b/tests/test_fp8/test_fp8_allgather.py @@ -1,4 +1,5 @@ import torch +import pytest import torch.distributed as dist from torch.distributed.distributed_c10d import _get_default_group from torch.testing import assert_close @@ -36,6 +37,7 @@ def run_dist(rank, world_size, port): check_4gpu() +@pytest.mark.skip("tested in corresponding sharderformer") @rerun_if_address_is_in_use() def test_all_gather(): spawn(run_dist, 4)