Add CurrentThreadId and NumThreads methods to Eigen threadpools and TensorDeviceThreadPool.

1 files changed, 4 insertions, 0 deletions

diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h b/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h
index d31b0ad38..90fded8ad 100644
--- a/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h
+++ b/unsupported/Eigen/CXX11/src/Tensor/TensorDeviceThreadPool.h
@@ -172,6 +172,10 @@ struct ThreadPoolDevice {
     pool_->Schedule(func);
   }
 
+  EIGEN_STRONG_INLINE size_t currentThreadId() const {
+    return pool_->CurrentThreadId();
+  }
+
   // 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