From 67aa5682cd756c9853e17d8baf3030d4607089a3 Mon Sep 17 00:00:00 2001 From: "dongduo5@huawei.com" Date: Tue, 13 Apr 2021 21:38:40 +0800 Subject: [PATCH] add fractal zn lstm --- .../format_transfer_hwcn_fractal_zn_lstm.cc | 6 +++--- 1 file changed, 3 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 12dd57e0..8cf0bdd8 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 @@ -92,7 +92,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in result.data = dst; result.length = static_cast(total_size); return SUCCESS; - + std::shared_ptr dst(new (std::nothrow) uint8_t[total_size], std::default_delete()); if (dst == nullptr) { GELOGE(OUT_OF_MEMORY, "Failed to trans format from %s to %s, can not alloc the memory for dst buf %ld, shape %s", @@ -116,7 +116,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in int64_t cn = c * n; int64_t wcn = w * cn; - for (int64_t c1_idx = 0; c1_idx < c1; c1_idx++) { + /*for (int64_t c1_idx = 0; c1_idx < c1; c1_idx++) { int64_t c1_head_addr = c1_idx * hwncoc0; for (int64_t h_idx = 0; h_idx < h; h_idx++) { int64_t h_head_addr = c1_head_addr + h_idx * wncoc0; @@ -163,7 +163,7 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in } } } - } + }*/ result.data = dst; result.length = static_cast(total_size); return SUCCESS;