nix.yml

name: "CI build"

on: [workflow_dispatch]

jobs:
  build:
    runs-on: ubuntu-latest-16-cores
    permissions:
      contents: 'read'
      id-token: 'write'
    steps:
    - uses: actions/checkout@v4
    - uses: DeterminateSystems/nix-installer-action@v10
    - name: Authenticate to Google Cloud Platform
      uses: google-github-actions/auth@v2
      with:
        project_id: yorick-dev-416917
        workload_identity_provider: projects/752785843927/locations/global/workloadIdentityPools/github/providers/github-actions-workload-prvdr
        service_account: github-actions-service-account@yorick-dev-416917.iam.gserviceaccount.com
    - name: Set up cache
      uses: zombiezen/setup-nix-cache-action@v0.4.0
      with:
        substituters: gs://replicate-nix-cache-dev
        secret_keys: ${{ secrets.NIX_PRIVATE_KEY }}
        use_nixcached: true
        nixcached_upload_options: -j12
    - name: Get ssh key from secret
      env:
        GIT_SSH_KEY: ${{ secrets.GIT_SSH_KEY }}
      run: |
        mkdir -p ~/.ssh
        cat > ~/.ssh/id_ed25519 <<< "$GIT_SSH_KEY"
        chmod 0600 ~/.ssh/id_ed25519
    - name: Build cog-triton-builder
      env:
        COG_TOKEN: ${{ secrets.COG_TOKEN }}
      run: |
        nix build --accept-flake-config ".#cog-triton-builder" -o cog-triton-builder
        ./cog-triton-builder push r8.im/replicate-internal/triton-builder
    - name: Build cog-triton-runner-80
      env:
        COG_TOKEN: ${{ secrets.COG_TOKEN }}
      run: |
        nix build --accept-flake-config ".#cog-triton-runner-80" -o cog-triton-runner-80
        ./cog-triton-runner-80 push r8.im/replicate-internal/triton-base-sm80
    # - name: Build cog-triton-runner-86
    #   env:
    #     COG_TOKEN: ${{ secrets.COG_TOKEN }}
    #   run: |
    #     nix build --accept-flake-config ".#cog-triton-runner-86" -o cog-triton-runner-86
    #     ./cog-triton-runner-86 push r8.im/replicate-internal/triton-base-sm86
    - name: Build cog-triton-runner-90
      env:
        COG_TOKEN: ${{ secrets.COG_TOKEN }}
      run: |
        nix build --accept-flake-config ".#cog-triton-runner-90" -o cog-triton-runner-90
        ./cog-triton-runner-90 push r8.im/replicate-internal/triton-base-sm90
    - run: nix path-info --closure-size --human-readable ./cog-triton-*