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

Fix integer conversion warnings

Browse Source
This commit is contained in:
Christoph Hertzberg 2018-09-28 19:33:39 +02:00
parent 2c083ace3e
commit 86ba50be39
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
unsupported/Eigen/CXX11/src/Tensor/TensorContractionThreadPool.h
View File

@ -791,7 +791,7 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT
// The underlying GEMM kernel assumes that k is a multiple of 8 and // The underlying GEMM kernel assumes that k is a multiple of 8 and
// subtle breakage occurs if this is violated. // subtle breakage occurs if this is violated.
Index block_size = 8 * divup<Index>(k, 8 * num_threads); Index block_size = 8 * divup<Index>(k, 8 * num_threads);
int num_blocks = divup<Index>(k, block_size); int num_blocks = internal::convert_index<int>(divup<Index>(k, block_size));
// we use 'result' for the first block's partial result. // we use 'result' for the first block's partial result.
MaxSizeVector<Scalar*> block_buffers(num_blocks - 1); MaxSizeVector<Scalar*> block_buffers(num_blocks - 1);
Barrier barrier(num_blocks); Barrier barrier(num_blocks);
@ -855,7 +855,7 @@ struct TensorEvaluator<const TensorContractionOp<Indices, LeftArgType, RightArgT
// result. // result.
double reduction_cost = TensorCostModel<ThreadPoolDevice>::totalCost( double reduction_cost = TensorCostModel<ThreadPoolDevice>::totalCost(
m * n, TensorOpCost(2, 1, 1, true, output_packet_size)); 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 min_cost = total_parallel_cost;
double kPerThreadOverHead = 4000; double kPerThreadOverHead = 4000;
double kFixedOverHead = 100000; double kFixedOverHead = 100000;