Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions .github/scripts/install-torch-tensorrt.sh
Original file line number Diff line number Diff line change
Expand Up @@ -43,6 +43,8 @@ PY
python -m pip uninstall -y torch torchvision
python -m pip install --force-reinstall --pre ${TORCHVISION} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
python -m pip install --force-reinstall --pre ${TORCH} --index-url ${INDEX_URL} --extra-index-url https://pypi.org/simple
# dynamo-torchao full/nightly suite
python -m pip install torchao

# If CUDA 13 (cu13), prepend venv's NVIDIA CUDA 13 libs to LD_LIBRARY_PATH
if [[ "${CU_VERSION}" == cu13* ]]; then
Expand Down
10 changes: 9 additions & 1 deletion tests/ci/suites.py
Original file line number Diff line number Diff line change
Expand Up @@ -237,9 +237,17 @@ def for_variant(self, variant: Variant) -> dict[str, Any]:
tier="l2",
lanes=("full", "nightly"),
paths=("models/",),
markers="not critical",
# TorchAO compile tests live in dynamo-torchao.
markers="not critical and not torchao",
jobs=_MODEL,
),
Suite(
"dynamo-torchao",
tier="l2",
lanes=("full", "nightly"),
paths=("models/test_torchao*.py",),
jobs=_HEAVY,
),
Suite(
"dynamo-llm",
tier="l2",
Expand Down
16 changes: 16 additions & 0 deletions tests/py/dynamo/models/conftest.py
Original file line number Diff line number Diff line change
Expand Up @@ -22,3 +22,19 @@ def pytest_addoption(parser):
def ir(request):
ir_opt = request.config.getoption("--ir")
return ir_opt[0] if ir_opt else "dynamo"


def pytest_configure(config):
config.addinivalue_line(
"markers",
"torchao: TorchAO quantization compile tests; collected by the dynamo-torchao full-lane suite",
)


def pytest_collection_modifyitems(config, items):
"""Mark TorchAO model tests so the dynamo-torchao suite can own them."""
marker = pytest.mark.torchao
for item in items:
path = str(getattr(item, "path", item.fspath))
if "test_torchao" in path:
item.add_marker(marker)
Loading