Use int64_t instead of int for in loop that may overflow. (pytorch#8435)

runtian-zhou · Jun 13, 2018 · 4d287f9 · 4d287f9
1 parent 2c9c48a
commit 4d287f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aten/src/ATen/native/cpu/ReduceOpsKernel.cpp b/aten/src/ATen/native/cpu/ReduceOpsKernel.cpp
@@ -102,7 +102,7 @@ struct Reduction {
       sum = std::accumulate(buf, buf + WIDTH, scalar_t(ident), ReduceScalar());
     }
 
-    for (int i = k * WIDTH; i != size; i++) {
+    for (int64_t i = k * WIDTH; i != size; i++) {
       sum = ReduceScalar()(sum, data[i]);
     }
     return sum;