From bf94b4dbcc4e8e0602715dce92f5053608674b43 Mon Sep 17 00:00:00 2001
From: ding <ding@localhost.localdomain>
Date: Fri, 16 Sep 2016 13:04:27 -0400
Subject: [PATCH] persist edges if their storage level is non in
 PeriodicGraphCheckpointer

---
 .../apache/spark/mllib/impl/PeriodicGraphCheckpointer.scala  | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/mllib/src/main/scala/org/apache/spark/mllib/impl/PeriodicGraphCheckpointer.scala b/mllib/src/main/scala/org/apache/spark/mllib/impl/PeriodicGraphCheckpointer.scala
index 20db6084d0e0d..80074897567eb 100644
--- a/mllib/src/main/scala/org/apache/spark/mllib/impl/PeriodicGraphCheckpointer.scala
+++ b/mllib/src/main/scala/org/apache/spark/mllib/impl/PeriodicGraphCheckpointer.scala
@@ -87,7 +87,10 @@ private[mllib] class PeriodicGraphCheckpointer[VD, ED](
 
   override protected def persist(data: Graph[VD, ED]): Unit = {
     if (data.vertices.getStorageLevel == StorageLevel.NONE) {
-      data.persist()
+      data.vertices.persist()
+    }
+    if (data.edges.getStorageLevel == StorageLevel.NONE) {
+      data.edges.persist()
     }
   }