Skip to content

Full GPU Testing

Full GPU Testing #3

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@v4
- name: Setup packages
uses: ./.github/actions/setup
with:
cuda-version: 'cu118'
- name: Print GPU information
run: |
nvidia-smi
- name: Install main package
run: |
pip install -e .[full,test]
- name: Run tests
run: |
FULL_TEST=1 pytest
shell: bash