Skip to content

Commit 5ccac5e

Browse files
juliagmt-googlejuliagmt-google
authored
Update benchmarks.yml
1 parent 9067033 commit 5ccac5e

File tree

1 file changed

+30
-30
lines changed

1 file changed

+30
-30
lines changed

‎.github/workflows/benchmarks.yml‎

Lines changed: 30 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -57,29 +57,29 @@ jobs:
5757
repository: openxla/xla # Replace with your fork if needed
5858
path: xla
5959

60-
# - name: Build XLA with GPU support with RBE
61-
# working-directory: xla
62-
# continue-on-error: true
63-
# run: bazel build --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 --nobuild -- //xla/... //build_tools/... @tsl//tsl/...
60+
- name: Build XLA with GPU support with RBE
61+
working-directory: xla
62+
continue-on-error: true
63+
run: bazel build --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 --nobuild -- //xla/... //build_tools/... @tsl//tsl/...
6464

65-
# - name: Run XLA tests with GPU support with RBE
66-
# working-directory: xla
67-
# continue-on-error: true
68-
# 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/...
65+
- name: Run XLA tests with GPU support with RBE
66+
working-directory: xla
67+
continue-on-error: true
68+
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/...
6969

70-
# - name: Run Profile Analysis
71-
# working-directory: xla
72-
# continue-on-error: true
73-
# run: bazel analyze-profile profile.json.gz
70+
- name: Run Profile Analysis
71+
working-directory: xla
72+
continue-on-error: true
73+
run: bazel analyze-profile profile.json.gz
7474

7575
- name: Create results directory
7676
working-directory: xla
7777
run: mkdir -p results
7878

79-
# - name: Get GPU spec
80-
# working-directory: xla
81-
# continue-on-error: true
82-
# run: nvidia-smi
79+
- name: Get GPU spec
80+
working-directory: xla
81+
continue-on-error: true
82+
run: nvidia-smi
8383

8484
- name: Configure XLA
8585
working-directory: xla
@@ -89,9 +89,9 @@ jobs:
8989
working-directory: xla
9090
run: echo "TF_CPP_MAX_VLOG_LEVEL=1" >> $GITHUB_ENV
9191

92-
# - name: Build run_hlo_module with GPU
93-
# working-directory: xla
94-
# run: bazel build -c opt --dynamic_mode=off xla/tools:run_hlo_module --config=cuda
92+
- name: Build run_hlo_module with GPU
93+
working-directory: xla
94+
run: bazel build -c opt --dynamic_mode=off xla/tools:run_hlo_module --config=cuda
9595

9696
# - name: Wait For Connection
9797
# uses: google-ml-infra/actions/ci_connection@main
@@ -112,17 +112,17 @@ jobs:
112112
echo "Running benchmark on $file" &> results/"$file".log
113113
# ./bazel-bin/xla/tools/run_hlo_module --input_format=hlo --platform=GPU "$file" &> results/"$filename".log
114114
done
115-
- name: Wait For Connection
116-
uses: google-ml-infra/actions/ci_connection@main
117-
with:
118-
halt-dispatch-input: ${{ inputs.halt-for-connection }}
119-
# - name: Run HLO Module Benchmarks withg GPU in xla/service/gpu
120-
# working-directory: xla
121-
# continue-on-error: true
122-
# run: |
123-
# find xla/service/gpu -name "*.hlo" -print0 | while IFS= read -r -d $'\0' file; do
124-
# ./bazel-bin/xla/tools/run_hlo_module --input_format=hlo --platform=GPU "$file" &> results/"$(basename "$file")".log
125-
# done
115+
# - name: Wait For Connection
116+
# uses: google-ml-infra/actions/ci_connection@main
117+
# with:
118+
# halt-dispatch-input: ${{ inputs.halt-for-connection }}
119+
- name: Run HLO Module Benchmarks withg GPU in xla/service/gpu
120+
working-directory: xla
121+
continue-on-error: true
122+
run: |
123+
find xla/service/gpu -name "*.hlo" -print0 | while IFS= read -r -d $'\0' file; do
124+
./bazel-bin/xla/tools/run_hlo_module --input_format=hlo --platform=GPU "$file" &> results/"$(basename "$file")".log
125+
done
126126
# - name: Build hlo_runner_main
127127
# working-directory: xla
128128
# run: bazel build -c opt --config=cuda --dynamic_mode=off //xla/tools/multihost_hlo_runner:hlo_runner_main

0 commit comments

Comments
 (0)