Skip to content

Commit 141e01f

Browse files
authored
[CI] Adds pytest-durations for test timing (#782)
* [Ci] Adds pytest-durations for test timing Adds `pytest-durations` to the test requirements and configures pytest to display test durations. This helps in identifying slow-running tests and optimizing the test suite for faster feedback. * add amd ci durations * Removes flash_attn installation from CI
1 parent b66f9aa commit 141e01f

File tree

3 files changed

+4
-4
lines changed

3 files changed

+4
-4
lines changed

.github/workflows/amd_ci.yml

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,6 @@ jobs:
4141
python -m pip install --upgrade pip --no-user
4242
[[ -f requirements-test.txt ]] && \
4343
PIP_NO_BUILD_ISOLATION=1 pip install -r requirements-test.txt --no-user
44-
pip install flash_attn==2.5.8 --no-user --no-build-isolation
4544
touch "$MARKER"
4645
fi
4746
@@ -116,4 +115,4 @@ jobs:
116115
source "${{ runner.tool_cache }}/${{ env.VENV_DIR }}/bin/activate"
117116
cd testing/python/amd
118117
unset PYTHONPATH
119-
python -m pytest -v test_tilelang_test_amd.py
118+
python -m pytest -v test_tilelang_test_amd.py --durations=0

.github/workflows/ci.yml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -111,11 +111,11 @@ jobs:
111111
source "${{ runner.tool_cache }}/${{ env.VENV_DIR }}/bin/activate"
112112
cd examples
113113
unset PYTHONPATH
114-
python -m pytest -n 4 **/test*.py -v -r fE
114+
python -m pytest -n 4 **/test*.py -v -r fE --durations=0
115115
116116
- name: Run tests
117117
run: |
118118
source "${{ runner.tool_cache }}/${{ env.VENV_DIR }}/bin/activate"
119119
cd testing/python
120120
unset PYTHONPATH
121-
python -m pytest -n 4 -v -r fE
121+
python -m pytest -n 4 -v -r fE --durations=0

requirements-test.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ dtlib
1212
numpy>=1.23.5
1313
pytest>=6.2.4
1414
pytest_xdist>=2.2.1
15+
pytest-durations
1516
packaging>=21.0
1617
PyYAML
1718
tqdm>=4.62.3

0 commit comments

Comments
 (0)