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 604566a8..9b4a8de2 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 @@ -111,11 +111,11 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in int index = 0; for (int64_t y_src = y_src_head; y_src < y_src_head + K; y_src++) { int64_t x_dst = x_src_head + index; + int64_t y_dst = x_src + y_src - x_dst; ++index; - int64_t src_idx = x_src * n + y_src; - auto src_offset = src_idx * size; - int64_t y_dst = x_src + y_src - x_dst; + auto src_idx = x_src * n + y_src; + auto src_offset = src_idx * size; auto dst_idx = x_dst * n + y_dst; auto dst_offset = dst_idx * size; @@ -142,6 +142,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in result.data = dst; result.length = static_cast(total_size); + return SUCCESS; } } // namespace