Skip to content

Commit 9b9830f

Browse files
Update benchmarks.yml
1 parent 3136300 commit 9b9830f

File tree

1 file changed

+1
-17
lines changed

1 file changed

+1
-17
lines changed

.github/workflows/benchmarks.yml

Lines changed: 1 addition & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -66,22 +66,6 @@ jobs:
6666
# uses: google-ml-infra/actions/ci_connection@main
6767
# with:
6868
# halt-dispatch-input: ${{ inputs.halt-for-connection }}
69-
70-
- name: Create gpu_hlo_backend.hlo
71-
working-directory: xla
72-
run: |
73-
cat << EOF > gpu_hlo_backend.hlo
74-
HloModule module
75-
// CHECK: is_scheduled=true
76-
77-
ENTRY computation {
78-
p = f32[5000,6000]{1,0} parameter(0)
79-
e = f32[5000,6000]{1,0} sqrt(p)
80-
c = f32[6000,5000] transpose(p), dimensions={1,0}
81-
r = f32[300,20,5000] reshape(c)
82-
ROOT out = (f32[5000,6000], f32[300,20,5000]) tuple(e,r)
83-
}
84-
EOF
8569

8670
# - name: Wait For Connection
8771
# uses: google-ml-infra/actions/ci_connection@main
@@ -92,7 +76,7 @@ jobs:
9276
working-directory: xla
9377
run: |
9478
cd ..
95-
./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/gpu_hlo_backend.hlo &> xla/results/gpu_hlo_backend.log
79+
./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 &> xla/results/gpu_hlo_backend.log
9680
9781
# - name: Wait For Connection
9882
# uses: google-ml-infra/actions/ci_connection@main

0 commit comments

Comments
 (0)