Skip to content
CI - Cloud TPU (presubmit)

Add reshard API in experimental. Currently for sharding_in_types we have 2 APIs: mesh_cast and reshard. Both work in sharding_in_types mode and affect the sharding of the aval. Following are the semantics of both: #1760

Sign in to view logs

Add reshard API in experimental. Currently for sharding_in_types we have 2 APIs: mesh_cast and reshard. Both work in sharding_in_types mode and affect the sharding of the aval. Following are the semantics of both:

Add reshard API in experimental. Currently for sharding_in_types we have 2 APIs: mesh_cast and reshard. Both work in sharding_in_types mode and affect the sharding of the aval. Following are the semantics of both: #1760

Triggered via pull request January 20, 2025 19:39
@copybara-servicecopybara-service[bot]
synchronize #25978
test_716879595
Status Success
Total duration 7m 13s
Artifacts –

cloud-tpu-ci-presubmit.yml

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