fix wcc_opt bugs in hash partition (alibaba#88)

Author: mengke-mk

examples/analytical_apps/cuda/wcc/wcc_opt.h

@@ -58,10 +58,10 @@ class WCCOptContext : public grape::VoidContext<FRAG_T> {
     if (frag.fid() == 0) {
       messages.InitBuffer(
           0, (frag.GetTotalVerticesNum() - frag.InnerVertices().size()) *
-                 sizeof(offset_t) * 2 * 1.5);
+                 sizeof(offset_t) * 2 * 2);
     } else {
-      messages.InitBuffer(
-          frag.InnerVertices().size() * sizeof(offset_t) * 2 * 1.5, 0);
+      messages.InitBuffer(frag.Vertices().size() * sizeof(offset_t) * 2 * 1.5,
+                          0);
     }
   }
 
@@ -118,6 +118,7 @@ class WCCOpt : public GPUAppBase<FRAG_T, WCCOptContext<FRAG_T>>,
     auto d_coo = ctx.coo_frag->DeviceObject();
     auto vertices = frag.Vertices();
     auto iv = frag.InnerVertices();
+    auto av = frag.Vertices();
     auto dev_mm = messages.DeviceObject();
     auto* d_parents = thrust::raw_pointer_cast(ctx.parents.data());
     auto& stream = messages.stream();
@@ -200,7 +201,7 @@ class WCCOpt : public GPUAppBase<FRAG_T, WCCOptContext<FRAG_T>>,
     }
 
     if (frag.fid() > 0) {
-      WorkSourceRange<vertex_t> ws_in(iv.begin(), iv.size());
+      WorkSourceRange<vertex_t> ws_in(av.begin(), av.size());
 
       ForEach(stream, ws_in, [=] __device__(vertex_t v) mutable {
         auto offset = vertex2offset(v);

misc/cuda_app_tests.sh

@@ -120,7 +120,16 @@ for np in ${proc_list}; do
     RunApp ${np} wcc -lb=${lb} ${SER} --serialization_prefix=./serial/${GRAPH}_wcc
     WCCVerify ${GRAPE_HOME}/dataset/${GRAPH}-WCC
 
-    RunApp ${np} wcc_opt -lb=${lb} ${SER} --serialization_prefix=./serial/${GRAPH}_wcc_opt
+    RunApp ${np} wcc_opt -segmented_partition=true -rebalance=true -lb=${lb} ${SER} --serialization_prefix=./serial/${GRAPH}_wcc_opt_tt
+    WCCVerify ${GRAPE_HOME}/dataset/${GRAPH}-WCC
+
+    RunApp ${np} wcc_opt -segmented_partition=true -rebalance=false -lb=${lb} ${SER} --serialization_prefix=./serial/${GRAPH}_wcc_opt_tf
+    WCCVerify ${GRAPE_HOME}/dataset/${GRAPH}-WCC
+
+    RunApp ${np} wcc_opt -segmented_partition=false -rebalance=true -lb=${lb} ${SER} --serialization_prefix=./serial/${GRAPH}_wcc_opt_ft
+    WCCVerify ${GRAPE_HOME}/dataset/${GRAPH}-WCC
+
+    RunApp ${np} wcc_opt -segmented_partition=false -rebalance=false -lb=${lb} ${SER} --serialization_prefix=./serial/${GRAPH}_wcc_opt_ff
     WCCVerify ${GRAPE_HOME}/dataset/${GRAPH}-WCC
   done
 done