1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
|
name: Full GPU Testing
on: # yamllint disable-line rule:truthy
workflow_dispatch:
# schedule:
# - cron: "0 6 * * *" # Everyday at 6:00am UTC/10:00pm PST
jobs:
full_gpu_pytest:
if: github.repository == 'pyg-team/pytorch_geometric'
runs-on: [self-hosted, nvidia]
steps:
- name: Checkout repository
uses: actions/checkout@v5
- name: Setup packages
uses: ./.github/actions/setup
with:
cuda-version: 'cu128'
- name: Print GPU information
run: |
nvidia-smi
- name: Install main package
run: |
uv pip install -e ".[full,test]"
- name: Check installation
run: |
uv run --no-project python -c "import torch; print('PyTorch:', torch.__version__)"
uv run --no-project python -c "import torch; print('CUDA available:', torch.cuda.is_available())"
uv run --no-project python -c "import torch; print('CUDA:', torch.version.cuda)"
shell: bash
- name: Run tests
timeout-minutes: 20
run: |
FULL_TEST=1 uv run --no-project pytest --durations 10
shell: bash
|
