Build fully connected graph which edges across called computations.
Restructured sharding passes to propagate sharding on pass-through instructions which now the placer does not assign anymore (GTEs, tuples, bitcast, parameters, ...). PiperOrigin-RevId: 203591020
Loading
Please sign in to comment