From b0354f616f7f49ee9b19f6b8e5d0dc775b05dba2 Mon Sep 17 00:00:00 2001
From: Burak Yavuz <brkyvz@gmail.com>
Date: Tue, 23 Dec 2014 23:33:51 -0800
Subject: [PATCH] [SPARK-4409] Incorporated mengxr's code

---
 .../src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala
index 733112c1fcc86..5a7281ec6dc3c 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/linalg/Matrices.scala
@@ -448,11 +448,11 @@ object SparseMatrix {
       // selection-rejection method
       var idx = 0L
       var numSelected = 0
-      var i = 0
       var j = 0
       val colPtrs = new Array[Int](numCols + 1)
       val rowIndices = new Array[Int](nnz)
       while (j < numCols && numSelected < nnz) {
+        var i = 0
         while (i < numRows && numSelected < nnz) {
           if (rng.nextDouble() < 1.0 * (nnz - numSelected) / (size - idx)) {
             rowIndices(numSelected) = i