Skip to content

GitLab

Unverified Commit e71d2452 authored by Hongxin Liu's avatar Hongxin Liu Committed by GitHub
Browse files

[test] ignore gpt2 shardformer test (#4619)

parent ec086680
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_shardformer/test_model/test_shard_gpt2.py
View file @ e71d2452
...@@ -102,6 +102,7 @@ def check_forward_backward(model_fn, data_gen_fn, output_transform_fn, loss_fn, ...@@ -102,6 +102,7 @@ def check_forward_backward(model_fn, data_gen_fn, output_transform_fn, loss_fn,
torch.cuda.empty_cache() torch.cuda.empty_cache()
@pytest.mark.skip(reason="This test will hang in CI")
@parameterize('test_config', [{ @parameterize('test_config', [{
'tp_size': 2, 'tp_size': 2,
'pp_size': 2, 'pp_size': 2,
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment