Skip to content
JAX Array API

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

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. #4613

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

jax-array-api.yml

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