| @@ -39,6 +39,8 @@ static std::map<int64_t, HcclDataType> kConstOpHcclDataType = { | |||||
| {ge::DT_FLOAT16, HCCL_DATA_TYPE_FP16}, | {ge::DT_FLOAT16, HCCL_DATA_TYPE_FP16}, | ||||
| {ge::DT_INT8, HCCL_DATA_TYPE_INT8}, | {ge::DT_INT8, HCCL_DATA_TYPE_INT8}, | ||||
| {ge::DT_INT32, HCCL_DATA_TYPE_INT32}, | {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 = { | 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_FP16, sizeof(float) / 2}, | ||||
| {HCCL_DATA_TYPE_INT8, sizeof(int8_t)}, | {HCCL_DATA_TYPE_INT8, sizeof(int8_t)}, | ||||
| {HCCL_DATA_TYPE_INT32, sizeof(int32_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 = { | static std::map<HorovodReduceOp, HcclReduceOp> kHorovodRedOpToHcclRedOp = { | ||||