From 5292a506a76d4a449d15bdd7255611d2283e02d8 Mon Sep 17 00:00:00 2001
From: wqtshg <wangtao123@huawei.com>
Date: Tue, 24 Nov 2020 19:53:35 +0800
Subject: [PATCH] update datatype

---
 ge/graph/manager/util/hcom_util.h | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/ge/graph/manager/util/hcom_util.h b/ge/graph/manager/util/hcom_util.h
index 064058f8..6ae1c019 100644
--- a/ge/graph/manager/util/hcom_util.h
+++ b/ge/graph/manager/util/hcom_util.h
@@ -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 = {