fix(mge): disable dedup for io callback oprs

GitOrigin-RevId: 424d347d83220a56547d643f2daf461124b1ead9

fix(mge): disable dedup for io callback oprs
GitOrigin-RevId: 424d347d83220a56547d643f2daf461124b1ead9
517c1ec0 · Megvii Engine Team · 939bf3b2 · 517c1ec0
显示空白变更内容
内联并排

Showing with 2 addition and 0 deletion

imperative/src/impl/opr_utility.cpp imperative/src/impl/opr_utility.cpp +2 -0

未找到文件。
--- a/imperative/src/impl/opr_utility.cpp
+++ b/imperative/src/impl/opr_utility.cpp
@@ -46,6 +46,7 @@ InputCallback::InputCallback(cg::ComputingGraph& graph, callback_t callback,
            ->add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE)
            .add_flag(VarNode::Flag::NO_SYS_MEM_ALLOC)
            .dtype(DType::from_enum(DTypeEnum::Byte));
+    add_equivalence_component<ScalarHash<void*>>(this);
 }

 SymbolVarArray InputCallback::make(cg::ComputingGraph& graph,
@@ -178,6 +179,7 @@ NopCallback::NopCallback(cg::ComputingGraph& graph, callback_t callback,
            ->add_flag(VarNode::Flag::ALLOW_EMPTY_SHAPE)
            .add_flag(VarNode::Flag::NO_SYS_MEM_ALLOC)
            .dtype(DType::from_enum(DTypeEnum::Byte));
+    add_equivalence_component<ScalarHash<void*>>(this);
 }

 SymbolVar NopCallback::make(cg::ComputingGraph& graph, callback_t callback,