Skip to content

[sharding_in_types] Add support for nary ops to propagate sharding when 1 input is sharded and all others are replicated. #1836

[sharding_in_types] Add support for nary ops to propagate sharding when 1 input is sharded and all others are replicated.

[sharding_in_types] Add support for nary ops to propagate sharding when 1 input is sharded and all others are replicated. #1836

Triggered via pull request October 10, 2024 03:42
Status Success
Total duration 53s
Artifacts

jax-array-api.yml

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