Skip to content

Commit 3e28c59

Browse files
committed
feat: add nightly ci gpu hpc
1 parent 420e70f commit 3e28c59

File tree

3 files changed

+31
-0
lines changed

3 files changed

+31
-0
lines changed

.github/ci-hpc-gpu-config.yml

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,5 @@
1+
build:
2+
python_dependencies:
3+
- ecmwf/earthkit-utils@refactor/backend_for_testing_only
4+
queue: ng
5+
gpus: 1
Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,22 @@
1+
name: nightly-hpc-gpu
2+
3+
on:
4+
workflow_dispatch:
5+
6+
# Run at 03:20 UTC every day (on default branch)
7+
schedule:
8+
- cron: "20 03 * * *"
9+
10+
jobs:
11+
test-hpc-gpu:
12+
runs-on: [self-hosted, linux, hpc]
13+
steps:
14+
- uses: ecmwf/reusable-workflows/ci-hpc@v2
15+
with:
16+
github_user: ${{ secrets.BUILD_PACKAGE_HPC_GITHUB_USER }}
17+
github_token: ${{ secrets.GH_REPO_READ_TOKEN }}
18+
troika_user: ${{ secrets.HPC_CI_TESTING_SSH_USER }}
19+
repository: ecmwf/earthkit-meteo@${{ github.event.pull_request.head.sha || github.sha }}
20+
build_config: .github/ci-hpc-gpu-config.yml
21+
python_version: "3.10"
22+
python_toml_opt_dep_sections: gpu,test

pyproject.toml

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -31,6 +31,10 @@ dependencies = [
3131
"earthkit-utils>=0.0.1",
3232
"numpy",
3333
]
34+
optional-dependencies.gpu = [
35+
"cupy",
36+
"torch",
37+
]
3438
optional-dependencies.test = [
3539
"pytest",
3640
"pytest-cov",

0 commit comments

Comments
 (0)