未验证 提交 5babcd02 编写于 作者: T Tao Luo 提交者: GitHub

Revert "remove unnecessary prepare_data (#17080)" (#17432)

This reverts commit aca60e9a.
上级 e336dc86
......@@ -1023,7 +1023,6 @@ Scope* OperatorWithKernel::PrepareData(
std::vector<std::string>* transfered_inplace_vars,
RuntimeContext* ctx) const {
Scope* new_scope = nullptr;
if (!need_prepare_data_) return new_scope;
std::unordered_set<std::string> no_buffer_ins;
if (info_) {
......@@ -1116,10 +1115,6 @@ Scope* OperatorWithKernel::PrepareData(
SetTensorToVariable(*var, out, trans_var);
}
}
// If new_scope = nullptr, it means that for each input of this Op, there is
// no TransformData. Thus, PrepareData could be skipped at the rest iterations
// of this Op's execution to save the elapsed time.
if (!new_scope) need_prepare_data_ = false;
return new_scope;
}
......
......@@ -507,7 +507,6 @@ class OperatorWithKernel : public OperatorBase {
mutable std::unique_ptr<OpKernelFunc> kernel_func_;
mutable std::unique_ptr<RuntimeContext> runtime_ctx_;
mutable const Scope* pre_scope_ = nullptr;
mutable bool need_prepare_data_ = true;
mutable bool enable_cache_runtime_context = false;
mutable bool enable_cache_expected_kernel = false;
mutable bool all_kernels_must_compute_runtime_shape = false;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册