communicator#

FedLab communication API

Package Contents#

dtype_torch2flab(torch_type)

dtype_flab2torch(fedlab_type)

HEADER_SENDER_RANK_IDX = 0#
HEADER_RECEIVER_RANK_IDX = 1#
HEADER_SLICE_SIZE_IDX = 2#
HEADER_MESSAGE_CODE_IDX = 3#
HEADER_DATA_TYPE_IDX = 4#
DEFAULT_RECEIVER_RANK#
DEFAULT_SLICE_SIZE = 0#
DEFAULT_MESSAGE_CODE_VALUE = 0#
HEADER_SIZE = 5#
INT8 = 0#
INT16 = 1#
INT32 = 2#
INT64 = 3#
FLOAT16 = 4#
FLOAT32 = 5#
FLOAT64 = 6#
dtype_torch2flab(torch_type)#
dtype_flab2torch(fedlab_type)#