
    Vh                         d dl mZ d dlmZ  G d dej
                        Zd Zej                  j                  deddg	       y)
    NFakeProcessGroupc                       e Zd ZdZy)	FakeStorez
    A fake store is a fake Key-Value store simply for initialization usage
    the of fake process group, one can either use FakeStore or HashStore.
    N)__name__
__module____qualname____doc__     [/home/dcms/DCMS/lib/python3.12/site-packages/torch/testing/_internal/distributed/fake_pg.pyr   r   
   s    r   r   c                     t        ||      S )a  
    A fake process group (not related to FakeTensor) is a process group which
    doesn't actually do any communication, it just hallucinates some
    communication.  You can run a single rank with a fake process group
    without needing multiple processes (simulates per-rank behavior)

    NOTE: This is not a real process group, and it would produce wrong results
    for every collective. It should be used as a convinient tool when playing
    with distributed but don't care about the actual data.
    r   )prefix_storerank
world_sizetimeouts       r   _create_fake_pgr      s     D*--r   fakecpucuda)devices)
torch.distributeddistributeddisttorch._C._distributed_c10dr   Storer   r   Backendregister_backendr   r   r   <module>r      sC    !


 .   fov  Or   