GitHub-Proxy/eigen
1
0
Fork 0
mirror of https://gitlab.com/libeigen/eigen.git synced 2025-08-12 11:49:02 +08:00
Code Issues Packages Projects Releases Wiki Activity

Remove shadow warnings in TensorDeviceThreadPool

Browse Source
This commit is contained in:
Eugene Zhulenev 2019-08-28 10:32:19 -07:00
parent 9aba527405
commit 6e77f9bef3
1 changed files with 6 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -90,7 +90,6 @@ struct ThreadPoolDevice {
// CPU cycles due to the threads competing for memory bandwidth, so we // CPU cycles due to the threads competing for memory bandwidth, so we
// statically schedule at most 4 block copies here. // statically schedule at most 4 block copies here.
const size_t kMinBlockSize = 32768; const size_t kMinBlockSize = 32768;
typedef TensorCostModel<ThreadPoolDevice> CostModel;
const size_t num_threads = CostModel::numThreads(n, TensorOpCost(1.0, 1.0, 0), 4); const size_t num_threads = CostModel::numThreads(n, TensorOpCost(1.0, 1.0, 0), 4);
if (n <= kMinBlockSize || num_threads < 2) { if (n <= kMinBlockSize || num_threads < 2) {
::memcpy(dst, src, n); ::memcpy(dst, src, n);
@ -302,9 +301,12 @@ struct ThreadPoolDevice {
// For parallelForAsync we must keep passed in closures on the heap, and // For parallelForAsync we must keep passed in closures on the heap, and
// delete them only after `done` callback finished. // delete them only after `done` callback finished.
struct ParallelForAsyncContext { struct ParallelForAsyncContext {
ParallelForAsyncContext(Index count, std::function<void(Index, Index)> f, ParallelForAsyncContext(Index block_count,
std::function<void()> done) std::function<void(Index, Index)> block_f,
: count(count), f(std::move(f)), done(std::move(done)) {} std::function<void()> done_callback)
: count(block_count),
f(std::move(block_f)),
done(std::move(done_callback)) {}
std::atomic<Index> count; std::atomic<Index> count;
std::function<void(Index, Index)> f; std::function<void(Index, Index)> f;