Skip to content

Commit 51792c6

Browse files
Update run_bazel_test.sh
Force GPU testing
1 parent 061ad8a commit 51792c6

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

ci/run_bazel_test.sh

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -58,6 +58,7 @@ if [[ $JAXCI_RUN_BAZEL_TEST_GPU_LOCAL == 1 ]]; then
5858
--config=non_multiaccelerator_local \
5959
--repo_env=HERMETIC_PYTHON_VERSION="$JAXCI_HERMETIC_PYTHON_VERSION" \
6060
--run_under "${JAXCI_JAX_GIT_DIR}/build/parallel_accelerator_execute.sh" --test_timeout=3000 \
61+
--test_env=JAX_PLATFORM_NAME="gpu" \
6162
//tests:gpu_tests //tests:backend_independent_tests //tests/pallas:gpu_tests //tests/pallas:backend_independent_tests || true
6263
echo "Finished running non-multiaccelerator tests..."
6364

0 commit comments

Comments
 (0)