fix

EnzymeAD · Nov 7, 2024 · 9c404f3 · 9c404f3
1 parent 2e7de65
commit 9c404f3
Showing 1 changed file with 17 additions and 3 deletions.
diff --git a/.buildkite/gpu_pipeline.yml b/.buildkite/gpu_pipeline.yml
@@ -6,6 +6,18 @@ steps:
       cuda: "*"
     if: build.message !~ /\[skip tests\]/
     timeout_in_minutes: 180
+    if: build.tag == null
+    plugins:
+      - cache#v0.6.0:
+          manifest: .buildkite/gpu_pipeline.yml
+          path: .local
+          restore: file
+          save: file
+      - cache#v0.6.0:
+          manifest: WORKSPACE
+          path: .baztmp
+          restore: file
+          save: file
     commands: |
       pwd
       env
@@ -25,10 +37,12 @@ steps:
       unset NV_LIBCUBLAS_DEV_PACKAGE
       unset NV_NVTX_VERSION
       
-      curl -fLO https://github.com/bazelbuild/bazelisk/releases/download/v1.19.0/bazelisk-linux-amd64
+      if [ -f ".local/bin/bazel" ]; then
+        curl -fLO https://github.com/bazelbuild/bazelisk/releases/download/v1.19.0/bazelisk-linux-amd64
+        mv bazel* .local/bin/bazel
+        chmod +x .local/bin/bazel
+      fi
 
-      mv bazel* .local/bin/bazel
-      chmod +x .local/bin/bazel
       export PATH="`pwd`/.local/bin:\$PATH"
 
       mkdir -p .baztmp