From e2ed0cf8abfe275bd147f3228e4c59f5242993f6 Mon Sep 17 00:00:00 2001 From: Penporn Koanantakool Date: Sat, 2 Jun 2018 12:07:49 -0700 Subject: Add a ThreadPoolInterface* getter for ThreadPoolDevice. --- unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h') diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h b/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h index ec6802e85..ca9ba402e 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h @@ -169,7 +169,7 @@ struct ThreadPoolDevice { // parallelFor executes f with [0, n) arguments in parallel and waits for // completion. F accepts a half-open interval [first, last). - // Block size is choosen based on the iteration cost and resulting parallel + // Block size is chosen based on the iteration cost and resulting parallel // efficiency. If block_align is not nullptr, it is called to round up the // block size. void parallelFor(Index n, const TensorOpCost& cost, @@ -261,6 +261,9 @@ struct ThreadPoolDevice { parallelFor(n, cost, nullptr, std::move(f)); } + // Thread pool accessor. + ThreadPoolInterface* getPool() const { return pool_; } + private: ThreadPoolInterface* pool_; int num_threads_; -- cgit v1.2.3