Update benchmarks.yml

Author: juliagmt-google

.github/workflows/benchmarks.yml

@@ -104,14 +104,11 @@ jobs:
       # Don't fail fast - want to see results for all builds even if one fails.
       fail-fast: false
       matrix:
-        job_info: [
-          {
-            os: "linux-x86-g2-48-l4-4gpu",
-            container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest",
-            pretty_name: "Linux X86 runner with 4 NVIDIA L4 GPUs",
-          },
+        job_info:
+          - os: "linux-x86-g2-48-l4-4gpu"
+            container: "us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build:latest"
+            pretty_name: "Linux X86 runner with 4 NVIDIA L4 GPUs"
           # Expect more GPU types in the future.
-        ]
     name: ${{ matrix.job_info.pretty_name }}
     runs-on: ${{ matrix.job_info.os }}
     container: ${{ matrix.job_info.container }}
@@ -122,7 +119,6 @@ jobs:
     steps:
       - name: Checkout XLA
         uses: actions/checkout@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1
-
       - name: Print machine specs
         run: |
           nvidia-smi
@@ -131,21 +127,16 @@ jobs:
           uname -a # Kernel information
       - name: Create results directory
         run: mkdir results
-
       - name: Configure XLA for GPU backend
-        run: |
-          ./configure.py --backend CUDA --nccl
+        run: ./configure.py --backend CUDA --nccl
       - 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
-
+        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: