[sharding_in_types] Enforce AxisTypes to always exist if set_mesh
is used.
#42
Job | Run time |
---|---|
3m 53s | |
4m 1s | |
2m 45s | |
2m 51s | |
13m 30s |
set_mesh
is used.
#42
Job | Run time |
---|---|
3m 53s | |
4m 1s | |
2m 45s | |
2m 51s | |
13m 30s |