Skip to content
JAX Array API

[Mosaic][TPU] Omit short circuiting of relayout (we should always relayout!) and implement product mismatch case for where we relayout from replicated to offset, and the number of vregs changes. #4442

Sign in to view logs

[Mosaic][TPU] Omit short circuiting of relayout (we should always relayout!) and implement product mismatch case for where we relayout from replicated to offset, and the number of vregs changes.

[Mosaic][TPU] Omit short circuiting of relayout (we should always relayout!) and implement product mismatch case for where we relayout from replicated to offset, and the number of vregs changes. #4442

Triggered via pull request November 14, 2024 07:50
@copybara-servicecopybara-service[bot]
synchronize #24887
test_696226548
Status Success
Total duration 59s
Artifacts –

jax-array-api.yml

on: pull_request
Matrix: build
Fit to window
Zoom out
Zoom in