IntelĀ® Machine Learning Scaling Library
2018
A library providing an efficient implementation of communication patterns used in deep learning.
|
A wrapper class for operation parameters. More...
#include <mlsl.hpp>
Public Member Functions | |
size_t | GetGlobalKernelCount () |
size_t | GetGlobalKernelOffset () |
size_t | GetLocalKernelCount () |
size_t | GetOwnedKernelCount () |
size_t | GetOwnedKernelOffset () |
DataType | GetDataType () |
size_t | GetKernelSize () |
bool | IsDistributedUpdate () |
void | StartGradientComm (void *buf) |
void | StartIncrementComm (void *buf) |
void * | WaitGradientComm () |
void * | TestGradientComm (bool *isCompleted) |
void * | WaitIncrementComm () |
A wrapper class for operation parameters.
Holds information about the shape of learnable parameters and allows performing associated communications.
Weights and biases should have separate instances.
DataType MLSL::ParameterSet::GetDataType | ( | ) |
size_t MLSL::ParameterSet::GetGlobalKernelCount | ( | ) |
size_t MLSL::ParameterSet::GetGlobalKernelOffset | ( | ) |
size_t MLSL::ParameterSet::GetKernelSize | ( | ) |
size_t MLSL::ParameterSet::GetLocalKernelCount | ( | ) |
size_t MLSL::ParameterSet::GetOwnedKernelCount | ( | ) |
size_t MLSL::ParameterSet::GetOwnedKernelOffset | ( | ) |
bool MLSL::ParameterSet::IsDistributedUpdate | ( | ) |
void MLSL::ParameterSet::StartGradientComm | ( | void * | buf | ) |
Starts the non-blocking exchange of the gradient with respect to parameters.
buf | the buffer containing the gradient |
void MLSL::ParameterSet::StartIncrementComm | ( | void * | buf | ) |
Starts the non-blocking exchange of parameters increment. Applicable only when distributedUpdate = true.
buf | the buffer containing the increment |
void* MLSL::ParameterSet::TestGradientComm | ( | bool * | isCompleted | ) |
Tests for completion of the exchange of gradients with respect to parameters.
isCompleted | the completion status of the request, true if request is completed, false otherwise |
void* MLSL::ParameterSet::WaitGradientComm | ( | ) |
Waits for completion of the exchange of gradients with respect to parameters.
void* MLSL::ParameterSet::WaitIncrementComm | ( | ) |
Waits for completion of the exchange of parameters increment.