From ca914147eb419b485fbf6b80e181f46a720c3064 Mon Sep 17 00:00:00 2001 From: flybird11111 <1829166702@qq.com> Date: Wed, 9 Apr 2025 14:01:47 +0800 Subject: [PATCH] Update test_fp16_torch.py --- tests/test_booster/test_mixed_precision/test_fp16_torch.py | 2 -- 1 file changed, 2 deletions(-) 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 bb76b354d..f6d6e8303 100644 --- a/tests/test_booster/test_mixed_precision/test_fp16_torch.py +++ b/tests/test_booster/test_mixed_precision/test_fp16_torch.py @@ -1,4 +1,3 @@ -import pytest import torch from torch.optim import Adam @@ -36,7 +35,6 @@ 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)