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

Method allreduce

test/distributed/test_c10d_common.py:1532–1536  ·  view source on GitHub ↗
(self, tensor_list, opts=None)

Source from the content-addressed store, hash-verified

1530 return DummyWork()
1531
1532 def allreduce(self, tensor_list, opts=None):
1533 for tensor in tensor_list:
1534 tensor.add_(2)
1535
1536 return DummyWork()
1537
1538 def barrier(self, opts=None):
1539 store = c10d._get_default_store()

Calls 1

DummyWorkClass · 0.85

Tested by

no test coverage detected