提交 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,
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));
}
}
......
......@@ -590,7 +590,6 @@ public:
MGB_DEFINE_OPR_CLASS(ParamPackSplit, cg::SingleCNOperatorNodeBase) // {
TensorShapeArray m_shapes;
std::vector<dt_int32> m_offsets;
std::vector<bool> m_mem_fwd_success;
void scn_do_execute() 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.
先完成此消息的编辑!
想要评论请 注册