Diff Coverage

    Returns:
        None.
    """
    return platform.set_model_state_dict(model, model_state_dict, options=options)


def hsdp_sync_stream():
    """Wait for hsdp gradient handle to be completed."""

        Raises:
            NotImplementedError: MindSpore support is not yet implemented.
        """
        raise NotImplementedError(
            "set_model_state_dict is not yet supported on MindSpore"
        )

    @staticmethod

        Raises:
            NotImplementedError: Platform subclasses must implement this method.
        """
        raise NotImplementedError(
            "Platform subclasses must implement set_model_state_dict"
        )

    @staticmethod

            unexpected_keys.append(key)
            continue
        if isinstance(target, DTensor):
            if isinstance(val, DTensor):
                scattered[key] = val
            else:
                # Slice a plain (global) tensor into a DTensor shard.
                placements = (
                    target.layout.alias_placements if target.layout else target.placements

        Delegates to torch-specific implementation that scatters full tensors
        into DTensor shards and performs an in-place load.
        """
        # pylint: disable=C0415
        from hyper_parallel.platform.torch.fully_shard.state_dict_utils import (
            set_model_state_dict as _set_model_state_dict,
        )
        return _set_model_state_dict(model, model_state_dict, options=options)

    @staticmethod
    def save_checkpoint(cell: Module, file_path: str, ckpt_format: str = "safetensors") -> None:
        if ckpt_format == "safetensors":