未验证 提交 82ce2d56 编写于 作者: A Allen Guo 提交者: GitHub

update new feature of mlperf (#735) (#42706)

* update new feature of mlperf
Co-authored-by: Nyaozhixin <522190855@qq.com>
上级 cde2b24d
......@@ -215,7 +215,12 @@ void Executor::Run(const std::vector<const Tensor *> &inputs,
popart::StepIO stepio(popart_inputs, popart_anchors);
VLOG(10) << "Running...";
session_->run(stepio);
if (ipu_strategy_->popart_options.createImplicitPipeliningFwdOnlyProgram &&
ipu_strategy_->runtime_options.enable_eval) {
session_->run("implicitPipeliningFwdOnly", stepio);
} else {
session_->run(stepio);
}
VLOG(10) << "Running...done";
}
......
......@@ -285,6 +285,8 @@ IpuStrategy::IpuStrategy() {
ADD_POPART_BOOL_OPTION_ALIAS(
schedule_non_weight_update_gradient_consumers_early,
scheduleNonWeightUpdateGradientConsumersEarly);
ADD_POPART_BOOL_OPTION_ALIAS(create_implicit_pipelining_fwd_only_program,
createImplicitPipeliningFwdOnlyProgram);
ADD_POPART_DOUBLE_OPTION_ALIAS(outline_sequence_break_cost,
outlineSequenceBreakCost);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册