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 2ccb7a2d..d73dd2a7 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 @@ -65,7 +65,7 @@ Status CheckArgsForHwcnToFrazlstm(const TransArgs &args) { GELOGE(PARAM_INVALID, "Failed to check src shape %s", ShapeToString(args.src_shape).c_str()); return PARAM_INVALID; } - if (!CheckShapeValid(args.dst_shape, kFracZnLstmDimsNum)) { + if (!CheckShapeValid(args.dst_shape, kFracZDimsNum)) { GELOGE(PARAM_INVALID, "Failed to check dst shape %s", ShapeToString(args.dst_shape).c_str()); return PARAM_INVALID; } @@ -96,6 +96,10 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in } auto ret = memcpy_s(dst.get(), static_cast(total_size), args.data, static_cast(total_size)); + if (ret != EOK) { + GELOGE(INTERNAL_ERROR, "Failed to copy data ==="); + return INTERNAL_ERROR; + } result.data = dst; result.length = static_cast(total_size); return SUCCESS;