Add reshard
API in experimental. Currently for sharding_in_types we have 2 APIs: mesh_cast
and reshard
. Both work in sharding_in_types mode and affect the sharding of the aval. Following are the semantics of both:
#70428
Annotations
1 error
Set up Python
WARNING: Running pip as the 'root' user can result in broken permissions and conflicting behaviour with the system package manager, possibly rendering your system unusable.It is recommended to use a virtual environment instead: https://pip.pypa.io/warnings/venv. Use the --root-user-action option if you know what you are doing and want to suppress this warning.
|
Loading