From 8751d19dd89de6a86de1660267325929095bbda3 Mon Sep 17 00:00:00 2001
From: zhaoxinxin <zhaoxinxin1@huawei.com>
Date: Tue, 17 Nov 2020 15:31:31 +0800
Subject: [PATCH] 	modified:   src/ge/host_kernels/concat_v2_kernel.cc

---
 src/ge/host_kernels/concat_v2_kernel.cc | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/ge/host_kernels/concat_v2_kernel.cc b/src/ge/host_kernels/concat_v2_kernel.cc
index c46b4277..35517c4d 100644
--- a/src/ge/host_kernels/concat_v2_kernel.cc
+++ b/src/ge/host_kernels/concat_v2_kernel.cc
@@ -32,7 +32,7 @@ namespace ge {
 namespace {
 const size_t kConcatV2InputNum = 3;
 const int kSupportEmptyTensorRank = 1;
-const std::set<DataType> concatv2_supported_type = {DT_INT32, DT_FLOAT};
+const std::set<DataType> concatv2_supported_type = {DT_INT32, DT_FLOAT, DT_INT64};
 
 template <typename T>
 void GetOutputData(std::vector<T> &y_data, int64_t loop, size_t &input_size,
@@ -88,6 +88,7 @@ Status ConcatV2Kernel::Compute(const ge::OpDescPtr op_desc_ptr, const vector<ge:
 
   std::vector<int32_t> y_data_int32_t;
   std::vector<float> y_data_float;
+  std::vector<int64_t> y_data_int64_t;
 
   // Index 0 can always gets a GeTensorDesc object from any OpDescPtr.
   auto output_tensor_desc = op_desc_ptr->GetOutputDesc(0);
@@ -106,6 +107,7 @@ Status ConcatV2Kernel::Compute(const ge::OpDescPtr op_desc_ptr, const vector<ge:
   switch (data_type) {
     SET_OUTPUT(DT_INT32, int32_t)
     SET_OUTPUT(DT_FLOAT, float)
+    SET_OUTPUT(DT_INT64, int64_t)
     default:
       break;
   }