5282: fix int overflow in the warning

apache · Jan 17, 2015 · 7afac23 · 7afac23
1 parent f38ef65
commit 7afac23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/RowMatrix.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/distributed/RowMatrix.scala
@@ -131,7 +131,7 @@ class RowMatrix(
       throw new IllegalArgumentException(s"Argument with more than 65535 cols: $cols")
     }
     if (cols > 10000) {
-      val mem = cols * cols * 8
+      val mem = (math.pow(cols, 2) * 8).formatted("%,.0f")
       logWarning(s"$cols columns will require at least $mem bytes of memory!")
     }
   }