Skip to content

Commit faade52

Browse files
ybaturinaGoogle-ML-Automation
authored andcommitted
Check H100/B200 test results.
PiperOrigin-RevId: 834325957
1 parent 00d707e commit faade52

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

.github/workflows/bazel_cuda_h100_b200.yml

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ jobs:
8181
--repo_env=HERMETIC_CUDA_UMD_VERSION="13.0.0" \
8282
--test_env=XLA_PYTHON_CLIENT_ALLOCATOR=platform \
8383
--run_under "$(pwd)/build/parallel_accelerator_execute.sh" \
84-
--test_output=errors \
84+
--test_output=all \
8585
--test_tag_filters=-multiaccelerator \
8686
--test_env=JAX_ACCELERATOR_COUNT=1 \
8787
--test_env=JAX_TESTS_PER_ACCELERATOR=8 \
@@ -102,7 +102,8 @@ jobs:
102102
//tests/pallas:gpu_tests \
103103
//tests/mosaic:gpu_tests
104104
run_multiaccelerator_tests:
105-
if: ${{ github.event.repository.fork == false && (github.event_name == 'schedule' || github.event_name == 'workflow_dispatch' || contains(github.event.pull_request.labels.*.name, 'CI Optional GPU Presubmit')) }}
105+
needs: changed_files
106+
if: ${{ github.event.repository.fork == false && (github.event_name == 'schedule' || needs.changed_files.outputs.any_changed == 'true' || github.event_name == 'workflow_dispatch' || contains(github.event.pull_request.labels.*.name, 'CI Optional GPU Presubmit')) }}
106107
runs-on: linux-x86-a3-8g-h100-8gpu
107108
container: 'us-docker.pkg.dev/ml-oss-artifacts-published/ml-public-container/ml-build:latest'
108109
name: "Bazel multiple H100 CUDA tests"
@@ -124,7 +125,7 @@ jobs:
124125
--repo_env=HERMETIC_CUDNN_VERSION="9.11.0" \
125126
--repo_env=HERMETIC_CUDA_UMD_VERSION="13.0.0" \
126127
--test_env=XLA_PYTHON_CLIENT_ALLOCATOR=platform \
127-
--test_output=errors \
128+
--test_output=all \
128129
--strategy=TestRunner=local \
129130
--local_test_jobs=8 \
130131
--test_env=JAX_EXCLUDE_TEST_TARGETS='PmapTest.testSizeOverflow|.*InterpretTest.*' \

0 commit comments

Comments
 (0)