File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 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.*' \
You can’t perform that action at this time.
0 commit comments