Use container for tests on the GPU runner.

NVIDIA · Sep 23, 2024 · d8742bf · d8742bf
1 parent 14aae79
commit d8742bf
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 3 deletions.
diff --git a/.github/actions/build/action.yml.j2 b/.github/actions/build/action.yml.j2
@@ -49,6 +49,16 @@ runs:
         ls -lahR ${{ env.ARTIFACTS_DIR }}
 <% endif %>
 
+    # TODO: Remove
+    - name: Cache build
+      id: cache-build
+      uses: actions/cache@v3
+      with:
+        path: |
+          ${{ env.ARTIFACTS_DIR }}
+        key: ${{ inputs.client-repo }}-${{ runner.os }}-build
+
+
     - if: ${{ inputs.use-container }}
       name: Build (in container)
       shell: bash --noprofile --norc -xeuo pipefail {0}
@@ -71,7 +81,7 @@ runs:
           --rm "${{ inputs.docker-image }}" \
           /bin/bash -c "${{ env.REPO_DIR }}/continuous_integration/scripts/entrypoint ${{ env.REPO_DIR }}/continuous_integration/scripts/build ${{ inputs.build-type}} ${{ inputs.target-device }}"
 
-    - if: ${{ !inputs.use-container }}
+    - if: ${{ !inputs.use-container && steps.cache-build.outputs.cache-hit != 'true'}}
       name: Build (without container)
       shell: bash --noprofile --norc -xeuo pipefail {0}
       run: |

diff --git a/.github/workflows/gh-build-and-test.yml b/.github/workflows/gh-build-and-test.yml
@@ -26,7 +26,7 @@ jobs:
       target-device: ${{ inputs.target-device }}
       runs-on: ${{ (inputs.host-platform == 'linux-x64' && 'linux-amd64-cpu16') || (inputs.host-platform == 'linux-aarch64' && 'linux-arm64-cpu16') || (inputs.host-platform == 'mac' && 'macos-latest') }}
       build-type: ${{ inputs.build-type }}
-      use-container: ${{ inputs.host-platform == 'linux-x64' || inputs.host-platform == 'linux-aarch64'}}
+      use-container: false
       host-platform: ${{ inputs.host-platform }}
       dependencies-file: ""
       build-mode: ${{ inputs.build-mode }}
@@ -46,7 +46,7 @@ jobs:
       runs-on: ${{ (inputs.host-platform == 'linux-x64' && 'linux-amd64-gpu-v100-latest-1') || (inputs.host-platform == 'linux-aarch64' && 'linux-arm64-cpu16') || (inputs.host-platform == 'mac' && 'macos-latest') }}
       runner-has-gpu: ${{ inputs.host-platform == 'linux-x64' }}
       build-type: ${{ inputs.build-type }}
-      use-container: false
+      use-container: ${{ inputs.host-platform == 'linux-x64' }}
       host-platform: ${{ inputs.host-platform }}
       dependencies-file: ""
       build-mode: ${{ inputs.build-mode }}