change kb to mb

maropu · May 6, 2015 · d91e5ed · d91e5ed
1 parent c688e3c
commit d91e5ed
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/core/src/main/scala/org/apache/spark/serializer/KryoSerializer.scala b/core/src/main/scala/org/apache/spark/serializer/KryoSerializer.scala
@@ -51,9 +51,9 @@ class KryoSerializer(conf: SparkConf)
 
   private val bufferSizeKb = conf.getSizeAsKb("spark.kryoserializer.buffer", "64k")
 
-  if (bufferSizeKb >= 2048) {
+  if (bufferSizeKb >= 2048 * 1024) {
     throw new IllegalArgumentException("spark.kryoserializer.buffer must be less than " +
-      s"2048 mb, got: + $bufferSizeKb mb.")
+      s"2048 mb, got: + ${bufferSizeKb/1024} mb.")
   }
   private val bufferSize = (bufferSizeKb * 1024).toInt