Skip to content

Commit baaf5cc

Browse files
fduwjjfduwjj
committed
[5/N] Remove Replicated Tensor class
[ghstack-poisoned]
1 parent d763780 commit baaf5cc

2 files changed

Lines changed: 1 addition & 174 deletions

File tree

‎torch/distributed/_shard/api.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -126,7 +126,7 @@ def shard_parameter(
126126
@contextmanager
127127
def load_with_process_group(process_group):
128128
"""
129-
Context manager to set the process group with which to load a ShardedTensor/ReplicatedTensor.
129+
Context manager to set the process group with which to load a ShardedTensor.
130130
"""
131131
global _CURRENT_PROCESS_GROUP
132132
if _CURRENT_PROCESS_GROUP is not None:

‎torch/distributed/_shard/replicated_tensor.py‎

Lines changed: 0 additions & 173 deletions
This file was deleted.

0 commit comments

Comments
 (0)