Searched refs:NcclManager (Results 1 – 5 of 5) sorted by relevance
47 struct NcclManager::NcclStream {71 struct NcclManager::CommunicatorMember {83 struct NcclManager::Communicator {130 struct NcclManager::Collective {182 NcclManager::NcclManager() {} in NcclManager() function in tensorflow::NcclManager183 NcclManager::~NcclManager() {} in ~NcclManager()184 NcclManager* NcclManager::instance() { in instance()185 static NcclManager* instance = new NcclManager(); in instance()189 string NcclManager::GenerateCommunicatorKey() { in GenerateCommunicatorKey()195 Status NcclManager::GetCommunicator(NcclManager::Collective* collective, in GetCommunicator()[all …]
215 NcclManager::DoneCallback CreateDoneCallback(TestCase* test_case) { in CreateDoneCallback()272 auto participant = absl::make_unique<NcclManager::Participant>( in TYPED_TEST()276 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()341 auto participant = absl::make_unique<NcclManager::Participant>( in TYPED_TEST()345 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()384 auto participant = absl::make_unique<NcclManager::Participant>( in TYPED_TEST()388 NcclManager::instance()->AddToAllGather( in TYPED_TEST()403 NcclManager::instance()->GenerateCommunicatorKey(); in TEST()414 std::vector<NcclManager> nccl_managers(num_nodes); in TYPED_TEST()439 auto participant = absl::make_unique<NcclManager::Participant>( in TYPED_TEST()[all …]
43 class NcclManager {46 NcclManager();47 ~NcclManager();49 static NcclManager* instance();228 TF_DISALLOW_COPY_AND_ASSIGN(NcclManager);
106 auto participant = absl::make_unique<NcclManager::Participant>( in ComputeAsync()110 NcclManager::instance()->AddToAllReduce( in ComputeAsync()138 auto participant = absl::make_unique<NcclManager::Participant>( in ComputeAsync()142 NcclManager::instance()->AddReduceSend( in ComputeAsync()175 auto participant = absl::make_unique<NcclManager::Participant>( in ComputeAsync()179 NcclManager::instance()->AddReduceRecv( in ComputeAsync()210 auto participant = absl::make_unique<NcclManager::Participant>( in ComputeAsync()214 NcclManager::instance()->AddBroadcastSend( in ComputeAsync()247 auto participant = absl::make_unique<NcclManager::Participant>( in ComputeAsync()251 NcclManager::instance()->AddBroadcastSend( in ComputeAsync()
56 NcclManager::instance()->GenerateCommunicatorKey(); in InitializeInstanceBeforeGroupDiscovery()144 auto participant = absl::make_unique<NcclManager::Participant>( in Run()155 NcclManager::instance()->AddToAllReduce( in Run()182 NcclManager::instance()->SignalMultiNodeReady(nccl_collective_key); in Run()