Searched refs:node_states (Results 1 – 3 of 3) sorted by relevance
304 std::vector<NodeState> node_states(num_nodes); in RunMultiNodeAllReduceTest() local305 RunMultiNodeAllReduceTest(node_states, num_ranks_per_node); in RunMultiNodeAllReduceTest()308 void RunMultiNodeAllReduceTest(std::vector<NodeState>& node_states, in RunMultiNodeAllReduceTest() argument310 const int num_nodes = node_states.size(); in RunMultiNodeAllReduceTest()320 node_states[0].nccl_manager.GenerateCommunicatorKey(); in RunMultiNodeAllReduceTest()329 &node_states, &communicator_key, &collective_key, in RunMultiNodeAllReduceTest()342 node_states[node].nccl_manager.AddToAllReduce( in RunMultiNodeAllReduceTest()352 node_states[node].nccl_manager.SignalMultiNodeReady(collective_key); in RunMultiNodeAllReduceTest()369 std::vector<NodeState> node_states(num_nodes); in RunMultiNodeBroadcastTest() local371 node_states[0].nccl_manager.GenerateCommunicatorKey(); in RunMultiNodeBroadcastTest()[all …]
11 module asm "\09.section \22___kcrctab+node_states\22, \22a\22\09"
2653 const auto* node_states = scheduler_->GetNodeStates(); in TEST_F() local2656 for (const auto& nodedef_node_state : *node_states) { in TEST_F()