From 11388e588816aac55f488db01d28cfc1517a36f7 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Fri, 17 Apr 2020 15:10:23 +0800 Subject: [PATCH] refactor(mgb/parampacksplit): remove m_mem_plan_fwd GitOrigin-RevId: 2e624b79d5c04874f65c0366e9462796b80c262b --- src/opr/impl/tensor_manip.cpp | 5 +---- src/opr/include/megbrain/opr/tensor_manip.h | 1 - 2 files changed, 1 insertion(+), 5 deletions(-) diff --git a/src/opr/impl/tensor_manip.cpp b/src/opr/impl/tensor_manip.cpp index f33f6f92..f493e241 100644 --- a/src/opr/impl/tensor_manip.cpp +++ b/src/opr/impl/tensor_manip.cpp @@ -1443,7 +1443,6 @@ ParamPackSplit::ParamPackSplit(VarNode* src, VarNode* offsets, mgb_assert(src->comp_node() == offsets->comp_node()); add_input({src}); add_input({offsets}); - m_mem_fwd_success.resize(m_shapes.size()); for (size_t i = 0; i < shapes.size(); i++) { mgb_assert(shapes[i].total_nr_elems(), "empty param is not allowed!"); @@ -1485,9 +1484,7 @@ void ParamPackSplit::mem_plan_fwd_in2out_readonly() { for (size_t i = 0; i < output().size(); i++) { auto layout = output(i)->layout(); auto spec = SubTensorSpec::make_from_offset_elem(layout, m_offsets[i * 2]); - m_mem_fwd_success[i] = output(i)->set_fwd_in2out_readonly( - input(0), spec); - mgb_assert(m_mem_fwd_success[i]); + mgb_assert(output(i)->set_fwd_in2out_readonly(input(0), spec)); } } diff --git a/src/opr/include/megbrain/opr/tensor_manip.h b/src/opr/include/megbrain/opr/tensor_manip.h index c9d1cb39..2bc23ff3 100644 --- a/src/opr/include/megbrain/opr/tensor_manip.h +++ b/src/opr/include/megbrain/opr/tensor_manip.h @@ -590,7 +590,6 @@ public: MGB_DEFINE_OPR_CLASS(ParamPackSplit, cg::SingleCNOperatorNodeBase) // { TensorShapeArray m_shapes; std::vector m_offsets; - std::vector m_mem_fwd_success; void scn_do_execute() override{}; void init_output_static_infer_desc() override; -- GitLab