diff options
| author |  Rasmus Larsen <rmlarsen@google.com> | 2019-01-11 10:20:55 -0800 |
|---|---|---|
| committer | Rasmus Larsen <rmlarsen@google.com> | 2019-01-11 10:20:55 -0800 |
| commit | 74882471d0012596e78bffa7ad4e74d940be5059 (patch) | |
| tree | a9284bcb78b905128afbdca84a701272bbabb80d /unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h | |
| parent | e15bb785adf756f3e48410ee681ca97ad5bb3e76 (diff) | |
| parent | 3c9add6598cc35e5317788627dfa81f517e89e07 (diff) | |
Merged eigen/eigen into default
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 9666bf167..d68409e26 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h @@ -890,7 +890,8 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT TENSOR_CONTRACTION_DISPATCH( this->template evalGemmPartialWithoutOutputKernel, Alignment, - (buf, begin, end, /*num_threads=*/num_blocks)); + (buf, begin, end, + /*num_threads=*/internal::convert_index<int>(num_blocks))); // Check if it was the last task in l0 range. const Index l0_index = block_idx / l0_size; |