o h@s*ddlZddlmZGdddejZdS)Nc@s,eZdZdZejdejdejfddZdS)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. modulereturncCsdS)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)selfrrrV/var/www/html/ai/venv/lib/python3.10/site-packages/torch/distributed/_shard/sharder.pyshards z Sharder.shardN) __name__ __module__ __qualname____doc__abcabstractmethodnnModulerrrrrrs r)r torch.nnrABCrrrrrs