ci.yml

name: Codebase tests (GPU) 

on:
  pull_request:
  push:
    branches:
      - eighty-devx-gpu-ci

  # 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:
      - uses: actions/checkout@v4

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

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

      - name: Authenticate gcloud
        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: Run tests
      #   env:
      #     PATH: "/home/runner/.pixi/bin:/home/runner/.local/bin:$PATH"
      #     CONDA_PREFIX: "/home/runner/.pixi/envs/default"
      #   run: pixi run all-tests-ci

      - name: Run tests
        uses: prefix-dev/setup-pixi@v0.8.1
        env:
          RUNNER_DEBUG: true
        with:
          pixi-version: v0.25.0
          manifest-path: pyproject.toml
          activate-environment: true
          cache: true
      - run: |
          cp assets/system/libEGL.so .pixi/envs/default/x86_64-conda-linux-gnu/sysroot/usr/lib64/
          pixi run all-tests-ci