Fix shorten-64-to-32 warning in TensorContractionThreadPool

This commit is contained in:
Eugene Zhulenev 2019-01-10 10:27:55 -08:00
parent d812f411c3
commit 0abe03764c

View File

@ -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;