Skip to content

Commit 176da98

Browse files
Update benchmarks.yml
1 parent d057d0f commit 176da98

File tree

1 file changed

+6
-1
lines changed

1 file changed

+6
-1
lines changed

.github/workflows/benchmarks.yml

Lines changed: 6 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -87,6 +87,11 @@ jobs:
8787
continue-on-error: true
8888
run: bazel test --build_tag_filters=-no_oss,requires-gpu-nvidia,gpu,-rocm-only --test_tag_filters=-no_oss,requires-gpu-nvidia,gpu,-rocm-only,requires-gpu-sm75-only,requires-gpu-sm60,requires-gpu-sm70,-requires-gpu-sm80,-requires-gpu-sm80-only,-requires-gpu-sm90,-requires-gpu-sm90-only,-requires-gpu-sm100,-requires-gpu-sm100-only,-requires-gpu-amd --config=warnings --config=rbe_linux_cuda_nvcc --run_under=//tools/ci_build/gpu_build:parallel_gpu_execute --repo_env=TF_CUDA_COMPUTE_CAPABILITIES=7.5 --@cuda_driver//:enable_forward_compatibility=true --test_output=errors --verbose_failures --keep_going --nobuild_tests_only --profile=profile.json.gz --flaky_test_attempts=3 --jobs=150 --bes_upload_mode=fully_async -- //xla/... //build_tools/... @tsl//tsl/...
8989

90+
- name: Run Profile Analysis
91+
working-directory: xla
92+
continue-on-error: true
93+
run: bazel analyze-profile profile.json.gz
94+
9095
- name: Get GPU spec
9196
working-directory: xla
9297
continue-on-error: true
@@ -96,7 +101,7 @@ jobs:
96101
working-directory: xla
97102
continue-on-error: true
98103
run: |
99-
for file in xla/tools/multihost_hlo_runner/data/*.hol; do
104+
for file in xla/tools/multihost_hlo_runner/data/*.hlo; do
100105
filename=$(basename "$file")
101106
echo "Running benchmark on $file"
102107
bazel-bin/xla/tools/multihost_hlo_runner/hlo_runner_main "$file"

0 commit comments

Comments
 (0)