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 concatv2_supported_type = {DT_INT32, DT_FLOAT}; +const std::set concatv2_supported_type = {DT_INT32, DT_FLOAT, DT_INT64}; template void GetOutputData(std::vector &y_data, int64_t loop, size_t &input_size, @@ -88,6 +88,7 @@ Status ConcatV2Kernel::Compute(const ge::OpDescPtr op_desc_ptr, const vector y_data_int32_t; std::vector y_data_float; + std::vector 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