jax

3ed2fa4b - [ROCm] Rework blocking PR gate: file-driven targets, rocm_gate config, explicit target files

[ROCm] Rework blocking PR gate: file-driven targets, rocm_gate config, explicit target files - Replace ad-hoc target list in ci_blocking_test_targets.txt with the 10-test curated set selected by coverage/priority analysis: export_back_compat, sparse, lax, lax_numpy, pjit, export_harnesses, lax_autodiff, pallas, lax_numpy_operators, lax_control_flow. - Add rocm_gate Bazel config to rocm.bazelrc: routes to the multi-GPU RBE pool (linux_x64_multigpu) without matching single_gpu/multi_gpu in the script, so no tag filter is appended and the explicit target file controls selection entirely (including pjit_test_gpu which is tagged multiaccelerator). - Switch blocking-gate runner to linux-x86-64-2gpu-amd so that dynamic RBE execution can dispatch multiaccelerator tests locally without hitting a single-GPU limitation. - Add gpu-config input to bazel_rocm.yml (default: single_gpu); blocking gate passes rocm_gate. Thread GPU_CONFIG env var to the run step. - Make all bazel_rocm.yml callers explicit about which target file to use via gate-targets-file: presubmit run-tests, continuous, and nightly all pass build/rocm/ci_test_targets.txt; blocking gate passes ci_blocking_test_targets.txt.