Skip to content
JAX Array API

Return SingleDeviceSharding instead of GSPMDShardings when there is only 1 device during compiled.input_shardings call. #4615

Sign in to view logs

Return SingleDeviceSharding instead of GSPMDShardings when there is only 1 device during compiled.input_shardings call.

Return SingleDeviceSharding instead of GSPMDShardings when there is only 1 device during compiled.input_shardings call. #4615

Triggered via pull request November 18, 2024 18:45
@copybara-servicecopybara-service[bot]
synchronize #24951
test_697667348
Status Success
Total duration 52s
Artifacts –

jax-array-api.yml

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