Skip to content

Commit 4a4a538

Browse files
Add build pymomentum in gpu environment
1 parent becb423 commit 4a4a538

File tree

1 file changed

+24
-2
lines changed

1 file changed

+24
-2
lines changed

.github/workflows/ci_windows.yml

+24-2
Original file line numberDiff line numberDiff line change
@@ -45,17 +45,39 @@ jobs:
4545
--parallel
4646
4747
pymomentum:
48-
name: py-win
48+
name: py-${{ matrix.pixi_env }}-win
4949
runs-on: windows-latest
50+
strategy:
51+
fail-fast: false
52+
matrix:
53+
include:
54+
- pixi_env: "cpu"
55+
- pixi_env: "gpu"
56+
cuda_version: "12.8.0"
57+
env:
58+
FULL_CUDA_VERSION: ${{ matrix.cuda_version }}
5059
steps:
5160
- name: Checkout
5261
uses: actions/checkout@v4
5362

63+
- name: Install CUDA Toolkit
64+
if: ${{ contains(matrix.pixi_env, 'cuda') || contains(matrix.pixi_env, 'gpu') }}
65+
uses: Jimver/[email protected]
66+
id: cuda-toolkit
67+
with:
68+
# Available versions: https://github.com/Jimver/cuda-toolkit/blob/v0.2.21/src/links/linux-links.ts
69+
cuda: ${{ matrix.cuda_version }}
70+
71+
- name: Check CUDA Version
72+
if: ${{ contains(matrix.pixi_env, 'cuda') || contains(matrix.pixi_env, 'gpu') }}
73+
run: |
74+
nvcc --version
75+
5476
- name: Set up pixi
5577
uses: prefix-dev/[email protected]
5678
with:
5779
cache: true
5880

5981
- name: Build and test PyMomentum
6082
run: |
61-
pixi run test_py
83+
pixi run -e ${{ matrix.pixi_env }} test_py

0 commit comments

Comments
 (0)