Unverified Commit 8ce27782 authored by Xuehai Pan's avatar Xuehai Pan Committed by GitHub
Browse files

[CI][Refactor] Merge test CI workflow files into one (#973)



* refactor: merge test CI workflow files into one

* chore: set `UV_INDEX_STRATEGY=unsafe-best-match`

* feat: add AST test with Python 3.8

* feat: implement manual caching mechanism for self-hosted runners

* refactor: simplify cache logic for self-hosted runners

* chore: clear uv cache on failure

* chore: print format.sh output to logs

* chore: improve uv caching

* chore: disable parallel test

* chore: use `PYTHONDEVMODE=1` in CI

* feat: enable coredump generation

* fix: fix perfbench condition

* Revert "feat: enable coredump generation"

This reverts commit c52da65cb572932e09905d08c43a39ec3cf47c54.

* chore: move example CI down

* Revert "chore: move example CI down"

This reverts commit 9d8e65055e01d955c5268a9a6705d270c2de0d57.

* chore: skip example `test_example_mha_sink_bwd_bhsd`

* chore: skip example `test_example_gqa_sink_bwd_bhsd`

* fix: fix example argument passing

* fix: loosen test criteria

* chore: rename `CMAKE_CONFIGURE_OPTIONS` -> `CLANG_TIDY_CMAKE_OPTIONS` for clarity

* feat: enable parallel testings

* chore: update pytest options

* remove skipped test as now been resolved

* chore: empty commit to re-trigger ci

* test for n 1

* chore: remove ` --numprocesses=1` option in example

* chore: disable failfast

* chore: update cibw selection

* fix: fix git submodule clone

* chore: update cibw commands

* fix: fix yapf multiprocessing

* chore: setup ccache for CIBW on macOS only

* chore: update comments

* chore: update artifact listing

* fix: do not fail if not found nvcc in PATH

* fix: fix flash-attn installation

* chore: update dist workflow trigger

* chore: remove outdated comments

* chore(workflows/dist): simplify build matrix strategy

* fix: fix CUDA path finding

* fix: fix CUDA path finding

* chore: imcrease CI timeout

* ci: disable failfast

* fix: hide path prefix

* chore: more verbose

* chore: disable PR trigger for dist workflow

* fix: seed for tests

* fix: use nightly torch for ROCm tests

* chore: enable PR trigger for dist workflow

* chore: stop uploading debug wheels as artifacts in PR

* chore: do not run workflows in forks

* chore: housekeep requirements

* chore: use Nightly-ROCm-6.3 for CI

* chore: use Nightly-ROCm-6.4 for CI

* Update ROCm toolkit version to 7.0

* chore: restore previous rocm-ci.yml for test

* fix: cleanup PYTHONPATH

* chore: remove previous rocm-ci.yml

* ci fix

* chore: remove previous rocm-ci.yml

* chore: enable parallel example run

---------
Co-authored-by: default avatarLeiWang1999 <leiwang1999@outlook.com>
Co-authored-by: default avataralex_xiao <xinyuxiao2024@gmail.com>
parent 80665cd1
# Lint requirements
--requirement requirements-lint.txt
# Common test requirements
--requirement requirements-test.txt
# ROCm specific requirements
# Currently: none
# lint requirements
-r requirements-lint.txt
# build requirements
Cython
# Lint requirements
--requirement requirements-lint.txt
# Build requirements
cython
cmake>=3.26
# runtime requirements
setuptools
scikit-build-core
ninja
# Runtime requirements
--requirement requirements.txt
# Test requirements
cffi
cpplint
Cython
cython
docutils
dtlib
numpy>=1.23.5
pytest>=6.2.4
pytest_xdist>=2.2.1
einops
packaging>=21.0
pytest-xdist>=2.2.1
pytest-durations
pytest-timeout
packaging>=21.0
PyYAML
tqdm>=4.62.3
typing_extensions>=4.10.0
pytest>=6.2.4
pyyaml
requests
cloudpickle
ml_dtypes
psutil
torch
tabulate
wheel
setuptools
einops
scipy
tabulate
tornado
wheel
# runtime requirements
numpy>=1.23.5
tqdm>=4.62.3
typing_extensions>=4.10.0
# Runtime requirements
cloudpickle
ml_dtypes
ml-dtypes
numpy>=1.23.5
psutil
torch
torch>=2.7; platform_system == 'Darwin'
tqdm>=4.62.3
typing-extensions>=4.10.0
import os
import random
os.environ["PYTHONHASHSEED"] = "0"
random.seed(0)
try:
import torch
except ImportError:
pass
else:
torch.manual_seed(0)
try:
import numpy as np
except ImportError:
pass
else:
np.random.seed(0)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment