devtools/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2026-04-10 11:34:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add a ThreadPoolInterface* getter for ThreadPoolDevice.

Browse Source
This commit is contained in:
Penporn Koanantakool
2018-06-02 12:07:49 -07:00
parent 84868da904
commit e2ed0cf8ab
1 changed files with 4 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h
View File

@@ -169,7 +169,7 @@ struct ThreadPoolDevice {
// parallelFor executes f with [0, n) arguments in parallel and waits for // parallelFor executes f with [0, n) arguments in parallel and waits for
// completion. F accepts a half-open interval [first, last). // 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 // efficiency. If block_align is not nullptr, it is called to round up the
// block size. // block size.
void parallelFor(Index n, const TensorOpCost& cost, void parallelFor(Index n, const TensorOpCost& cost,
@@ -261,6 +261,9 @@ struct ThreadPoolDevice {
parallelFor(n, cost, nullptr, std::move(f)); parallelFor(n, cost, nullptr, std::move(f));
} }
// Thread pool accessor.
ThreadPoolInterface* getPool() const { return pool_; }
private: private:
ThreadPoolInterface* pool_; ThreadPoolInterface* pool_;
int num_threads_; int num_threads_;