Searched refs:NcclManager (Results 1 – 8 of 8) sorted by relevance
50 int NcclManager::instance_count = 0;72 struct NcclManager::NcclStream : public core::RefCounted {99 struct NcclManager::CommunicatorMember {111 struct NcclManager::Communicator {162 struct NcclManager::Collective : public core::RefCounted {181 if (NcclManager::instance_count > 1) { in Collective()228 NcclManager::NcclManager() { in NcclManager() function in tensorflow::NcclManager234 NcclManager::~NcclManager() { in ~NcclManager()250 NcclManager* NcclManager::instance() { in instance()251 static NcclManager* instance = new NcclManager(); in instance()[all …]
287 NcclManager::DoneCallback CreateDoneCallback(TestCase* test_case) { in CreateDoneCallback()298 NcclManager nccl_manager;338 auto participant = absl::make_unique<NcclManager::Participant>( in RunMultiNodeAllReduceTest()393 auto participant = absl::make_unique<NcclManager::Participant>( in RunMultiNodeBroadcastTest()480 auto participant = absl::make_unique<NcclManager::Participant>( in TYPED_TEST()484 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()547 auto participant = absl::make_unique<NcclManager::Participant>( in TYPED_TEST()551 NcclManager::instance()->AddToAllReduce( in TYPED_TEST()589 auto participant = absl::make_unique<NcclManager::Participant>( in TYPED_TEST()593 NcclManager::instance()->AddToAllGather( in TYPED_TEST()[all …]
49 class NcclManager {52 NcclManager();53 ~NcclManager();55 static NcclManager* instance();259 TF_DISALLOW_COPY_AND_ASSIGN(NcclManager);
37 NcclManager nccl_manager_;80 auto participant = absl::make_unique<NcclManager::Participant>( in Enqueue()106 NcclManager::Context context( in Enqueue()
110 auto participant = absl::make_unique<NcclManager::Participant>( in ComputeAsync()113 NcclManager::instance()->AddToAllReduce( in ComputeAsync()141 auto participant = absl::make_unique<NcclManager::Participant>( in ComputeAsync()144 NcclManager::instance()->AddReduceSend( in ComputeAsync()177 auto participant = absl::make_unique<NcclManager::Participant>( in ComputeAsync()180 NcclManager::instance()->AddReduceRecv( in ComputeAsync()211 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()250 NcclManager::instance()->AddBroadcastRecv( in ComputeAsync()
73 NcclManager::instance()->GenerateCommunicatorKey(); in InitializeCollectiveGroupRuntimeDetails()
25 class NcclManager; variable
39 class NcclManager; variable