github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-15 22:53:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
flybird11111 2025-04-01 15:32:11 +08:00
parent 822556a8ca
commit 4b8b67ae23
1 changed files with 2 additions and 0 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,5 +1,6 @@
import torch import torch
from torch.optim import Adam from torch.optim import Adam
import pytest
import colossalai import colossalai
from colossalai.booster.mixed_precision import FP16TorchMixedPrecision 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 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)