Fix test failure in GPU CI if NCCL_DEBUG is enabled.
Fix test failure in GPU CI if NCCL_DEBUG is enabled.

If NCCL_DEBUG is enabled, NCCL prints extra status information. Make
test accept this.

google/jax,google/jax,google/jax,google/jax