From 6478015786bb2974879213bf32b7ed1c8d18e48a Mon Sep 17 00:00:00 2001 From: wangcong Date: Wed, 13 May 2020 18:07:28 +0800 Subject: [PATCH] fix compile error --- mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc | 4 ++-- mindspore/ccsrc/operator/ops.h | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc b/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc index 825126672..c302a11cd 100644 --- a/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc +++ b/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc @@ -722,8 +722,8 @@ size_t TbeKernelBuild::GetOptionalInput(const mindspore::CNodePtr &cnode, bool i } std::string TbeKernelBuild::GetRealOpType(const std::string &origin_type) { - static std::map TbeKernelBuild::buffer_fussion_op_map = { - {"DepthwiseConv2dNative", "DepthwiseConv2D"}, {"TensorAdd", "Add"}}; + static std::map buffer_fussion_op_map = {{"DepthwiseConv2dNative", "DepthwiseConv2D"}, + {"TensorAdd", "Add"}}; string result = origin_type; if (buffer_fussion_op_map.find(origin_type) != buffer_fussion_op_map.end()) { result = buffer_fussion_op_map[origin_type]; diff --git a/mindspore/ccsrc/operator/ops.h b/mindspore/ccsrc/operator/ops.h index f46f76bfc..d7394be31 100755 --- a/mindspore/ccsrc/operator/ops.h +++ b/mindspore/ccsrc/operator/ops.h @@ -204,6 +204,7 @@ extern const PrimitivePtr kPrimOneHot; extern const PrimitivePtr kPrimGelu; extern const PrimitivePtr kPrimGeluGrad; extern const PrimitivePtr kPrimRelu; +extern const PrimitivePtr kPrimReluV2; extern const PrimitivePtr kPrimActivation; extern const PrimitivePtr kPrimZerosLikeTensor; extern const PrimitivePtr kPrimFakeBprop; -- GitLab