Skip to content

GitLab

Commit 7466180f authored by Sengxian's avatar Sengxian
Browse files

Skip test when there are no enough gpu

parent dad99892
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
tests/test_ddp.py
View file @ 7466180f
...@@ -11,6 +11,8 @@ from test_numerical import test_fmoe_linear as _test_fmoe_linear ...@@ -11,6 +11,8 @@ from test_numerical import test_fmoe_linear as _test_fmoe_linear
def _run_distributed(func, world_size, args: Dict): def _run_distributed(func, world_size, args: Dict):
if torch.cuda.device_count() < world_size:
pytest.skip("No enough GPU")
import subprocess import subprocess
import os import os
......
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