[XLA] Increase the time limit for cpu_benchmarks.yml to 9 hours (540 …

…minutes). PiperOrigin-RevId: 732240277
openxla · Feb 28, 2025 · 83baf9b · 83baf9b
1 parent 0c04616
commit 83baf9b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/.github/workflows/cpu_benchmarks.yml b/.github/workflows/cpu_benchmarks.yml
@@ -49,7 +49,7 @@ jobs:
     defaults:
       run:
         shell: bash
-    timeout-minutes: 360
+    timeout-minutes: 540
     steps:
       - name: Print machine specs
         run: |
@@ -108,6 +108,7 @@ jobs:
           cd tmp_hlo
           wget https://storage.googleapis.com/xla-benchmarking-temp/gemma2_2b_keras_jax.hlo
           cd ..
+          ./bazel-bin/xla/tools/run_hlo_module --input_format=hlo --platform=CPU tmp_hlo/gemma2_2b_keras_jax.hlo
 
       - name: Compute the cost of gemma2_2b_keras_jax.hlo
         run: |