From 488d5c3aea421a2c306ba675eefb16217e67ef1f Mon Sep 17 00:00:00 2001 From: "dongduo5@huawei.com" Date: Mon, 12 Apr 2021 15:26:15 +0800 Subject: [PATCH] add lstm cmakelist --- .../format_transfer_hwcn_fractal_zn_lstm.cc | 12 ++++++------ ge/common/ge_common.mk | 1 + 2 files changed, 7 insertions(+), 6 deletions(-) diff --git a/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc b/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc index 064f4d40..669b7895 100644 --- a/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc +++ b/ge/common/formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc @@ -43,9 +43,9 @@ Status TransShapeHwcnToFrazlstm(const DataType &data_type, const std::vector &dst_shape) { if (src_format == FORMAT_HWCN && CheckDataTypeSupported(data_type)) { if (!CheckShapeValid(src_shape, kHwcnDimsNum)) { - GELOGE(ACL_ERROR_GE_TRANSSHAPE_SHAPE_INVALID, "Failed to check src shape %s", + GELOGE(ACL_ERROR_GE_SHAPE_INVALID, "Failed to check src shape %s", ShapeToString(src_shape).c_str()); - return ACL_ERROR_GE_TRANSSHAPE_SHAPE_INVALID; + return ACL_ERROR_GE_SHAPE_INVALID; } return TransShapeHwcnToFrazlstm(data_type, src_shape, dst_shape); } else if (src_format != FORMAT_HWCN) { - return ACL_ERROR_GE_TRANSSHAPE_FORMAT_INVALID; + return ACL_ERROR_GE_SHAPE_INVALID; } else { - return ACL_ERROR_GE_TRANSSHAPE_DATATYPE_INVALID; + return ACL_ERROR_GE_SHAPE_INVALID; } } diff --git a/ge/common/ge_common.mk b/ge/common/ge_common.mk index e28090ad..7a7a3801 100755 --- a/ge/common/ge_common.mk +++ b/ge/common/ge_common.mk @@ -31,6 +31,7 @@ GE_COMMON_LOCAL_SRC_FILES := \ formats/format_transfers/format_transfer_dhwcn_fracz3D.cc \ formats/format_transfers/format_transfer_dhwnc_fracz3D_transpose.cc \ formats/format_transfers/format_transfer_nchw_fz_c04.cc \ + formats/format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc \ formats/formats.cc \ ge_format_util.cc \ fmk_error_codes.cc \