diff --git a/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc b/mindspore/ccsrc/kernel/tbe/tbe_kernel_build.cc index 825126672bd8796f5019c5c1e404afce08991022..c302a11cd677336494031b265c94de5d6310539a 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 f46f76bfcee36a8ba20f1ec19e7af6869bd86dfd..d7394be31008ec1298b65fcdf4072c29e6227e4c 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;