github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-09 13:00:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

[fx] recovered skipped pipeline tests (#1338)

Browse Source
This commit is contained in:
Frank Lee
2022-07-19 09:49:50 +08:00
committed by GitHub
parent 0c51ff2c13
commit f3ce7b8336
2 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_fx/test_comm_size_compute.py
View File

@@ -30,7 +30,6 @@ class MLP(torch.nn.Module):
return x
@pytest.mark.skip('skip due to CI environment')
def test_comm_size_compute():
model = MLP(MODEL_DIM)
input_sample = torch.rand(BATCH_SIZE, MODEL_DIM)

1
tests/test_fx/test_pipeline_passes.py
View File

@@ -39,7 +39,6 @@ def pipeline_pass_test_helper(model, data, pass_func):
assert output.equal(origin_output)
@pytest.mark.skip('skip due to CI environment')
def test_pipeline_passes():
model = MLP(MODEL_DIM)
data = torch.rand(BATCH_SIZE, MODEL_DIM)