Skip to content

Commit

Permalink
Initial testing with cirun based GPU CI
Browse files Browse the repository at this point in the history
  • Loading branch information
akshaysubr committed Aug 9, 2024
1 parent 02a6e9d commit ff40d3c
Show file tree
Hide file tree
Showing 2 changed files with 57 additions and 0 deletions.
9 changes: 9 additions & 0 deletions .cirun.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
runners:
- name: "zarr-gpu-runner"
cloud: "gcp"
gpu: "nvidia-tesla-t4"
instance_type: "n1-standard-1"
machine_image: "common-cu123-v20240730-ubuntu-2204-py310"
preemptible: true
labels:
- "cirun-gpu-runner"
48 changes: 48 additions & 0 deletions .github/workflows/gpu_test.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
# This workflow will install Python dependencies, run tests and lint with a variety of Python versions
# For more information see: https://help.github.com/actions/language-and-framework-guides/using-python-with-github-actions

name: Test V3

on:
push:
branches: [ v3 ]
pull_request:
branches: [ v3 ]
workflow_dispatch:

concurrency:
group: ${{ github.workflow }}-${{ github.ref }}
cancel-in-progress: true

jobs:
test:
name: py=${{ matrix.python-version }}, np=${{ matrix.numpy-version }}, deps=${{ matrix.dependency-set }}

runs-on: "cirun-gpu-runner--${{ github.run_id }}"
strategy:
matrix:
python-version: ['3.10', '3.11', '3.12']
numpy-version: ['1.24', '1.26', '2.0']
dependency-set: ["minimal", "optional"]

steps:
- uses: actions/checkout@v4
- name: GPU check
run: |
nvidia-smi
- name: Set up Python
uses: actions/setup-python@v5
with:
python-version: ${{ matrix.python-version }}
cache: 'pip'
- name: Install Hatch
run: |
python -m pip install --upgrade pip
pip install hatch
- name: Set Up Hatch Env
run: |
hatch env create test.py${{ matrix.python-version }}-${{ matrix.numpy-version }}-${{ matrix.dependency-set }}
hatch env run -e test.py${{ matrix.python-version }}-${{ matrix.numpy-version }}-${{ matrix.dependency-set }} list-env
- name: Run Tests
run: |
hatch env run --env test.py${{ matrix.python-version }}-${{ matrix.numpy-version }}-${{ matrix.dependency-set }} run

0 comments on commit ff40d3c

Please sign in to comment.