[sharding_in_types] Enforce AxisTypes to always exist if set_mesh
is used.
#48
Job | Run time |
---|---|
2m 49s | |
2m 12s | |
1m 47s | |
1m 45s | |
8m 33s |
set_mesh
is used.
#48
Job | Run time |
---|---|
2m 49s | |
2m 12s | |
1m 47s | |
1m 45s | |
8m 33s |