-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Conversation
@@ -121,6 +121,19 @@ double EvaluateWorloads(const std::vector<Workload>& workloads, | |||
return dmlc::GetTime() - t; | |||
} | |||
|
|||
TEST(Engine, start_stop) { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please add documentation
|
||
for (int i = 0; i < num_engine; ++i) { | ||
engine[i]->Stop(); | ||
engine[i]->Start(); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Could you verify whether the actions have actually been successful?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Without adding new API, I cannot think of a way to verify except sanity test.
@@ -0,0 +1,44 @@ | |||
# Licensed to the Apache Software Foundation (ASF) under one |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please don't create a new test directory.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Done.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM pending on other reviews being sufficiently addressed.
@piiswrong Can you have a look at the engine part to see if it's correct? |
* fix engine start/stop * add tests * fix test * fix * fix tests
* fix engine start/stop * add tests * fix test * fix * fix tests
streams_.reset(new StreamManager<kMaxNumGpus, kNumStreamsPerGpu>()); | ||
task_queue_.reset(new dmlc::ConcurrentBlockingQueue<OprBlock*>()); | ||
io_task_queue_.reset(new dmlc::ConcurrentBlockingQueue<OprBlock*>()); | ||
thread_pool_.reset(new ThreadPool(kNumWorkingThreads, [this]() { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Shouldn't this contain change to another constructor of ThreadPool? Only the one accepting std::shared_ptr<dmlc::ManualEvent> ready_event
will wait until ready, which might be the issue fixed in #8995.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Fixed in #11065
* fix engine start/stop * add tests * fix test * fix * fix tests
Description
Continued fixes as in #10820.
Checklist
Essentials
Please feel free to remove inapplicable items for your PR.
Changes
Comments