Copy data only when necessary

markhamstra · May 6, 2014 · 64e7198 · 64e7198
1 parent b85806c
commit 64e7198
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Vectors.scala
@@ -136,7 +136,12 @@ object Vectors {
           new DenseVector(v.toArray)  // Can't use underlying array directly, so make a new one
         }
       case v: BSV[Double] =>
-        new SparseVector(v.length, v.index.slice(0, v.used), v.data.slice(0, v.used))
+        if (v.index.length == v.used) {
+          new SparseVector(v.length, v.index, v.data)
+        }
+        else {
+          new SparseVector(v.length, v.index.slice(0, v.used), v.data.slice(0, v.used))
+        }
       case v: BV[_] =>
         sys.error("Unsupported Breeze vector type: " + v.getClass.getName)
     }