Browse Source

fix(mge): disable dedup for io callback oprs

GitOrigin-RevId: 424d347d83
tags/v1.0.0-rc1
Megvii Engine Team 5 years ago
parent
commit
517c1ec007
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      imperative/src/impl/opr_utility.cpp

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

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


Loading…
Cancel
Save