PMIx based wire-up for PyTorch
Something went wrong. Try again.
123456789101112131415161718192021222324252627import pytest
def pytest_addoption(parser): parser.addoption( "--nprocs", action="store", default="10", help="Number of MPI processes to spawn for isolated tests", ) parser.addoption( "--nprocs-gpu", action="store", default="1", help="Number of MPI processes to spawn for isolated tests (GPU)", )
@pytest.fixturedef nprocs(request): return request.config.getoption("--nprocs")
@pytest.fixturedef nprocs_gpu(request): return request.config.getoption("--nprocs-gpu")