MCPcopy Create free account
hub / github.com/pytorch/pytorch / reduce_scatter

Method reduce_scatter

test/distributed/test_c10d_common.py:1563–1567  ·  view source on GitHub ↗
(self, output_tensor_list, input_tensor_lists, opts=None)

Source from the content-addressed store, hash-verified

1561 return DummyWork()
1562
1563 def reduce_scatter(self, output_tensor_list, input_tensor_lists, opts=None):
1564 for output_tensor, input_tensor_list in zip(output_tensor_list, input_tensor_lists):
1565 output_tensor.copy_(input_tensor_list[self.rank()])
1566
1567 return DummyWork()
1568
1569 def send(self, tensor_list, dst, tag=0):
1570 for tensor in tensor_list:

Callers 9

test_reduce_scatterMethod · 0.45
test_collectivesMethod · 0.45
comm_fnMethod · 0.45
test_reduce_scatterMethod · 0.45
test_reduce_scatterMethod · 0.45
test_reduce_scatterMethod · 0.45

Calls 3

rankMethod · 0.95
DummyWorkClass · 0.85
copy_Method · 0.45

Tested by

no test coverage detected