提交 11388e58 编写于 作者: M Megvii Engine Team 提交者: Xinran Xu

refactor(mgb/parampacksplit): remove m_mem_plan_fwd

GitOrigin-RevId: 2e624b79d5c04874f65c0366e9462796b80c262b
上级 d929dd0a
...@@ -1443,7 +1443,6 @@ ParamPackSplit::ParamPackSplit(VarNode* src, VarNode* offsets, ...@@ -1443,7 +1443,6 @@ ParamPackSplit::ParamPackSplit(VarNode* src, VarNode* offsets,
mgb_assert(src->comp_node() == offsets->comp_node()); mgb_assert(src->comp_node() == offsets->comp_node());
add_input({src}); add_input({src});
add_input({offsets}); add_input({offsets});
m_mem_fwd_success.resize(m_shapes.size());
for (size_t i = 0; i < shapes.size(); i++) { for (size_t i = 0; i < shapes.size(); i++) {
mgb_assert(shapes[i].total_nr_elems(), "empty param is not allowed!"); mgb_assert(shapes[i].total_nr_elems(), "empty param is not allowed!");
...@@ -1485,9 +1484,7 @@ void ParamPackSplit::mem_plan_fwd_in2out_readonly() { ...@@ -1485,9 +1484,7 @@ void ParamPackSplit::mem_plan_fwd_in2out_readonly() {
for (size_t i = 0; i < output().size(); i++) { for (size_t i = 0; i < output().size(); i++) {
auto layout = output(i)->layout(); auto layout = output(i)->layout();
auto spec = SubTensorSpec::make_from_offset_elem(layout, m_offsets[i * 2]); auto spec = SubTensorSpec::make_from_offset_elem(layout, m_offsets[i * 2]);
m_mem_fwd_success[i] = output(i)->set_fwd_in2out_readonly( mgb_assert(output(i)->set_fwd_in2out_readonly(input(0), spec));
input(0), spec);
mgb_assert(m_mem_fwd_success[i]);
} }
} }
......
...@@ -590,7 +590,6 @@ public: ...@@ -590,7 +590,6 @@ public:
MGB_DEFINE_OPR_CLASS(ParamPackSplit, cg::SingleCNOperatorNodeBase) // { MGB_DEFINE_OPR_CLASS(ParamPackSplit, cg::SingleCNOperatorNodeBase) // {
TensorShapeArray m_shapes; TensorShapeArray m_shapes;
std::vector<dt_int32> m_offsets; std::vector<dt_int32> m_offsets;
std::vector<bool> m_mem_fwd_success;
void scn_do_execute() override{}; void scn_do_execute() override{};
void init_output_static_infer_desc() override; void init_output_static_infer_desc() override;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册