From 8c95eb3f964631c2f91cada352e2d119a9445f74 Mon Sep 17 00:00:00 2001
From: Klaus Zimmermann <klaus.zimmermann@quansight.com>
Date: Thu, 13 Nov 2025 17:51:54 +0100
Subject: [PATCH] Add checkout mode to linux job v2

---
 .github/workflows/linux_job_v2.yml      | 14 +++++++++++++-
 .github/workflows/test_linux_job_v2.yml |  1 +
 2 files changed, 14 insertions(+), 1 deletion(-)

diff --git a/.github/workflows/linux_job_v2.yml b/.github/workflows/linux_job_v2.yml
index 76f2c60d4c..83bc75cbfc 100644
--- a/.github/workflows/linux_job_v2.yml
+++ b/.github/workflows/linux_job_v2.yml
@@ -40,6 +40,14 @@ on:
         description: 'Number of commits to fetch, defaults to 1 similar to actions/checkout'
         default: 1
         type: number
+      single-branch:
+        description: 'Whether to fetch only a single branch, defaults to true'
+        default: true
+        type: boolean
+      checkout-mode:
+        description: 'Mode of checkout; one of "normal", "blobless", "treeless".'
+        default: "normal"
+        type: string
       submodules:
         description:
           Same as actions/checkout, set to `true` to checkout submodules or `recursive` to
@@ -124,6 +132,7 @@ jobs:
       # Will be blank outside of this
       PR_NUMBER: ${{ github.event.pull_request.number }}
       SCRIPT: ${{ inputs.script }}
+      filter: ${{ inputs.checkout-mode == 'blobless' && 'blob:none' || inputs.checkout-mode == 'treeless' && 'tree:0' || null }}
     runs-on: ${{ inputs.runner }}
     # TODO: Eventually this should run in a container, we need to make a container that matches up
     #       with the users for our self hosted runner infra since using actions/checkout with a root
@@ -177,14 +186,17 @@ jobs:
                docker exec -it $(docker container ps --format '{{.ID}}') bash
 
       - name: Checkout repository (${{ inputs.repository || github.repository }}@${{ inputs.ref }})
-        uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
+        uses: ./test-infra/.github/actions/checkout
         with:
           # Support the use case where we need to checkout someone's fork
           repository: ${{ inputs.repository || github.repository }}
           ref: ${{ inputs.ref || github.ref }}
+          single-branch: ${{ inputs.single-branch }}
           path: ${{ inputs.repository || github.repository }}
           fetch-depth: ${{ inputs.fetch-depth }}
           submodules: ${{ inputs.submodules }}
+          filter: ${{ env.filter }}
+          submodules-filter: ${{ env.filter }}
 
       - name: Calculate docker image
         id: calculate-docker-image
diff --git a/.github/workflows/test_linux_job_v2.yml b/.github/workflows/test_linux_job_v2.yml
index e078619df5..e1aee8b892 100644
--- a/.github/workflows/test_linux_job_v2.yml
+++ b/.github/workflows/test_linux_job_v2.yml
@@ -19,6 +19,7 @@ jobs:
       runner: linux.2xlarge
       test-infra-repository: ${{ github.repository }}
       test-infra-ref: ${{ github.ref }}
+      checkout-mode: "treeless"
       submodules: "recursive"
       gpu-arch-type: cpu
       gpu-arch-version: ""