diff --git a/.github/workflows/ci.yml b/.github/workflows/ci.yml index d4c02d03..78ed48f5 100644 --- a/.github/workflows/ci.yml +++ b/.github/workflows/ci.yml @@ -91,7 +91,7 @@ jobs: - tags: tilelang-metax-runner name: self-hosted-metax # Format: MACA-.[.]. E.g., "MACA-3.0". - toolkit: MACA-3.6 + toolkit: MACA-3.7 - tags: [macos-latest] name: macos-latest toolkit: Metal # or Nightly-Metal @@ -279,7 +279,7 @@ jobs: if [[ "${UV_INDEX}" == *"/nightly/"* ]]; then uv pip install --prerelease=allow -v torch fi - uv pip install -v -r requirements-test.txt + uv pip install -r requirements-test.txt echo "import torch; print(f'torch: {torch.__version__}')" | uv run --no-project --script - if [[ "${{ matrix.runner.toolkit }}" == *"CUDA"* ]]; then uv pip install --no-build-isolation-package=flash-attn -v -r requirements-test-cuda.txt @@ -287,8 +287,8 @@ jobs: elif [[ "${{ matrix.runner.toolkit }}" == *"ROCm"* ]]; then uv pip install -v -r requirements-test-rocm.txt elif [[ "${{ matrix.runner.toolkit }}" == *"MACA"* ]]; then - uv pip install -v -r requirements-test-maca.txt - uv pip install -v --no-deps --python-version 3.10.0 flash_linear_attention==0.4.0+metax3.5.3.9torch2.8 -i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com + uv pip install -r requirements-test-maca.txt + uv pip install --no-deps --python-version 3.10.0 flash_linear_attention==0.4.0 -i https://repos.metax-tech.com/r/maca-pypi/simple --trusted-host repos.metax-tech.com elif [[ "${{ matrix.runner.toolkit }}" == *"Metal"* ]]; then uv pip install -v -r requirements-test-metal.txt else @@ -357,7 +357,7 @@ jobs: uv run --no-project -m -- pytest --verbose --color=yes --durations=0 --showlocals --cache-clear ) - "${PYTEST[@]}" --maxfail=3 --numprocesses=4 \ + "${PYTEST[@]}" --maxfail=3 \ ../examples # NVIDIA CUDA tests