Browse Source

refactor(mgb/parampacksplit): remove m_mem_plan_fwd

GitOrigin-RevId: 2e624b79d5
tags/v0.4.0
Megvii Engine Team Xinran Xu 5 years ago
parent
commit
11388e5888
2 changed files with 1 additions and 5 deletions
  1. +1
    -4
      src/opr/impl/tensor_manip.cpp
  2. +0
    -1
      src/opr/include/megbrain/opr/tensor_manip.h

+ 1
- 4
src/opr/impl/tensor_manip.cpp View File

@@ -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));
}
}



+ 0
- 1
src/opr/include/megbrain/opr/tensor_manip.h View File

@@ -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;


Loading…
Cancel
Save