Browse Source

feat(mge/imperative): add thread name

GitOrigin-RevId: 61eece55b8
release-1.2
Megvii Engine Team 4 years ago
parent
commit
93f4977c78
3 changed files with 13 additions and 1 deletions
  1. +4
    -0
      imperative/python/src/helper.h
  2. +6
    -1
      imperative/src/impl/interpreter_impl.h
  3. +3
    -0
      imperative/src/impl/physical_tensor.cpp

+ 4
- 0
imperative/python/src/helper.h View File

@@ -69,6 +69,10 @@ class PyTaskDipatcher {
pybind11::gil_scoped_acquire _;
f();
}

void on_async_queue_worker_thread_start() override {
mgb::sys::set_thread_name("py_task_worker");
}
};
Queue queue;
bool finalized = false;


+ 6
- 1
imperative/src/impl/interpreter_impl.h View File

@@ -143,10 +143,15 @@ private:
size_t m_enable_evict = 0;

struct WorkQueue : AsyncQueueSC<Command, WorkQueue> {
WorkQueue(ChannelImpl* owner) : m_owner(owner) {}
WorkQueue(ChannelImpl* owner) : m_owner(owner) {
sys::set_thread_name("interpreter");
}
void process_one_task(Command& cmd) {
m_owner->process_one_task(cmd);
}
void on_async_queue_worker_thread_start() override {
sys::set_thread_name("worker");
}
private:
ChannelImpl* m_owner;
} m_worker;


+ 3
- 0
imperative/src/impl/physical_tensor.cpp View File

@@ -44,6 +44,9 @@ class AsyncReleaser : public CompNodeDepedentObject {
std::this_thread::sleep_for(1us);
add_task(std::move(param));
}
void on_async_queue_worker_thread_start() override {
sys::set_thread_name("releaser");
}
};
Waiter m_waiter{this};



Loading…
Cancel
Save