diff --git a/tests/test_booster/test_mixed_precision/test_fp16_torch.py b/tests/test_booster/test_mixed_precision/test_fp16_torch.py index f6d6e8303..d98171a3d 100644 --- a/tests/test_booster/test_mixed_precision/test_fp16_torch.py +++ b/tests/test_booster/test_mixed_precision/test_fp16_torch.py @@ -1,5 +1,6 @@ import torch from torch.optim import Adam +import pytest import colossalai from colossalai.booster.mixed_precision import FP16TorchMixedPrecision @@ -35,6 +36,7 @@ def run_torch_amp(rank, world_size, port): del model, optimizer, criterion, data, output, mixed_precision +@pytest.mark.skip("test ci.") @rerun_if_address_is_in_use() def test_torch_ddp_plugin(): spawn(run_torch_amp, 1)