aboutsummaryrefslogtreecommitdiffhomepage
path: root/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h
diff options
context:
space:
mode:
authorGravatar Eugene Zhulenev <ezhulenev@google.com>2019-01-11 11:41:53 -0800
committerGravatar Eugene Zhulenev <ezhulenev@google.com>2019-01-11 11:41:53 -0800
commit1e6d15b55b67d6170d4548fa7f20acbb327814bf (patch)
treec969a3fc7fb79f00f82baa435bbd7ca0aecfc979 /unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h
parentdf29511ac0486639e23fe65c7edafecc2d9f1579 (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.h3
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);