From 1aa5ed78718c0d2d1574334964ca74171266d085 Mon Sep 17 00:00:00 2001 From: "dongduo5@huawei.com" Date: Tue, 13 Apr 2021 21:47:06 +0800 Subject: [PATCH] add fractal zn lstm --- .../format_transfer_hwcn_fractal_zn_lstm.cc | 15 +++++++-------- 1 file changed, 7 insertions(+), 8 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 e7a9c55e..2ccb7a2d 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 @@ -87,12 +87,6 @@ Status CheckArgsForHwcnToFrazlstm(const TransArgs &args) { } Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const int size, const int64_t total_size) { - - auto ret = memcpy_s(dst.get(), static_cast(total_size), args.data, static_cast(total_size)); - 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", @@ -101,6 +95,11 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in return OUT_OF_MEMORY; } + auto ret = memcpy_s(dst.get(), static_cast(total_size), args.data, static_cast(total_size)); + result.data = dst; + result.length = static_cast(total_size); + return SUCCESS; + /*auto h = args.src_shape.at(kHwcnH); auto w = args.src_shape.at(kHwcnW); auto c = args.src_shape.at(kHwcnC); @@ -163,10 +162,10 @@ Status GetDstDataAfterTrans(const TransArgs &args, TransResult &result, const in } } } - }*/ + } result.data = dst; result.length = static_cast(total_size); - return SUCCESS; + return SUCCESS;*/ } } // namespace