Searched refs:nccl_manager (Results 1 – 2 of 2) sorted by relevance
298 NcclManager nccl_manager; member320 node_states[0].nccl_manager.GenerateCommunicatorKey(); in RunMultiNodeAllReduceTest()342 node_states[node].nccl_manager.AddToAllReduce( in RunMultiNodeAllReduceTest()352 node_states[node].nccl_manager.SignalMultiNodeReady(collective_key); in RunMultiNodeAllReduceTest()371 node_states[0].nccl_manager.GenerateCommunicatorKey(); in RunMultiNodeBroadcastTest()397 node_states[node].nccl_manager.AddBroadcastSend( in RunMultiNodeBroadcastTest()402 node_states[node].nccl_manager.AddBroadcastRecv( in RunMultiNodeBroadcastTest()410 node_states[node].nccl_manager.SignalMultiNodeReady(collective_key); in RunMultiNodeBroadcastTest()886 nodes[node].nccl_manager.AddToAllReduce( in TYPED_TEST()892 nodes[node].nccl_manager.SignalMultiNodeReady(collective_key); in TYPED_TEST()[all …]
31 "nccl_manager.cc",35 "nccl_manager.h",