Update rbe.yml

TraceMachina · Sep 11, 2024 · 2f2db4c · 2f2db4c
1 parent 1ecd6b2
commit 2f2db4c
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/.github/workflows/rbe.yml b/.github/workflows/rbe.yml
@@ -8,7 +8,7 @@ permissions: read-all
 
 jobs:
   tensorflow-rbe-test:
-    runs-on: ubuntu-22.04
+    runs-on: ubuntu-22.04-16core
     env:
       CC: gcc-12
       CXX: g++-12
@@ -37,7 +37,6 @@ jobs:
           --bes_header=x-nativelink-api-key=${{ secrets.NATIVELINK_COM_SECRET }} \
           --bes_results_url=https://app.nativelink.com/a/bclark8923/build \
           --remote_header=x-nativelink-project=tensorflow \
+          --remote_cache=grpcs://cas-bclark8923-d21fad.build-faster.nativelink.net \
+          --remote_default_exec_properties="container-image=docker://tensorflow/tensorflow:nightly" \
           //tensorflow/tools/pip_package:wheel
-
-      # --remote_cache=grpcs://cas-bclark8923-d21fad.build-faster.nativelink.net \
-      # --remote_default_exec_properties="container-image=docker://tensorflow/tensorflow:devel"