|
|
|
@ -39,6 +39,8 @@ static std::map<int64_t, HcclDataType> kConstOpHcclDataType = {
|
|
|
|
|
{ge::DT_FLOAT16, HCCL_DATA_TYPE_FP16},
|
|
|
|
|
{ge::DT_INT8, HCCL_DATA_TYPE_INT8},
|
|
|
|
|
{ge::DT_INT32, HCCL_DATA_TYPE_INT32},
|
|
|
|
|
{ge::DT_INT64, HCCL_DATA_TYPE_INT64},
|
|
|
|
|
{ge::DT_UINT64, HCCL_DATA_TYPE_UINT64},
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
static std::map<HcclDataType, int32_t> kConstOpHcclDataTypeSize = {
|
|
|
|
@ -46,6 +48,8 @@ static std::map<HcclDataType, int32_t> kConstOpHcclDataTypeSize = {
|
|
|
|
|
{HCCL_DATA_TYPE_FP16, sizeof(float) / 2},
|
|
|
|
|
{HCCL_DATA_TYPE_INT8, sizeof(int8_t)},
|
|
|
|
|
{HCCL_DATA_TYPE_INT32, sizeof(int32_t)},
|
|
|
|
|
{HCCL_DATA_TYPE_INT64, sizeof(int64_t)},
|
|
|
|
|
{HCCL_DATA_TYPE_UINT64, sizeof(uint64_t)},
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
static std::map<HorovodReduceOp, HcclReduceOp> kHorovodRedOpToHcclRedOp = {
|
|
|
|
|