Yash Katariya
|
cc5af7ed98
|
Rename ReshapeableDevicesSharding to PositionalSharding and add an alias NamedSharding for MeshPspecSharding.
`MeshPspecSharding` name will be replaced with `NamedSharding` in 3 months.
PiperOrigin-RevId: 485753078
|
2022-11-02 19:13:13 -07:00 |
|
Matthew Johnson
|
95eb4249bb
|
tweaks to DevicesSharding
1. rename DevicesSharding -> ReshapeableDevicesSharding
2. fix repr to print device order faithfully
3. respect shape of np.ndarray argument to __init__
|
2022-10-25 14:28:48 -07:00 |
|
Matthew Johnson
|
43098f906a
|
initial commit of DevicesSharding (fka SimpleSharding)
need to add tests!
Co-authored-by: Yash Katariya <yashkatariya@google.com>
Co-authored-by: Sharad Vikram <sharad.vikram@gmail.com>
|
2022-10-18 21:10:24 -07:00 |
|
Yash Katariya
|
9e4114f0f1
|
Move array.py and sharding.py from experimental/ to _src/.
PiperOrigin-RevId: 477201711
|
2022-09-27 10:06:52 -07:00 |
|