Prevent potential overflow.

This commit is contained in:
Benoit Steiner 2016-03-28 09:18:04 -07:00
parent 74f91ed06c
commit 78f83d6f6a

View File

@ -132,7 +132,7 @@ class TensorExecutor<Expression, ThreadPoolDevice, Vectorizable>
device.enqueue_with_barrier(&barrier, &EvalRange<Evaluator, Index, Vectorizable>::run, evaluator, i*blocksize, (i+1)*blocksize); device.enqueue_with_barrier(&barrier, &EvalRange<Evaluator, Index, Vectorizable>::run, evaluator, i*blocksize, (i+1)*blocksize);
} }
if (numblocks * blocksize < size) { if (static_cast<Index>(numblocks) * blocksize < size) {
EvalRange<Evaluator, Index, Vectorizable>::run(evaluator, numblocks * blocksize, size); EvalRange<Evaluator, Index, Vectorizable>::run(evaluator, numblocks * blocksize, size);
} }