TensorRT-LLMs/tests/unittest/pytest.ini
Omer Ullman Argov a32f7083b4
[ci] parallelize torch unittests (#5714)
Signed-off-by: Omer Ullman Argov <118735753+omera-nv@users.noreply.github.com>
2025-07-09 11:05:57 +03:00

24 lines
654 B
INI

[pytest]
asyncio_default_fixture_loop_scope = module
threadleak = True
threadleak_exclude = asyncio_\d+
addopts = --durations=0 -W ignore::DeprecationWarning
pythonpath =
_torch/auto_deploy/_utils_test
../../examples/auto_deploy
../../examples/models/core
../../examples
env =
D:AUTO_DEPLOY_LOG_LEVEL=INFO
markers =
part0
part1
part2
part3
part4
gpu2: this test uses 2 GPUs
gpu4: this test uses 4 GPUs
post_merge: this test should only run in post merge
high_cuda_memory: this test uses a lot of CUDA memory (typically more than 12GB)
no_xdist: this test should not run when using pytest-xdist