def broadcast_global_state_dict(
            self, state_dict: Dict[str, Any]
        ) -> Dict[str, Any]:
            """
            Broadcasts the entire state_dict to all other ranks
            each rank is responsible to load their own partition of data
            """
            return utils.broadcast_object(
                state_dict,
                src_rank=0,
                group=self.group,
            )

            Broadcasts the entire state_dict to all other ranks
            each rank is responsible to load their own partition of data
            
structured_prompting/fairseq-version/fairseq/fairseq/optim/shard.py