Skip to content
CI - Cloud TPU (presubmit)

[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level #1777

Sign in to view logs

[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level

[sharding_in_types] Move the calculation of new_mesh inside the decorator so that functions can be decorated with hidden_axes and visible_axes at the top level #1777

Run time

Learn about OS pricing on GitHub Actions
Job Run time
TPU test (jaxlib=head, v5e-8)
6m 5s
6m 5s