提交 8624ec22 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: f76a096832ae033eb3d69dbc69360ae1a6b1bb4d
上级 099d1b87
......@@ -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(),
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册