Update test_fp16_torch.py

This commit is contained in:
flybird11111
2025-04-09 14:01:47 +08:00
committed by GitHub
parent 3491a9f7e3
commit ca914147eb

View File

@@ -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)