github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-27 15:57:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update test_fp16_torch.py

Browse Source
This commit is contained in:
flybird11111 2025-04-09 14:01:47 +08:00 committed by GitHub
parent 3491a9f7e3
commit ca914147eb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_booster/test_mixed_precision/test_fp16_torch.py
View File

@ -1,4 +1,3 @@
import pytest
import torch import torch
from torch.optim import Adam 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 del model, optimizer, criterion, data, output, mixed_precision
@pytest.mark.skip("test ci.")
@rerun_if_address_is_in_use() @rerun_if_address_is_in_use()
def test_torch_ddp_plugin(): def test_torch_ddp_plugin():
spawn(run_torch_amp, 1) spawn(run_torch_amp, 1)