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

fix test_pipeline, test=develop (#25808)

* fix test_pipeline, test=develop
上级 d47304e6
...@@ -335,6 +335,7 @@ class SectionWorker : public DeviceWorker { ...@@ -335,6 +335,7 @@ class SectionWorker : public DeviceWorker {
void SetSkipVars(const std::vector<std::string>& skip_vars) { void SetSkipVars(const std::vector<std::string>& skip_vars) {
skip_vars_ = skip_vars; skip_vars_ = skip_vars;
} }
static void ResetBatchId() { batch_id_ = 0; }
static std::atomic<int> cpu_id_; static std::atomic<int> cpu_id_;
......
...@@ -250,6 +250,7 @@ void PipelineTrainer::Finalize() { ...@@ -250,6 +250,7 @@ void PipelineTrainer::Finalize() {
} }
} }
root_scope_->DropKids(); root_scope_->DropKids();
SectionWorker::ResetBatchId();
} }
Scope* PipelineTrainer::GetWorkerScope(int thread_id) { Scope* PipelineTrainer::GetWorkerScope(int thread_id) {
......
...@@ -248,7 +248,6 @@ list(REMOVE_ITEM TEST_OPS test_fuse_bn_act_pass) ...@@ -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_mnist)
list(REMOVE_ITEM TEST_OPS test_imperative_static_runner_while) 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_conv3d_transpose_op)
list(REMOVE_ITEM TEST_OPS test_pipeline)
# disable this unittest temporarily # disable this unittest temporarily
list(REMOVE_ITEM TEST_OPS test_imperative_data_loader_exception) 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.
先完成此消息的编辑!
想要评论请 注册