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

Function _backend

imperative/python/megengine/distributed/functional.py:37–41  ·  view source on GitHub ↗
()

Source from the content-addressed store, hash-verified

35
36
37def _backend():
38 if group._sd.backend == "auto":
39 return _device2backend[what_is_xpu()]
40 else:
41 return group._sd.backend
42
43
44def collective_comm(inp, mode, group, device):

Callers 5

collective_commFunction · 0.85
remote_sendFunction · 0.85
remote_recvFunction · 0.85
_remote_send_nobackwardFunction · 0.85
_remote_recv_nobackwardFunction · 0.85

Calls 1

what_is_xpuFunction · 0.85

Tested by

no test coverage detected