| 24 | /* ===================== dv_helper ===================== */ |
| 25 | |
| 26 | void dv_helper::add_output( |
| 27 | cg::OperatorNodeBase& opr, DType dtype, const Maybe<std::string>& name) { |
| 28 | mgb_assert(dtype.valid()); |
| 29 | opr.add_output(name) |
| 30 | ->add_flag(VarNode::Flag::NO_MEM_RECLAIM) |
| 31 | .add_flag(VarNode::Flag::PERSISTENT_DEVICE_VALUE) |
| 32 | .add_flag(VarNode::Flag::DISALLOW_RT_FORCE_DYNAMIC_MEM_ALLOC) |
| 33 | .dtype(dtype); |
| 34 | } |
| 35 | |
| 36 | void dv_helper::init_output_mem_plan( |
| 37 | const DeviceTensorND& val, cg::OperatorNodeBase& opr, bool dynamic, |