Skip to content
GitLab

added changes for resharding MP parts

Merged Administrator requested to merge reshard_mp_changes into main

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