From fc324c2336955a86a3c4bf249a5b7b32337831e1 Mon Sep 17 00:00:00 2001 From: "dongduo5@huawei.com" Date: Thu, 15 Apr 2021 15:08:10 +0800 Subject: [PATCH] Add transfer HWCN to Fractal_Zn_Lstm --- .../format_transfers/format_transfer_hwcn_fractal_zn_lstm.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 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 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