Created by: ngoyal2707
These changes just make the function glue_megatron_parts
to be more general reshard_megatron_parts
so that it can produce different model parts as output
Created by: ngoyal2707
These changes just make the function glue_megatron_parts
to be more general reshard_megatron_parts
so that it can produce different model parts as output