MCPcopy Create free account
hub / github.com/alibaba/graph-learn / tensor_to_addr

Function tensor_to_addr

graphlearn/python/nn/pytorch/data/utils.py:92–95  ·  view source on GitHub ↗
(tensor)

Source from the content-addressed store, hash-verified

90 return addr_tensor
91
92 def tensor_to_addr(tensor):
93 addr_array = tensor.tolist()
94 addr = '.'.join([str(i) for i in addr_array[:-1]]) + ':' + str(addr_array[-1])
95 return addr
96
97 def exchange_gl_server_info(addr_tensor ,world_size, rank):
98 comm_tensor = torch.zeros([world_size, 5], dtype=torch.int32)

Callers 1

exchange_gl_server_infoFunction · 0.85

Calls 1

joinMethod · 0.80

Tested by

no test coverage detected