sgBddlZddlmZGddejZy)NcfeZdZdZej dejdejfdZy)Sharderap This is an interface which allows user to create more advanced sharding strategies that are not easily be composed by the `ShardingSpec`. :class:`torch.distributed._shard.sharding_plan.ShardingPlan` could take an object of the `Sharder` and call `shard` to shard the module, then replace the original module with sharded module returned. modulereturncy)aq Shard a module base on the implementation of this method, and return the sharded version of the module. Args: module (:class:`torch.nn.Module`): The module to apply sharding to. Returns: A :class:`torch.nn.Module` object that represents a module that's already been sharded. N)selfrs S/var/www/html/venv/lib/python3.12/site-packages/torch/distributed/_shard/sharder.pyshardz Sharder.shardsN) __name__ __module__ __qualname____doc__abcabstractmethodnnModuler rr r rrs7   BII  "))    r r)rtorch.nnrABCrrr r rs  cgg r