diff --git a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h index 4553c3785..3dc95621c 100644 --- a/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h +++ b/unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h @@ -791,7 +791,7 @@ struct TensorEvaluator(k, 8 * num_threads); - int num_blocks = divup(k, block_size); + int num_blocks = internal::convert_index(divup(k, block_size)); // we use 'result' for the first block's partial result. MaxSizeVector block_buffers(num_blocks - 1); Barrier barrier(num_blocks); @@ -855,7 +855,7 @@ struct TensorEvaluator::totalCost( m * n, TensorOpCost(2, 1, 1, true, output_packet_size)); - Index num_threads = 1; + int num_threads = 1; double min_cost = total_parallel_cost; double kPerThreadOverHead = 4000; double kFixedOverHead = 100000;