Home
last modified time | relevance | path

Searched refs:collective_key (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/nccl/
Dnccl_manager.h134 Context(const string& collective_key, int num_local_devices, in Context()
137 : collective_key(collective_key), in Context()
144 const string& collective_key; member
192 void SignalMultiNodeReady(const string& collective_key);
237 bool CheckReady(const string& collective_key, Collective* collective)
Dnccl_manager.cc167 : collective_key(collective_key_in), in Collective()
188 const string collective_key; // A unique key for debugging. member
470 void NcclManager::SignalMultiNodeReady(const string& collective_key) { in SignalMultiNodeReady() argument
474 auto collective_it = collectives_.find(collective_key); in SignalMultiNodeReady()
478 if (CheckReady(collective_key, collective)) { in SignalMultiNodeReady()
481 VLOG(2) << "SignalMultiNodeReady collective " << collective_key in SignalMultiNodeReady()
505 auto collective_it = collectives_.find(context.collective_key); in AddParticipant()
509 context.collective_key, data_type, collective_type, reduction_op, in AddParticipant()
512 collectives_.emplace(context.collective_key, collective); in AddParticipant()
569 "Collective ", collective->collective_key, in AddParticipant()
[all …]
Dnccl_manager_test.cc312 const string collective_key = "allreduce"; in RunMultiNodeAllReduceTest() local
329 &node_states, &communicator_key, &collective_key, in RunMultiNodeAllReduceTest()
344 {collective_key, num_ranks_per_node, num_global_ranks, in RunMultiNodeAllReduceTest()
352 node_states[node].nccl_manager.SignalMultiNodeReady(collective_key); in RunMultiNodeAllReduceTest()
368 const string collective_key = "broadcast"; in RunMultiNodeBroadcastTest() local
381 &collective_key, &communicator_key, &test_case]() { in RunMultiNodeBroadcastTest()
399 {collective_key, num_ranks_per_node, num_global_ranks, in RunMultiNodeBroadcastTest()
404 {collective_key, num_ranks_per_node, num_global_ranks, in RunMultiNodeBroadcastTest()
410 node_states[node].nccl_manager.SignalMultiNodeReady(collective_key); in RunMultiNodeBroadcastTest()
874 const string collective_key = "allreduce"; in TYPED_TEST() local
[all …]