Browse Source

fix(mgb): fix param merge bug that caused the weight statistics error

GitOrigin-RevId: f76a096832
tags/v1.8.0
Megvii Engine Team 3 years ago
parent
commit
8624ec224b
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      src/core/impl/graph/cg_impl_seq.cpp

+ 2
- 1
src/core/impl/graph/cg_impl_seq.cpp View File

@@ -539,7 +539,8 @@ void ComputingGraphImpl::ComputingSequence::do_regist() const {
auto& mc = mp.chunk();
if (mp.valid() && mc.mem_alloc_status.is_from_owner_var()) {
auto size = mgb::get_aligned_power2(
mc.size(), j->comp_node().get_mem_addr_alignment());
mp.layout().span().dist_byte(),
j->comp_node().get_mem_addr_alignment());

recorder.regist_memory_chunk(
{chunk_id++, size, 0, this->m_opr_seq->size(),


Loading…
Cancel
Save