diff options
author | Eugene Zhulenev <ezhulenev@google.com> | 2019-01-11 11:41:53 -0800 |
---|---|---|
committer | Eugene Zhulenev <ezhulenev@google.com> | 2019-01-11 11:41:53 -0800 |
commit | 1e6d15b55b67d6170d4548fa7f20acbb327814bf (patch) | |
tree | c969a3fc7fb79f00f82baa435bbd7ca0aecfc979 /unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h | |
parent | df29511ac0486639e23fe65c7edafecc2d9f1579 (diff) |
Fix shorten-64-to-32 warning in TensorContractionThreadPool
Diffstat (limited to 'unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h')
-rw-r--r-- | unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h index d68409e26..e06099957 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h @@ -879,7 +879,8 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT // Keep count of pending gemm tasks for each l0 range. MaxSizeVector<std::atomic<int>> l0_state(l0_ranges); for (int i = 0; i < l0_ranges; ++i) { - l0_state.emplace_back(actual_range_size(l0_ranges, l0_size, i)); + const Index num_pending_tasks = actual_range_size(l0_ranges, l0_size, i); + l0_state.emplace_back(internal::convert_index<int>(num_pending_tasks)); } MaxSizeVector<Scalar*> block_buffers(num_blocks); |