update datatype

pull/405/head
wqtshg 4 years ago
parent 61f5835f2e
commit 5292a506a7

@ -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 = {

Loading…
Cancel
Save