Arcane::NCCLVariableSynchronizeDispatcher Class Reference

NCCL implementation of synchronization. More...

Inheritance diagram for Arcane::NCCLVariableSynchronizeDispatcher:

Collaboration diagram for Arcane::NCCLVariableSynchronizeDispatcher:

Classes
class	Factory

Public Member Functions
	NCCLVariableSynchronizeDispatcher (Factory *f)
Public Member Functions inherited from Arcane::AbstractDataSynchronizeImplementation
void	setDataSynchronizeInfo (DataSynchronizeInfo *sync_info) final

Protected Member Functions
void	compute () override
void	beginSynchronize (IDataSynchronizeBuffer *ds_buf) override
void	endSynchronize (IDataSynchronizeBuffer *ds_buf) override
Protected Member Functions inherited from Arcane::AbstractDataSynchronizeImplementation
DataSynchronizeInfo *	_syncInfo () const

Private Attributes
IParallelMng *	m_parallel_mng = nullptr
ncclComm_t	m_nccl_communicator

Detailed Description

NCCL implementation of synchronization.

NCCL only supports one GPU per rank.

Definition at line 54 of file NCCLVariableSynchronizeDispatcher.cc.

Constructor & Destructor Documentation

NCCLVariableSynchronizeDispatcher()

Arcane::NCCLVariableSynchronizeDispatcher::NCCLVariableSynchronizeDispatcher ( Factory * f )

explicit

Definition at line 110 of file NCCLVariableSynchronizeDispatcher.cc.

Member Function Documentation

beginSynchronize()

void Arcane::NCCLVariableSynchronizeDispatcher::beginSynchronize ( IDataSynchronizeBuffer * ds_buf )

overrideprotectedvirtual

Implements Arcane::IDataSynchronizeImplementation.

Definition at line 132 of file NCCLVariableSynchronizeDispatcher.cc.

compute()

void Arcane::NCCLVariableSynchronizeDispatcher::compute ( )

inlineoverrideprotectedvirtual

Implements Arcane::IDataSynchronizeImplementation.

Definition at line 64 of file NCCLVariableSynchronizeDispatcher.cc.

endSynchronize()

void Arcane::NCCLVariableSynchronizeDispatcher::endSynchronize ( IDataSynchronizeBuffer * ds_buf )

overrideprotectedvirtual

Implements Arcane::IDataSynchronizeImplementation.

Definition at line 182 of file NCCLVariableSynchronizeDispatcher.cc.

Member Data Documentation

m_nccl_communicator

ncclComm_t Arcane::NCCLVariableSynchronizeDispatcher::m_nccl_communicator

private

Definition at line 71 of file NCCLVariableSynchronizeDispatcher.cc.

m_parallel_mng

IParallelMng* Arcane::NCCLVariableSynchronizeDispatcher::m_parallel_mng = nullptr

private

Definition at line 70 of file NCCLVariableSynchronizeDispatcher.cc.

---

The documentation for this class was generated from the following file:

• NCCLVariableSynchronizeDispatcher.cc