aboutsummaryrefslogtreecommitdiff
path: root/graph
diff options
context:
space:
mode:
authorAnkur Dave <ankurdave@gmail.com>2013-10-16 23:56:15 -0700
committerAnkur Dave <ankurdave@gmail.com>2013-10-16 23:56:15 -0700
commit2282d27cf1da04bfdad5173c412e77a51d7c36e6 (patch)
treef287646a1225a7ff5dd41cd8ee9487059e2ad07b /graph
parentbc234bf0e1ecc5c9bba2cd629644780c9d5c3511 (diff)
downloadspark-2282d27cf1da04bfdad5173c412e77a51d7c36e6.tar.gz
spark-2282d27cf1da04bfdad5173c412e77a51d7c36e6.tar.bz2
spark-2282d27cf1da04bfdad5173c412e77a51d7c36e6.zip
Cache msgsByPartition
Diffstat (limited to 'graph')
-rw-r--r--graph/src/main/scala/org/apache/spark/graph/impl/GraphImpl.scala3
1 files changed, 2 insertions, 1 deletions
diff --git a/graph/src/main/scala/org/apache/spark/graph/impl/GraphImpl.scala b/graph/src/main/scala/org/apache/spark/graph/impl/GraphImpl.scala
index 5398631d26..a3cdb97cea 100644
--- a/graph/src/main/scala/org/apache/spark/graph/impl/GraphImpl.scala
+++ b/graph/src/main/scala/org/apache/spark/graph/impl/GraphImpl.scala
@@ -486,7 +486,8 @@ object GraphImpl {
.flatMap { case (vid, (vdata, pids)) =>
pids.iterator.map { pid => MessageToPartition(pid, (vid, vdata)) }
}
- .partitionBy(eTable.partitioner.get) //@todo assert edge table has partitioner
+ .partitionBy(eTable.partitioner.get).cache()
+ // @todo assert edge table has partitioner
val vTableReplicationMap: IndexedRDD[Pid, VertexIdToIndexMap] =
msgsByPartition.mapPartitionsWithIndex( (pid, iter) => {