ci: Limit cpus in docker processes (#3251)

Signed-off-by: Dillon Cullinan <dcullinan@nvidia.com>

.github/actions/docker-build/action.yml

@@ -40,7 +40,10 @@ runs:
   using: "composite"
   steps:
     - name: Set up Docker Buildx
-      uses: docker/setup-buildx-action@v3
+      uses: docker/setup-buildx-action@e468171a9de216ec08956ac3ada2f0791b6bd435 #v3.11.1
+      with:
+        # This is not # of CPUs, this is CPU priority, 1024 is the default
+        driver-opts: 'cpu-shares=100'
     - name: Login to NGC
       if: github.event.pull_request.head.repo.full_name == github.repository || github.event_name == 'push'
       shell: bash

.github/actions/pytest/action.yml

@@ -8,6 +8,10 @@ inputs:
   image_tag:
     description: 'Image Tag to run tests on'
     required: true
+  cpu_limit:
+    description: 'Maximum number of cores available to docker'
+    required: false
+    default: '10'
 
 
 runs:
@@ -16,11 +20,13 @@ runs:
     - name: Run tests
       shell: bash
       env:
+        NUM_CPUS: ${{ inputs.cpu_limit }}
         CONTAINER_ID: test_${{ github.run_id }}_${{ github.run_attempt }}_${{ github.job }}
         PYTEST_XML_FILE: pytest_test_report.xml
         HF_HOME: /runner/_work/_temp
       run: |
         docker run --runtime=nvidia --rm --gpus all -w /workspace \
+          --cpus=${NUM_CPUS} \
           --network host \
           --name ${{ env.CONTAINER_ID }}_pytest \
           ${{ inputs.image_tag }} \