Skip to content

Commit d56c1db

Browse files
Update benchmarks.yml
1 parent 89deed1 commit d56c1db

File tree

1 file changed

+23
-24
lines changed

1 file changed

+23
-24
lines changed

.github/workflows/benchmarks.yml

Lines changed: 23 additions & 24 deletions
Original file line numberDiff line numberDiff line change
@@ -91,25 +91,24 @@ jobs:
9191
working-directory: xla
9292
continue-on-error: true
9393
run: nvidia-smi
94-
- name: Wait For Connection
95-
uses: google-ml-infra/actions/ci_connection@main
96-
with:
97-
halt-dispatch-input: ${{ inputs.halt-for-connection }}
98-
99-
- name: Configure XLA
100-
working-directory: xla
101-
run: ./configure.py --backend CUDA --nccl
102-
103-
- name: Set TF_CPP_MAX_VLOG_LEVEL
104-
working-directory: xla
105-
run: echo "TF_CPP_MAX_VLOG_LEVEL=1" >> $GITHUB_ENV
10694

95+
# - name: Configure XLA
96+
# working-directory: xla
97+
# run: ./configure.py --backend CUDA --nccl
10798

99+
# - name: Set TF_CPP_MAX_VLOG_LEVEL
100+
# working-directory: xla
101+
# run: echo "TF_CPP_MAX_VLOG_LEVEL=1" >> $GITHUB_ENV
108102

109103
- name: Build run_hlo_module with GPU
110104
working-directory: xla
111105
run: bazel build -c opt --dynamic_mode=off xla/tools:run_hlo_module --config=cuda
112106

107+
- name: Wait For Connection
108+
uses: google-ml-infra/actions/ci_connection@main
109+
with:
110+
halt-dispatch-input: ${{ inputs.halt-for-connection }}
111+
113112
- name: Run HLO Module Benchmarks withg GPU in xla/tests/fuzz
114113
working-directory: xla
115114
continue-on-error: true
@@ -135,16 +134,16 @@ jobs:
135134
echo "--------------------------------------------------"
136135
done
137136
138-
- name: Build hlo_runner_main
139-
working-directory: xla
140-
run: bazel build -c opt --config=cuda --dynamic_mode=off //xla/tools/multihost_hlo_runner:hlo_runner_main
137+
# - name: Build hlo_runner_main
138+
# working-directory: xla
139+
# run: bazel build -c opt --config=cuda --dynamic_mode=off //xla/tools/multihost_hlo_runner:hlo_runner_main
141140

142-
- name: Run XLA GPU microbenchmarks with hlo_runner_main
143-
working-directory: xla
144-
continue-on-error: true
145-
run: |
146-
for file in xla/tools/multihost_hlo_runner/data/*.hlo; do
147-
filename=$(basename "$file")
148-
echo "Running benchmark on $file"
149-
bazel-bin/xla/tools/multihost_hlo_runner/hlo_runner_main "$file"
150-
done
141+
# - name: Run XLA GPU microbenchmarks with hlo_runner_main
142+
# working-directory: xla
143+
# continue-on-error: true
144+
# run: |
145+
# for file in xla/tools/multihost_hlo_runner/data/*.hlo; do
146+
# filename=$(basename "$file")
147+
# echo "Running benchmark on $file"
148+
# bazel-bin/xla/tools/multihost_hlo_runner/hlo_runner_main "$file"
149+
# done

0 commit comments

Comments
 (0)