Skip to content

Commit abda341

Browse files
Update benchmarks.yml
1 parent 08a8951 commit abda341

File tree

1 file changed

+14
-14
lines changed

1 file changed

+14
-14
lines changed

.github/workflows/benchmarks.yml

Lines changed: 14 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -125,20 +125,20 @@ jobs:
125125
free -h # Memory information
126126
df -h # Disk space information
127127
uname -a # Kernel information
128-
- name: Create results directory
129-
run: mkdir results
130-
- name: Configure XLA for GPU backend
131-
run: ./configure.py --backend CUDA --nccl
128+
# - name: Create results directory
129+
# run: mkdir results
130+
# - name: Configure XLA for GPU backend
131+
# run: ./configure.py --backend CUDA --nccl
132132
# - name: Set TF_CPP_MAX_VLOG_LEVEL
133133
# env:
134134
# TF_CPP_MAX_VLOG_LEVEL: 1
135-
- name: Build hlo_runner_main
136-
run: bazel build -c opt --config=cuda --dynamic_mode=off //xla/tools/multihost_hlo_runner:hlo_runner_main
137-
# TODO(juliagmt): Add more performance-criticalHLOs to benchmark.
138-
- name: Run xla/tools/hlo_opt/tests/gpu_hlo_pass.hlo
139-
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
140-
- name: Upload XSpace
141-
uses: actions/upload-artifact@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1
142-
with:
143-
name: gpu-xla-benchmarks-xspace
144-
path: xla/results/xspace.pbtxt
135+
# - name: Build hlo_runner_main
136+
# run: bazel build -c opt --config=cuda --dynamic_mode=off //xla/tools/multihost_hlo_runner:hlo_runner_main
137+
# # TODO(juliagmt): Add more performance-criticalHLOs to benchmark.
138+
# - name: Run xla/tools/hlo_opt/tests/gpu_hlo_pass.hlo
139+
# 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
140+
# - name: Upload XSpace
141+
# uses: actions/upload-artifact@b4ffde65f46336ab88eb53be808477a3936bae11 # v4.1.1
142+
# with:
143+
# name: gpu-xla-benchmarks-xspace
144+
# path: xla/results/xspace.pbtxt

0 commit comments

Comments
 (0)