Skip to content

Commit 7697d6e

Browse files
committed
ci: switch to uv in cupy GA
1 parent 4e3e04c commit 7697d6e

2 files changed

Lines changed: 20 additions & 61 deletions

File tree

.github/workflows/buildcupy.yaml

Lines changed: 20 additions & 23 deletions
Original file line numberDiff line numberDiff line change
@@ -13,29 +13,26 @@ jobs:
1313
runs-on: self-hosted
1414
steps:
1515
- name: Check out source repository
16-
uses: actions/checkout@v4
17-
- name: Set up Python environment and Install dependencies and pylops
16+
uses: actions/checkout@v6
17+
with:
18+
fetch-depth: 0
19+
- name: Install uv with Python 3.11
20+
uses: astral-sh/setup-uv@v6
21+
with:
22+
python-version: 3.11
23+
- name: Install dependencies and pylops
1824
run: |
19-
python3 -m venv .venv-pylops
20-
# write install scripts
21-
cat << 'EOF' > pylops_tests.sh
22-
#!/bin/bash
23-
PYTHON=.venv-pylops/bin/python3
24-
PIP=.venv-pylops/bin/pip
25-
ls $PYTHON
26-
ls $PIP
27-
$PIP install
28-
$PIP install --upgrade pip setuptools
29-
$PIP install install flake8 pytest setuptools-scm
30-
$PIP install -r requirements-dev-gpu.txt
31-
$PIP install cupy-cuda12x
32-
$PYTHON -m setuptools_scm
33-
$PIP install .
34-
EOF
35-
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 bash pylops_tests.sh
36-
- name: Tests with pytest
25+
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 bash -c '
26+
uv sync --locked --extra advanced \
27+
--extra stat --extra gpu-cu12 \
28+
--extra deep-cu128 --all-groups
29+
'
30+
echo "done!"
31+
- name: Test with pytest
3732
run: |
38-
export CUPY_PYLOPS=1; export TEST_CUPY_PYLOPS=1;
39-
export PYTEST=.venv-pylops/bin/pytest
40-
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 $PYTEST
33+
srun --account=yuxilab -n 1 -N 1 --gres=gpu:L40S:1 bash -c '
34+
export CUPY_PYLOPS=1
35+
export TEST_CUPY_PYLOPS=1
36+
uv run pytest --color=yes pytests/
37+
'
4138
echo "done!"

.github/workflows/buildcupy1.yaml

Lines changed: 0 additions & 38 deletions
This file was deleted.

0 commit comments

Comments
 (0)