github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-29 21:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

[unittest] added doc for the pytest wrapper (#1704)

Browse Source
This commit is contained in:
Frank Lee
2022-10-14 10:56:17 +08:00
committed by GitHub
parent 451cd72dea
commit 91cd34e6e0
2 changed files with 25 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_auto_parallel/test_shape_consistency_pass.py
View File

@@ -18,6 +18,7 @@ from colossalai.device.device_mesh import DeviceMesh
from colossalai.fx.passes.experimental.adding_shape_consistency_pass_v2 import shape_consistency_pass, solution_annotatation_pass
from colossalai.auto_parallel.solver.solver import Solver_V2
from colossalai.auto_parallel.solver.options import SolverOptions
from colossalai.testing.pytest_wrapper import run_on_environment_flag
class ConvModel(nn.Module):
@@ -73,7 +74,7 @@ def check_apply(rank, world_size, port):
assert output.equal(origin_output)
@pytest.mark.skip("for higher testing speed")
@run_on_environment_flag(name='AUTO_PARALLEL')
@pytest.mark.dist
@rerun_if_address_is_in_use()
def test_apply():