Skip to content
CI - Cloud TPU (presubmit)

Make device_put resharding on single device array input work under use_mesh. Fixes https://github.com/jax-ml/jax/issues/26552 #3801

# to view logs

Make device_put resharding on single device array input work under use_mesh. Fixes https://github.com/jax-ml/jax/issues/26552

Make device_put resharding on single device array input work under use_mesh. Fixes https://github.com/jax-ml/jax/issues/26552 #3801

Triggered via pull request February 15, 2025 05:33
@copybara-servicecopybara-service[bot]
synchronize #26553
test_727172487
Status Success
Total duration 7m 53s
Artifacts –

cloud-tpu-ci-presubmit.yml

on: pull_request
Matrix: cloud-tpu-test
Fit to window
Zoom out
Zoom in