diff options
Diffstat (limited to 'caffe2/utils')
| -rw-r--r-- | caffe2/utils/thread_pool.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/caffe2/utils/thread_pool.h b/caffe2/utils/thread_pool.h index cd8ba2c49a..7047dc3d43 100644 --- a/caffe2/utils/thread_pool.h +++ b/caffe2/utils/thread_pool.h @@ -9,7 +9,9 @@ namespace caffe2 { class CAFFE2_API TaskThreadPool : public c10::ThreadPool { public: - explicit TaskThreadPool(std::size_t pool_size, int numa_node_id = -1) + explicit TaskThreadPool( + std::size_t pool_size, + int numa_node_id = -1) : ThreadPool(pool_size, numa_node_id) {} // TODO move this to ATen/core/thread_pool.h |