ci.yml

name: Codebase tests (GPU) 

on:
  pull_request:
  push:
    branches:
      - main

  # Allows you to run this workflow manually from the Actions tab
  workflow_dispatch:

env:
  XLA_FLAGS: --xla_gpu_enable_command_buffer=
  XLA_PYTHON_CLIENT_PREALLOCATE: "false"
  XLA_PYTHON_CLIENT_ALLOCATOR: "platform"
  CPLUS_INCLUDE_PATH: "$CONDA_PREFIX/targets/x86_64-linux/include"

jobs:
  test:
    strategy:
      matrix:
        runner: [ParallelHoss]
    runs-on: ${{ matrix.runner }}
    steps:
      - name: Clone repo
      - uses: actions/checkout@v4

      - id: auth
        uses: google-github-actions/auth@v2
        with:
          credentials_json: "${{ secrets.ARTIFACT_REGISTRY_KEY }}"

      - name: Setup gcloud
        uses: google-github-actions/setup-gcloud@v2

      # - name: Install system environment and project dependencies
      #   env:
      #     GITHUB_CI: "true"
      #   run: ./install.sh

      # - name: Setup caching for test assets
      #   uses: actions/cache@v4
      #   with:
      #     path: assets
      #     key: ${{ runner.os }}-${{ matrix.runner }}-assets
      #     restore-keys: ${{ runner.os }}-${{ matrix.runner }}-assets

      # - name: Run tests
      #   uses: prefix-dev/setup-pixi@v0.8.1
      #   with:
      #     pixi-version: v0.25.0
      #     cache: true
      #   run: pixi run all-tests-ci