IntelĀ® Machine Learning Scaling Library
2018
A library providing an efficient implementation of communication patterns used in deep learning.
|
A wrapper class for operation input and output activations. More...
#include <mlsl.hpp>
Public Member Functions | |
size_t | GetGlobalFmCount () |
size_t | GetGlobalFmOffset () |
size_t | GetLocalFmCount () |
size_t | GetPackBlockCount () |
size_t | GetUnpackBlockCount () |
CommBlockInfo * | GetPackBlock (size_t idx) |
CommBlockInfo * | GetUnpackBlock (size_t idx) |
DataType | GetDataType () |
size_t | GetFmSize () |
void * | GetCommBuf () |
size_t | GetCommBufSize () |
void | StartComm (void *buf) |
void * | WaitComm () |
A wrapper class for operation input and output activations.
Holds information about the input/output activation shape and allows performing associated communications.
void* MLSL::Activation::GetCommBuf | ( | ) |
size_t MLSL::Activation::GetCommBufSize | ( | ) |
DataType MLSL::Activation::GetDataType | ( | ) |
size_t MLSL::Activation::GetFmSize | ( | ) |
size_t MLSL::Activation::GetGlobalFmCount | ( | ) |
size_t MLSL::Activation::GetGlobalFmOffset | ( | ) |
size_t MLSL::Activation::GetLocalFmCount | ( | ) |
CommBlockInfo* MLSL::Activation::GetPackBlock | ( | size_t | idx | ) |
Returns the CommBlockInfo object containing information for packing.
idx | the object index |
size_t MLSL::Activation::GetPackBlockCount | ( | ) |
CommBlockInfo* MLSL::Activation::GetUnpackBlock | ( | size_t | idx | ) |
Returns the CommBlockInfo object containing information for unpacking.
idx | the object index |
size_t MLSL::Activation::GetUnpackBlockCount | ( | ) |
void MLSL::Activation::StartComm | ( | void * | buf | ) |
Starts the non-blocking activation/gradient exchange with respect to activation.
buf | the buffer containing the packed activation |
void* MLSL::Activation::WaitComm | ( | ) |
Waits for completion of the activation/gradient exchange with respect to activation.