| 38 | } |
| 39 | |
| 40 | void ThreadPool::AddWork(const std::function<void()>& work) { |
| 41 | const size_t cur_chan_idx = |
| 42 | work_cnt_.fetch_add(1, std::memory_order_relaxed) % work_chans_.size(); |
| 43 | work_chans_.at(cur_chan_idx).Send(work); |
| 44 | } |
| 45 | |
| 46 | } // namespace oneflow |