Update benchmarks.yml

Author: juliagmt-google

.github/workflows/benchmarks.yml

@@ -134,13 +134,15 @@ jobs:
       - name: Set TF_CPP_MAX_VLOG_LEVEL
         env:
           TF_CPP_MAX_VLOG_LEVEL: 1
-      # - name: Build hlo_runner_main
-      #   run: bazel build -c opt --config=cuda --dynamic_mode=off //xla/tools/multihost_hlo_runner:hlo_runner_main
-      # # TODO(juliagmt): Add more performance-criticalHLOs to benchmark.
-      # - name: Run xla/tools/hlo_opt/tests/gpu_hlo_pass.hlo
-      #   run: ./bazel-bin/xla/tools/multihost_hlo_runner/hlo_runner_main --device_type=gpu --log_output=True --use_spmd_partitioning --xla_gpu_dump_xspace_to=xla/results/xspace.pbtxt xla/tools/hlo_opt/tests/gpu_hlo_pass.hlo
-      # - name: Upload XSpace
-      #   uses: actions/upload-artifact@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1
-      #   with:
-      #     name: gpu-xla-benchmarks-xspace
-      #     path: xla/results/xspace.pbtxt
+        run: |
+          echo "TF_CPP_MAX_VLOG_LEVEL is: $TF_CPP_MAX_VLOG_LEVEL"
+      - name: Build hlo_runner_main
+        run: bazel build -c opt --config=cuda --dynamic_mode=off //xla/tools/multihost_hlo_runner:hlo_runner_main
+      # TODO(juliagmt): Add more performance-criticalHLOs to benchmark.
+      - name: Run xla/tools/hlo_opt/tests/gpu_hlo_pass.hlo
+        run: ./bazel-bin/xla/tools/multihost_hlo_runner/hlo_runner_main --device_type=gpu --log_output=True --use_spmd_partitioning --xla_gpu_dump_xspace_to=xla/results/xspace.pbtxt xla/tools/hlo_opt/tests/gpu_hlo_pass.hlo
+      - name: Upload XSpace
+        uses: actions/upload-artifact@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1
+        with:
+          name: gpu-xla-benchmarks-xspace
+          path: xla/results/xspace.pbtxt