未验证 提交 5f30e57c 编写于 作者: L lilong12 提交者: GitHub

fix test_pipeline, test=develop (#25808)

* fix test_pipeline, test=develop
上级 d47304e6
......@@ -335,6 +335,7 @@ class SectionWorker : public DeviceWorker {
void SetSkipVars(const std::vector<std::string>& skip_vars) {
skip_vars_ = skip_vars;
}
static void ResetBatchId() { batch_id_ = 0; }
static std::atomic<int> cpu_id_;
......
......@@ -250,6 +250,7 @@ void PipelineTrainer::Finalize() {
}
}
root_scope_->DropKids();
SectionWorker::ResetBatchId();
}
Scope* PipelineTrainer::GetWorkerScope(int thread_id) {
......
......@@ -248,7 +248,6 @@ list(REMOVE_ITEM TEST_OPS test_fuse_bn_act_pass)
list(REMOVE_ITEM TEST_OPS test_imperative_static_runner_mnist)
list(REMOVE_ITEM TEST_OPS test_imperative_static_runner_while)
list(REMOVE_ITEM TEST_OPS test_conv3d_transpose_op)
list(REMOVE_ITEM TEST_OPS test_pipeline)
# disable this unittest temporarily
list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册