MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / _SendRecvGroup

Class _SendRecvGroup

imperative/python/megengine/distributed/functional.py:778–790  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

776
777
778class _SendRecvGroup:
779 def __init__(self, rank_from, rank_to):
780 self.key = "{}->{}".format(rank_from, rank_to)
781 self.rank_from = rank_from
782 self.rank_to = rank_to
783 self.size = 2
784
785 @property
786 def rank(self):
787 if get_rank() == self.rank_from:
788 return 0
789 else:
790 return 1
791
792
793class _RemoteSend(Function):

Callers 2

remote_sendFunction · 0.85
remote_recvFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected