build-gpu.yml

name: Build GPU

on:
  workflow_dispatch:
  push:
    branches:
      - main

jobs:
  build:

    runs-on: "cirun-gpu-runner--${{ github.run_id }}"
    defaults:
      run:
        shell: bash -l {0}
    strategy:
      matrix:
        python-version: ["3.9", "3.10"]

    steps:
      - uses: actions/checkout@v2

      - name: Run Nvidia-smi
        run: |
          nvidia-smi

      - name: Set up Python
        uses: conda-incubator/setup-miniconda@v2.2.0
        env:
          CONDA: /home/runnerx/miniconda3
        with:
          python-version: ${{ matrix.python-version }}
          miniconda-version: "latest"

      - name: Conda info
        run: |
          conda info
          conda list

      - name: Installing cudatoolkit and dependencies
        run: |
          conda install -c nvidia cudatoolkit
          pip install -r requirements.txt -r requirements-dev.txt

      - name: Numba Information
        run: |
          numba -s

      - name: Run GPU tagged tests
        run: |
          pytest -m gpu -v