Have all coloring tests passing

trilinos · Jul 5, 2021 · e4f1c7e · e4f1c7e
1 parent 3e4eddc
commit e4f1c7e
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/packages/zoltan2/core/src/algorithms/color/Zoltan2_AlgHybridPD2.hpp b/packages/zoltan2/core/src/algorithms/color/Zoltan2_AlgHybridPD2.hpp
@@ -159,6 +159,7 @@ class AlgPartialDistance2 : public AlgTwoGhostLayer<Adapter> {
 			    bool recolor_degrees){
 
       Kokkos::RangePolicy<ExecutionSpace> policy(0,boundary_verts_view.extent(0));
+      size_t local_recoloring_size;
       Kokkos::parallel_reduce("PD2 conflict detection",policy, KOKKOS_LAMBDA(const uint64_t& i,size_t& recoloring_size){
 	  //we only detect conflicts for vertices in the boundary
           const size_t curr_lid = boundary_verts_view(i);
@@ -219,7 +220,8 @@ class AlgPartialDistance2 : public AlgTwoGhostLayer<Adapter> {
             }      //              to completely move on to the next vertex.    |
             if(found) break;//<--------------------------------------------------
           }
-        },recoloringSize(0));
+        },local_recoloring_size);
+      Kokkos::deep_copy(recoloringSize, local_recoloring_size);
         Kokkos::fence();
 	//update the verts_to_send and verts_to_recolor views
         Kokkos::parallel_for("rebuild verts_to_send and verts_to_recolor",