[Tests][TIRx] Localize hardware test gates#19985
Conversation
There was a problem hiding this comment.
Code Review
This pull request removes the suite-level hardware gate for TIRX tests, updates target handling in CUDA codegen tests, and adds @pytest.mark.gpu decorators to several test cases. Feedback focuses on correcting an inaccurate return type annotation in _get_source and removing the @pytest.mark.gpu decorators from compile-only tests. Since these compile-only tests do not execute on physical hardware, removing the decorators and specifying concrete target architectures (like sm_90a or sm_100a) inside the tests will allow them to run in CPU-only CI environments.
Important
The consumer version of Gemini Code Assist on GitHub is being sunset. Starting June 18, 2026, new organization installations will be blocked, and all code review activity will officially cease on July 17, 2026.
For more details on the timeline and next steps, please review the Help Documentation.
5329af1 to
afc143a
Compare
|
@tvm-bot rerun |
7dd8061 to
eee3e5e
Compare
eee3e5e to
409f503
Compare
41829d3 to
7942ab3
Compare
edcf5d6 to
74d2592
Compare
9f6cc6e to
a463a1b
Compare
a463a1b to
9ba68e1
Compare
This PR removes the suite-wide TIRx compute-capability gate and localizes hardware skips to CUDA codegen and tile-primitive tests. It keeps the original test parameterization unchanged, allowing parser, printer, IR, transform, and other non-hardware TIRx tests to run in regular CI while device-dependent cases are skipped when SM100 hardware is unavailable. CUDA codegen helpers use explicit target architectures where needed, and the run-only benchmark utility tests retain a local SM100 gate.
This intentionally avoids adding separate compile/run parameter cases. A follow-up PR can audit slow frontend execution tests and define a focused runtime regression budget for real TIRx kernels.
Local validation:
pytest -n auto -m "not gpu" tests/python/tirx: 486 passed, 79 skipped.pytest -n auto -m gpu tests/python/tirxwithout matching hardware: 1509 skipped.