[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

Author: Cunxiao2002

.github/workflows/amd_ci.yml

@@ -41,7 +41,6 @@ jobs:
           python -m pip install --upgrade pip --no-user
           [[ -f requirements-test.txt ]] && \
             PIP_NO_BUILD_ISOLATION=1 pip install -r requirements-test.txt --no-user
-          pip install flash_attn==2.5.8 --no-user --no-build-isolation
           touch "$MARKER"
         fi
 
@@ -116,4 +115,4 @@ jobs:
         source "${{ runner.tool_cache }}/${{ env.VENV_DIR }}/bin/activate"
         cd testing/python/amd
         unset PYTHONPATH
-        python -m pytest -v test_tilelang_test_amd.py
+        python -m pytest -v test_tilelang_test_amd.py --durations=0

.github/workflows/ci.yml

@@ -111,11 +111,11 @@ jobs:
         source "${{ runner.tool_cache }}/${{ env.VENV_DIR }}/bin/activate"
         cd examples
         unset PYTHONPATH
-        python -m pytest -n 4 **/test*.py -v -r fE
+        python -m pytest -n 4 **/test*.py -v -r fE --durations=0
 
     - name: Run tests
       run: |
         source "${{ runner.tool_cache }}/${{ env.VENV_DIR }}/bin/activate"
         cd testing/python
         unset PYTHONPATH
-        python -m pytest -n 4 -v -r fE
+        python -m pytest -n 4 -v -r fE --durations=0

requirements-test.txt

@@ -12,6 +12,7 @@ dtlib
 numpy>=1.23.5
 pytest>=6.2.4
 pytest_xdist>=2.2.1
+pytest-durations
 packaging>=21.0
 PyYAML
 tqdm>=4.62.3