1674ae819SStefano Zampini #include <petsc-private/petscimpl.h> 2674ae819SStefano Zampini #include "bddcprivate.h" 3674ae819SStefano Zampini #include "bddcstructs.h" 4674ae819SStefano Zampini 5674ae819SStefano Zampini #define NEUMANN_MARK -1 6674ae819SStefano Zampini #define DIRICHLET_MARK -2 7674ae819SStefano Zampini #define SPECIAL_MARK -3 8674ae819SStefano Zampini 9674ae819SStefano Zampini #undef __FUNCT__ 10674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphASCIIView" 11*e49050b4SStefano Zampini PetscErrorCode PCBDDCGraphASCIIView(PCBDDCGraph graph, PetscInt verbosity_level, PetscViewer viewer) 12674ae819SStefano Zampini { 13674ae819SStefano Zampini PetscInt i,j; 14674ae819SStefano Zampini PetscErrorCode ierr; 15674ae819SStefano Zampini 16674ae819SStefano Zampini PetscFunctionBegin; 17674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedAllow(viewer,PETSC_TRUE);CHKERRQ(ierr); 18674ae819SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer,"--------------------------------------------------\n");CHKERRQ(ierr); 19674ae819SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 20674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Local BDDC graph for subdomain %04d\n",PetscGlobalRank);CHKERRQ(ierr); 21674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Number of vertices %d\n",graph->nvtxs);CHKERRQ(ierr); 22*e49050b4SStefano Zampini if (verbosity_level > 1) { 23674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 24674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"%d:\n",i);CHKERRQ(ierr); 25674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," touched: %d\n",graph->touched[i]);CHKERRQ(ierr); 26674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," which_dof: %d\n",graph->which_dof[i]);CHKERRQ(ierr); 27674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," neighbours: %d\n",graph->count[i]);CHKERRQ(ierr); 28674ae819SStefano Zampini if (graph->count[i]) { 29674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," set of neighbours:");CHKERRQ(ierr); 30674ae819SStefano Zampini for (j=0;j<graph->count[i];j++) { 31674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->neighbours_set[i][j]);CHKERRQ(ierr); 32674ae819SStefano Zampini } 33674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 34674ae819SStefano Zampini } 35*e49050b4SStefano Zampini if (verbosity_level > 2) { 36674ae819SStefano Zampini if (graph->xadj && graph->adjncy) { 37674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," local adj list:");CHKERRQ(ierr); 38674ae819SStefano Zampini for (j=graph->xadj[i];j<graph->xadj[i+1];j++) { 39674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->adjncy[j]);CHKERRQ(ierr); 40674ae819SStefano Zampini } 41674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 42674ae819SStefano Zampini } 43*e49050b4SStefano Zampini } 44674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," interface subset id: %d\n",graph->subset[i]);CHKERRQ(ierr); 45674ae819SStefano Zampini if (graph->subset[i] && graph->subset_ncc) { 46674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," ncc for subset: %d\n",graph->subset_ncc[graph->subset[i]-1]);CHKERRQ(ierr); 47674ae819SStefano Zampini } 48674ae819SStefano Zampini } 49*e49050b4SStefano Zampini } 50674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Total number of connected components %d\n",graph->ncc);CHKERRQ(ierr); 51674ae819SStefano Zampini for (i=0;i<graph->ncc;i++) { 52674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d (neighs:",i);CHKERRQ(ierr); 53674ae819SStefano Zampini PetscInt node_num=graph->queue[graph->cptr[i]]; 54674ae819SStefano Zampini for (j=0;j<graph->count[node_num];j++) { 55674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->neighbours_set[node_num][j]);CHKERRQ(ierr); 56674ae819SStefano Zampini } 57674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"):");CHKERRQ(ierr); 58674ae819SStefano Zampini for (j=graph->cptr[i];j<graph->cptr[i+1];j++) { 59674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer," %d",graph->queue[j]);CHKERRQ(ierr); 60674ae819SStefano Zampini } 61674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"\n");CHKERRQ(ierr); 62674ae819SStefano Zampini } 63*e49050b4SStefano Zampini if (graph->custom_minimal_size > 1 && verbosity_level > 1) { 64674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Custom minimal size %d\n",graph->custom_minimal_size);CHKERRQ(ierr); 65674ae819SStefano Zampini } 66674ae819SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 67674ae819SStefano Zampini PetscFunctionReturn(0); 68674ae819SStefano Zampini } 69674ae819SStefano Zampini 70674ae819SStefano Zampini #undef __FUNCT__ 71674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphGetCandidatesIS" 72674ae819SStefano Zampini PetscErrorCode PCBDDCGraphGetCandidatesIS(PCBDDCGraph graph, PetscBool use_faces, PetscBool use_edges, PetscBool use_vertices, PetscInt *n_faces, IS *FacesIS[], PetscInt *n_edges, IS *EdgesIS[], IS *VerticesIS) 73674ae819SStefano Zampini { 74674ae819SStefano Zampini IS *ISForFaces,*ISForEdges,ISForVertices; 75674ae819SStefano Zampini PetscInt i,j,nfc,nec,nvc,*idx; 76674ae819SStefano Zampini PetscBool twodim_flag; 77674ae819SStefano Zampini PetscErrorCode ierr; 78674ae819SStefano Zampini 79674ae819SStefano Zampini PetscFunctionBegin; 80674ae819SStefano Zampini /* loop on ccs to evalute number of faces, edges and vertices */ 81674ae819SStefano Zampini nfc = 0; 82674ae819SStefano Zampini nec = 0; 83674ae819SStefano Zampini nvc = 0; 84674ae819SStefano Zampini twodim_flag = PETSC_FALSE; 85674ae819SStefano Zampini for (i=0;i<graph->ncc;i++) { 86674ae819SStefano Zampini if (graph->cptr[i+1]-graph->cptr[i] > graph->custom_minimal_size) { 87674ae819SStefano Zampini if (graph->count[graph->queue[graph->cptr[i]]] == 1) { 88674ae819SStefano Zampini nfc++; 89674ae819SStefano Zampini } else { /* note that nec will be zero in 2d */ 90674ae819SStefano Zampini nec++; 91674ae819SStefano Zampini } 92674ae819SStefano Zampini } else { 93674ae819SStefano Zampini nvc += graph->cptr[i+1]-graph->cptr[i]; 94674ae819SStefano Zampini } 95674ae819SStefano Zampini } 96674ae819SStefano Zampini if (!nec) { /* we are in a 2d case -> no faces, only edges */ 97674ae819SStefano Zampini nec = nfc; 98674ae819SStefano Zampini nfc = 0; 99674ae819SStefano Zampini twodim_flag = PETSC_TRUE; 100674ae819SStefano Zampini } 101674ae819SStefano Zampini /* allocate IS arrays for faces, edges. Vertices need a single index set. */ 102674ae819SStefano Zampini ISForFaces = 0; 103674ae819SStefano Zampini ISForEdges = 0; 104674ae819SStefano Zampini ISForVertices = 0; 105674ae819SStefano Zampini if (use_faces) { 106674ae819SStefano Zampini ierr = PetscMalloc(nfc*sizeof(IS),&ISForFaces);CHKERRQ(ierr); 107674ae819SStefano Zampini } 108674ae819SStefano Zampini if (use_edges) { 109674ae819SStefano Zampini ierr = PetscMalloc(nec*sizeof(IS),&ISForEdges);CHKERRQ(ierr); 110674ae819SStefano Zampini } 111674ae819SStefano Zampini if (use_vertices) { 112674ae819SStefano Zampini ierr = PetscMalloc(nvc*sizeof(PetscInt),&idx);CHKERRQ(ierr); 113674ae819SStefano Zampini } 114674ae819SStefano Zampini /* loop on ccs to compute index sets for faces and edges */ 115674ae819SStefano Zampini nfc = 0; 116674ae819SStefano Zampini nec = 0; 117674ae819SStefano Zampini nvc = 0; 118674ae819SStefano Zampini for (i=0;i<graph->ncc;i++) { 119674ae819SStefano Zampini if (graph->cptr[i+1]-graph->cptr[i] > graph->custom_minimal_size) { 120674ae819SStefano Zampini if (graph->count[graph->queue[graph->cptr[i]]] == 1) { 121674ae819SStefano Zampini if (twodim_flag) { 122674ae819SStefano Zampini if (use_edges) { 123674ae819SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,graph->cptr[i+1]-graph->cptr[i],&graph->queue[graph->cptr[i]],PETSC_COPY_VALUES,&ISForEdges[nec]);CHKERRQ(ierr); 124674ae819SStefano Zampini nec++; 125674ae819SStefano Zampini } 126674ae819SStefano Zampini } else { 127674ae819SStefano Zampini if (use_faces) { 128674ae819SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,graph->cptr[i+1]-graph->cptr[i],&graph->queue[graph->cptr[i]],PETSC_COPY_VALUES,&ISForFaces[nfc]);CHKERRQ(ierr); 129674ae819SStefano Zampini nfc++; 130674ae819SStefano Zampini } 131674ae819SStefano Zampini } 132674ae819SStefano Zampini } else { 133674ae819SStefano Zampini if (use_edges) { 134674ae819SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,graph->cptr[i+1]-graph->cptr[i],&graph->queue[graph->cptr[i]],PETSC_COPY_VALUES,&ISForEdges[nec]);CHKERRQ(ierr); 135674ae819SStefano Zampini nec++; 136674ae819SStefano Zampini } 137674ae819SStefano Zampini } 138674ae819SStefano Zampini } 139674ae819SStefano Zampini } 140674ae819SStefano Zampini /* index set for vertices */ 141674ae819SStefano Zampini if (use_vertices) { 142674ae819SStefano Zampini for (i=0;i<graph->ncc;i++) { 143674ae819SStefano Zampini if (graph->cptr[i+1]-graph->cptr[i] <= graph->custom_minimal_size) { 144674ae819SStefano Zampini for (j=graph->cptr[i];j<graph->cptr[i+1];j++) { 145674ae819SStefano Zampini idx[nvc]=graph->queue[j]; 146674ae819SStefano Zampini nvc++; 147674ae819SStefano Zampini } 148674ae819SStefano Zampini } 149674ae819SStefano Zampini } 150674ae819SStefano Zampini /* sort vertex set (by local ordering) */ 151674ae819SStefano Zampini ierr = PetscSortInt(nvc,idx);CHKERRQ(ierr); 152674ae819SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,nvc,idx,PETSC_OWN_POINTER,&ISForVertices);CHKERRQ(ierr); 153674ae819SStefano Zampini } 154674ae819SStefano Zampini /* get back info */ 155674ae819SStefano Zampini *n_faces = nfc; 156674ae819SStefano Zampini *FacesIS = ISForFaces; 157674ae819SStefano Zampini *n_edges = nec; 158674ae819SStefano Zampini *EdgesIS = ISForEdges; 159674ae819SStefano Zampini *VerticesIS = ISForVertices; 160674ae819SStefano Zampini PetscFunctionReturn(0); 161674ae819SStefano Zampini } 162674ae819SStefano Zampini 163674ae819SStefano Zampini #undef __FUNCT__ 164674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphComputeConnectedComponents" 165674ae819SStefano Zampini PetscErrorCode PCBDDCGraphComputeConnectedComponents(PCBDDCGraph graph) 166674ae819SStefano Zampini { 167674ae819SStefano Zampini PetscInt adapt_interface,adapt_interface_reduced; 168674ae819SStefano Zampini MPI_Comm interface_comm; 169674ae819SStefano Zampini MPI_Request *send_requests; 170674ae819SStefano Zampini MPI_Request *recv_requests; 171674ae819SStefano Zampini PetscInt *aux_new_xadj,*new_xadj,*new_adjncy,**temp_buffer; 172674ae819SStefano Zampini PetscInt i,j,k,s,sum_requests,buffer_size,size_of_recv,temp_buffer_size; 173674ae819SStefano Zampini PetscMPIInt rank,neigh,tag,mpi_buffer_size; 174674ae819SStefano Zampini PetscInt *cum_recv_counts,*subset_to_nodes_indices,*recv_buffer_subset,*nodes_to_temp_buffer_indices; 175674ae819SStefano Zampini PetscInt *send_buffer,*recv_buffer,*queue_in_global_numbering,*sizes_of_sends,*add_to_subset; 176674ae819SStefano Zampini PetscInt start_of_recv,start_of_send,size_of_send,global_subset_counter,ins_val; 177674ae819SStefano Zampini PetscBool *subset_cc_adapt,same_set; 178674ae819SStefano Zampini PetscErrorCode ierr; 179674ae819SStefano Zampini 180674ae819SStefano Zampini PetscFunctionBegin; 181674ae819SStefano Zampini /* compute connected components locally */ 182674ae819SStefano Zampini ierr = PetscObjectGetComm((PetscObject)(graph->l2gmap),&interface_comm);CHKERRQ(ierr); 183674ae819SStefano Zampini ierr = PCBDDCGraphComputeConnectedComponentsLocal(graph);CHKERRQ(ierr); 184674ae819SStefano Zampini /* check consistency of connected components among neighbouring subdomains -> it adapt them in case it is needed */ 185674ae819SStefano Zampini adapt_interface = 0; 186674ae819SStefano Zampini adapt_interface_reduced = 0; 187674ae819SStefano Zampini for (i=0;i<graph->n_subsets;i++) { 188674ae819SStefano Zampini /* We are not sure that on a given subset of the local interface, 189674ae819SStefano Zampini with two connected components, the latters be the same among sharing subdomains */ 190674ae819SStefano Zampini if (graph->subset_ncc[i] > 1) { 191674ae819SStefano Zampini adapt_interface=1; 192674ae819SStefano Zampini break; 193674ae819SStefano Zampini } 194674ae819SStefano Zampini } 195674ae819SStefano Zampini ierr = MPI_Allreduce(&adapt_interface,&adapt_interface_reduced,1,MPIU_INT,MPI_LOR,interface_comm);CHKERRQ(ierr); 196674ae819SStefano Zampini 197674ae819SStefano Zampini if (graph->n_subsets && adapt_interface_reduced) { 198674ae819SStefano Zampini /* Retrict adjacency graph using information from previously computed connected components */ 199674ae819SStefano Zampini ierr = PetscMalloc(graph->nvtxs*sizeof(PetscInt),&aux_new_xadj);CHKERRQ(ierr); 200674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 201674ae819SStefano Zampini aux_new_xadj[i]=1; 202674ae819SStefano Zampini } 203674ae819SStefano Zampini for (i=0;i<graph->ncc;i++) { 204674ae819SStefano Zampini k = graph->cptr[i+1]-graph->cptr[i]; 205674ae819SStefano Zampini for (j=0;j<k;j++) { 206674ae819SStefano Zampini aux_new_xadj[graph->queue[graph->cptr[i]+j]]=k; 207674ae819SStefano Zampini } 208674ae819SStefano Zampini } 209674ae819SStefano Zampini j = 0; 210674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 211674ae819SStefano Zampini j += aux_new_xadj[i]; 212674ae819SStefano Zampini } 213674ae819SStefano Zampini ierr = PetscMalloc((graph->nvtxs+1)*sizeof(PetscInt),&new_xadj);CHKERRQ(ierr); 214674ae819SStefano Zampini ierr = PetscMalloc(j*sizeof(PetscInt),&new_adjncy);CHKERRQ(ierr); 215674ae819SStefano Zampini new_xadj[0]=0; 216674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 217674ae819SStefano Zampini new_xadj[i+1]=new_xadj[i]+aux_new_xadj[i]; 218674ae819SStefano Zampini if (aux_new_xadj[i]==1) { 219674ae819SStefano Zampini new_adjncy[new_xadj[i]]=i; 220674ae819SStefano Zampini } 221674ae819SStefano Zampini } 222674ae819SStefano Zampini ierr = PetscFree(aux_new_xadj);CHKERRQ(ierr); 223674ae819SStefano Zampini for (i=0;i<graph->ncc;i++) { 224674ae819SStefano Zampini k = graph->cptr[i+1]-graph->cptr[i]; 225674ae819SStefano Zampini for (j=0;j<k;j++) { 226674ae819SStefano Zampini ierr = PetscMemcpy(&new_adjncy[new_xadj[graph->queue[graph->cptr[i]+j]]],&graph->queue[graph->cptr[i]],k*sizeof(PetscInt));CHKERRQ(ierr); 227674ae819SStefano Zampini } 228674ae819SStefano Zampini } 229674ae819SStefano Zampini /* set new CSR into graph */ 230674ae819SStefano Zampini ierr = PetscFree(graph->xadj);CHKERRQ(ierr); 231674ae819SStefano Zampini ierr = PetscFree(graph->adjncy);CHKERRQ(ierr); 232674ae819SStefano Zampini graph->xadj = new_xadj; 233674ae819SStefano Zampini graph->adjncy = new_adjncy; 234674ae819SStefano Zampini /* allocate some space */ 235674ae819SStefano Zampini ierr = MPI_Comm_rank(interface_comm,&rank);CHKERRQ(ierr); 236674ae819SStefano Zampini ierr = PetscMalloc((graph->n_subsets+1)*sizeof(*cum_recv_counts),&cum_recv_counts);CHKERRQ(ierr); 237674ae819SStefano Zampini ierr = PetscMemzero(cum_recv_counts,(graph->n_subsets+1)*sizeof(*cum_recv_counts));CHKERRQ(ierr); 238674ae819SStefano Zampini ierr = PetscMalloc(graph->n_subsets*sizeof(*subset_to_nodes_indices),&subset_to_nodes_indices);CHKERRQ(ierr); 239674ae819SStefano Zampini /* first count how many neighbours per connected component I will receive from */ 240674ae819SStefano Zampini cum_recv_counts[0]=0; 241674ae819SStefano Zampini for (i=1;i<graph->n_subsets+1;i++) { 242674ae819SStefano Zampini j = 0; 243674ae819SStefano Zampini while (graph->subset[j] != i) { 244674ae819SStefano Zampini j++; 245674ae819SStefano Zampini } 246674ae819SStefano Zampini subset_to_nodes_indices[i-1]=j; 247674ae819SStefano Zampini /* We don't want sends/recvs_to/from_self -> here I don't count myself */ 248674ae819SStefano Zampini cum_recv_counts[i]=cum_recv_counts[i-1]+graph->count[j]; 249674ae819SStefano Zampini } 250674ae819SStefano Zampini ierr = PetscMalloc(2*cum_recv_counts[graph->n_subsets]*sizeof(*recv_buffer_subset),&recv_buffer_subset);CHKERRQ(ierr); 251674ae819SStefano Zampini ierr = PetscMalloc(cum_recv_counts[graph->n_subsets]*sizeof(MPI_Request),&send_requests);CHKERRQ(ierr); 252674ae819SStefano Zampini ierr = PetscMalloc(cum_recv_counts[graph->n_subsets]*sizeof(MPI_Request),&recv_requests);CHKERRQ(ierr); 253674ae819SStefano Zampini for (i=0;i<cum_recv_counts[graph->n_subsets];i++) { 254674ae819SStefano Zampini send_requests[i]=MPI_REQUEST_NULL; 255674ae819SStefano Zampini recv_requests[i]=MPI_REQUEST_NULL; 256674ae819SStefano Zampini } 257674ae819SStefano Zampini /* exchange with my neighbours the number of my connected components on the shared interface */ 258674ae819SStefano Zampini sum_requests = 0; 259674ae819SStefano Zampini for (i=0;i<graph->n_subsets;i++) { 260674ae819SStefano Zampini j = subset_to_nodes_indices[i]; 261674ae819SStefano Zampini for (k=0;k<graph->count[j];k++) { 262674ae819SStefano Zampini ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr); 263674ae819SStefano Zampini ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr); 264674ae819SStefano Zampini ierr = MPI_Isend(&graph->subset_ncc[i],1,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr); 265674ae819SStefano Zampini ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr); 266674ae819SStefano Zampini ierr = MPI_Irecv(&recv_buffer_subset[sum_requests],1,MPIU_INT,neigh,tag,interface_comm,&recv_requests[sum_requests]);CHKERRQ(ierr); 267674ae819SStefano Zampini sum_requests++; 268674ae819SStefano Zampini } 269674ae819SStefano Zampini } 270674ae819SStefano Zampini ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr); 271674ae819SStefano Zampini ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr); 272674ae819SStefano Zampini /* determine the connected component I need to adapt */ 273674ae819SStefano Zampini ierr = PetscMalloc(graph->n_subsets*sizeof(*subset_cc_adapt),&subset_cc_adapt);CHKERRQ(ierr); 274674ae819SStefano Zampini ierr = PetscMemzero(subset_cc_adapt,graph->n_subsets*sizeof(*subset_cc_adapt));CHKERRQ(ierr); 275674ae819SStefano Zampini for (i=0;i<graph->n_subsets;i++) { 276674ae819SStefano Zampini for (j=cum_recv_counts[i];j<cum_recv_counts[i+1];j++){ 277674ae819SStefano Zampini /* The first condition is natural (someone has a different number of ccs than me), the second one is just to be safe */ 278674ae819SStefano Zampini if ( graph->subset_ncc[i] != recv_buffer_subset[j] || graph->subset_ncc[i] > 1 ) { 279674ae819SStefano Zampini subset_cc_adapt[i] = PETSC_TRUE; 280674ae819SStefano Zampini break; 281674ae819SStefano Zampini } 282674ae819SStefano Zampini } 283674ae819SStefano Zampini } 284674ae819SStefano Zampini buffer_size = 0; 285674ae819SStefano Zampini for (i=0;i<graph->n_subsets;i++) { 286674ae819SStefano Zampini if (subset_cc_adapt[i]) { 287674ae819SStefano Zampini for (j=i;j<graph->ncc;j++) { 288674ae819SStefano Zampini if (graph->subset[graph->queue[graph->cptr[j]]] == i+1) { /* WARNING -> subset values goes from 1 to graph->n_subsets included */ 289674ae819SStefano Zampini buffer_size += 1 + graph->cptr[j+1]-graph->cptr[j]; 290674ae819SStefano Zampini } 291674ae819SStefano Zampini } 292674ae819SStefano Zampini } 293674ae819SStefano Zampini } 294674ae819SStefano Zampini ierr = PetscMalloc(buffer_size*sizeof(*send_buffer),&send_buffer);CHKERRQ(ierr); 295674ae819SStefano Zampini /* now get from neighbours their ccs (in global numbering) and adapt them (in case it is needed) */ 296674ae819SStefano Zampini ierr = PetscMalloc(graph->nvtxs*sizeof(*queue_in_global_numbering),&queue_in_global_numbering);CHKERRQ(ierr); 297674ae819SStefano Zampini ierr = ISLocalToGlobalMappingApply(graph->l2gmap,graph->cptr[graph->ncc],graph->queue,queue_in_global_numbering);CHKERRQ(ierr); 298674ae819SStefano Zampini /* determine how much data to send (size of each queue plus the global indices) and communicate it to neighbours */ 299674ae819SStefano Zampini ierr = PetscMalloc(graph->n_subsets*sizeof(*sizes_of_sends),&sizes_of_sends);CHKERRQ(ierr); 300674ae819SStefano Zampini ierr = PetscMemzero(sizes_of_sends,graph->n_subsets*sizeof(*sizes_of_sends));CHKERRQ(ierr); 301674ae819SStefano Zampini sum_requests = 0; 302674ae819SStefano Zampini start_of_send = 0; 303674ae819SStefano Zampini start_of_recv = cum_recv_counts[graph->n_subsets]; 304674ae819SStefano Zampini for (i=0;i<graph->n_subsets;i++) { 305674ae819SStefano Zampini if (subset_cc_adapt[i]) { 306674ae819SStefano Zampini size_of_send = 0; 307674ae819SStefano Zampini for (j=i;j<graph->ncc;j++) { 308674ae819SStefano Zampini if (graph->subset[graph->queue[graph->cptr[j]]] == i+1) { /* WARNING -> subset values goes from 1 to graph->n_subsets included */ 309674ae819SStefano Zampini send_buffer[start_of_send+size_of_send]=graph->cptr[j+1]-graph->cptr[j]; 310674ae819SStefano Zampini size_of_send += 1; 311674ae819SStefano Zampini ierr = PetscMemcpy(&send_buffer[start_of_send+size_of_send], 312674ae819SStefano Zampini &queue_in_global_numbering[graph->cptr[j]], 313674ae819SStefano Zampini (graph->cptr[j+1]-graph->cptr[j])*sizeof(*send_buffer));CHKERRQ(ierr); 314674ae819SStefano Zampini size_of_send = size_of_send+graph->cptr[j+1]-graph->cptr[j]; 315674ae819SStefano Zampini } 316674ae819SStefano Zampini } 317674ae819SStefano Zampini j = subset_to_nodes_indices[i]; 318674ae819SStefano Zampini sizes_of_sends[i] = size_of_send; 319674ae819SStefano Zampini for (k=0;k<graph->count[j];k++) { 320674ae819SStefano Zampini ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr); 321674ae819SStefano Zampini ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr); 322674ae819SStefano Zampini ierr = MPI_Isend(&sizes_of_sends[i],1,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr); 323674ae819SStefano Zampini ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr); 324674ae819SStefano Zampini ierr = MPI_Irecv(&recv_buffer_subset[sum_requests+start_of_recv],1,MPIU_INT,neigh,tag,interface_comm,&recv_requests[sum_requests]);CHKERRQ(ierr); 325674ae819SStefano Zampini sum_requests++; 326674ae819SStefano Zampini } 327674ae819SStefano Zampini start_of_send += size_of_send; 328674ae819SStefano Zampini } 329674ae819SStefano Zampini } 330674ae819SStefano Zampini ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr); 331674ae819SStefano Zampini ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr); 332674ae819SStefano Zampini buffer_size = 0; 333674ae819SStefano Zampini for (k=0;k<sum_requests;k++) { 334674ae819SStefano Zampini buffer_size += recv_buffer_subset[start_of_recv+k]; 335674ae819SStefano Zampini } 336674ae819SStefano Zampini ierr = PetscMalloc(buffer_size*sizeof(*recv_buffer),&recv_buffer);CHKERRQ(ierr); 337674ae819SStefano Zampini /* now exchange the data */ 338674ae819SStefano Zampini start_of_recv = 0; 339674ae819SStefano Zampini start_of_send = 0; 340674ae819SStefano Zampini sum_requests = 0; 341674ae819SStefano Zampini for (i=0;i<graph->n_subsets;i++) { 342674ae819SStefano Zampini if (subset_cc_adapt[i]) { 343674ae819SStefano Zampini size_of_send = sizes_of_sends[i]; 344674ae819SStefano Zampini j = subset_to_nodes_indices[i]; 345674ae819SStefano Zampini for (k=0;k<graph->count[j];k++) { 346674ae819SStefano Zampini ierr = PetscMPIIntCast(graph->neighbours_set[j][k],&neigh);CHKERRQ(ierr); 347674ae819SStefano Zampini ierr = PetscMPIIntCast((PetscInt)(rank+1)*graph->count[j],&tag);CHKERRQ(ierr); 348674ae819SStefano Zampini ierr = PetscMPIIntCast(size_of_send,&mpi_buffer_size);CHKERRQ(ierr); 349674ae819SStefano Zampini ierr = MPI_Isend(&send_buffer[start_of_send],mpi_buffer_size,MPIU_INT,neigh,tag,interface_comm,&send_requests[sum_requests]);CHKERRQ(ierr); 350674ae819SStefano Zampini size_of_recv = recv_buffer_subset[cum_recv_counts[graph->n_subsets]+sum_requests]; 351674ae819SStefano Zampini ierr = PetscMPIIntCast((PetscInt)(neigh+1)*graph->count[j],&tag);CHKERRQ(ierr); 352674ae819SStefano Zampini ierr = PetscMPIIntCast(size_of_recv,&mpi_buffer_size);CHKERRQ(ierr); 353674ae819SStefano Zampini ierr = MPI_Irecv(&recv_buffer[start_of_recv],mpi_buffer_size,MPIU_INT,neigh,tag,interface_comm,&recv_requests[sum_requests]);CHKERRQ(ierr); 354674ae819SStefano Zampini start_of_recv += size_of_recv; 355674ae819SStefano Zampini sum_requests++; 356674ae819SStefano Zampini } 357674ae819SStefano Zampini start_of_send += size_of_send; 358674ae819SStefano Zampini } 359674ae819SStefano Zampini } 360674ae819SStefano Zampini ierr = MPI_Waitall(sum_requests,recv_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr); 361674ae819SStefano Zampini ierr = MPI_Waitall(sum_requests,send_requests,MPI_STATUSES_IGNORE);CHKERRQ(ierr); 362674ae819SStefano Zampini for (j=0;j<buffer_size;) { 363674ae819SStefano Zampini ierr = ISGlobalToLocalMappingApply(graph->l2gmap,IS_GTOLM_MASK,recv_buffer[j],&recv_buffer[j+1],&recv_buffer[j],&recv_buffer[j+1]);CHKERRQ(ierr); 364674ae819SStefano Zampini k = recv_buffer[j]+1; 365674ae819SStefano Zampini j += k; 366674ae819SStefano Zampini } 367674ae819SStefano Zampini sum_requests = cum_recv_counts[graph->n_subsets]; 368674ae819SStefano Zampini start_of_recv = 0; 369674ae819SStefano Zampini ierr = PetscMalloc(graph->nvtxs*sizeof(*nodes_to_temp_buffer_indices),&nodes_to_temp_buffer_indices);CHKERRQ(ierr); 370674ae819SStefano Zampini global_subset_counter = 0; 371674ae819SStefano Zampini for (i=0;i<graph->n_subsets;i++) { 372674ae819SStefano Zampini if (subset_cc_adapt[i]) { 373674ae819SStefano Zampini temp_buffer_size = 0; 374674ae819SStefano Zampini /* find nodes on the shared interface we need to adapt */ 375674ae819SStefano Zampini for (j=0;j<graph->nvtxs;j++) { 376674ae819SStefano Zampini if (graph->subset[j]==i+1) { 377674ae819SStefano Zampini nodes_to_temp_buffer_indices[j] = temp_buffer_size; 378674ae819SStefano Zampini temp_buffer_size++; 379674ae819SStefano Zampini } else { 380674ae819SStefano Zampini nodes_to_temp_buffer_indices[j] = -1; 381674ae819SStefano Zampini } 382674ae819SStefano Zampini } 383674ae819SStefano Zampini /* allocate some temporary space */ 384674ae819SStefano Zampini ierr = PetscMalloc(temp_buffer_size*sizeof(PetscInt*),&temp_buffer);CHKERRQ(ierr); 385674ae819SStefano Zampini ierr = PetscMalloc(temp_buffer_size*(cum_recv_counts[i+1]-cum_recv_counts[i])*sizeof(PetscInt),&temp_buffer[0]);CHKERRQ(ierr); 386674ae819SStefano Zampini ierr = PetscMemzero(temp_buffer[0],temp_buffer_size*(cum_recv_counts[i+1]-cum_recv_counts[i])*sizeof(PetscInt));CHKERRQ(ierr); 387674ae819SStefano Zampini for (j=1;j<temp_buffer_size;j++) { 388674ae819SStefano Zampini temp_buffer[j] = temp_buffer[j-1]+cum_recv_counts[i+1]-cum_recv_counts[i]; 389674ae819SStefano Zampini } 390674ae819SStefano Zampini /* analyze contributions from neighbouring subdomains for i-th conn comp 391674ae819SStefano Zampini temp buffer structure: 392674ae819SStefano Zampini supposing part of the interface has dimension 5 (for example with global dofs 0,1,2,3,4) 393674ae819SStefano Zampini 3 neighs procs with structured connected components: 394674ae819SStefano Zampini neigh 0: [0 1 4], [2 3]; (2 connected components) 395674ae819SStefano Zampini neigh 1: [0 1], [2 3 4]; (2 connected components) 396674ae819SStefano Zampini neigh 2: [0 4], [1], [2 3]; (3 connected components) 397674ae819SStefano Zampini tempbuffer (row-oriented) will be filled as: 398674ae819SStefano Zampini [ 0, 0, 0; 399674ae819SStefano Zampini 0, 0, 1; 400674ae819SStefano Zampini 1, 1, 2; 401674ae819SStefano Zampini 1, 1, 2; 402674ae819SStefano Zampini 0, 1, 0; ]; 403674ae819SStefano Zampini This way we can simply find intersections of ccs among neighs. 404674ae819SStefano Zampini For the example above, the graph->subset array will be modified to reproduce the following 4 connected components [0], [1], [2 3], [4]; 405674ae819SStefano Zampini */ 406674ae819SStefano Zampini for (j=0;j<cum_recv_counts[i+1]-cum_recv_counts[i];j++) { 407674ae819SStefano Zampini ins_val = 0; 408674ae819SStefano Zampini size_of_recv = recv_buffer_subset[sum_requests]; /* total size of recv from neighs */ 409674ae819SStefano Zampini for (buffer_size=0;buffer_size<size_of_recv;) { /* loop until all data from neighs has been taken into account */ 410674ae819SStefano Zampini for (k=1;k<recv_buffer[buffer_size+start_of_recv]+1;k++) { /* filling properly temp_buffer using data from a single recv */ 411674ae819SStefano Zampini temp_buffer[nodes_to_temp_buffer_indices[recv_buffer[start_of_recv+buffer_size+k]]][j] = ins_val; 412674ae819SStefano Zampini } 413674ae819SStefano Zampini buffer_size += k; 414674ae819SStefano Zampini ins_val++; 415674ae819SStefano Zampini } 416674ae819SStefano Zampini start_of_recv += size_of_recv; 417674ae819SStefano Zampini sum_requests++; 418674ae819SStefano Zampini } 419674ae819SStefano Zampini ierr = PetscMalloc(temp_buffer_size*sizeof(*add_to_subset),&add_to_subset);CHKERRQ(ierr); 420674ae819SStefano Zampini ierr = PetscMemzero(add_to_subset,temp_buffer_size*sizeof(*add_to_subset));CHKERRQ(ierr); 421674ae819SStefano Zampini for (j=0;j<temp_buffer_size;j++) { 422674ae819SStefano Zampini if (!add_to_subset[j]) { /* found a new cc */ 423674ae819SStefano Zampini global_subset_counter++; 424674ae819SStefano Zampini add_to_subset[j] = global_subset_counter; 425674ae819SStefano Zampini for (k=j+1;k<temp_buffer_size;k++) { /* check for other nodes in new cc */ 426674ae819SStefano Zampini same_set = PETSC_TRUE; 427674ae819SStefano Zampini for (s=0;s<cum_recv_counts[i+1]-cum_recv_counts[i];s++) { 428674ae819SStefano Zampini if (temp_buffer[j][s]!=temp_buffer[k][s]) { 429674ae819SStefano Zampini same_set = PETSC_FALSE; 430674ae819SStefano Zampini break; 431674ae819SStefano Zampini } 432674ae819SStefano Zampini } 433674ae819SStefano Zampini if (same_set) { 434674ae819SStefano Zampini add_to_subset[k] = global_subset_counter; 435674ae819SStefano Zampini } 436674ae819SStefano Zampini } 437674ae819SStefano Zampini } 438674ae819SStefano Zampini } 439674ae819SStefano Zampini /* insert new data in subset array */ 440674ae819SStefano Zampini temp_buffer_size = 0; 441674ae819SStefano Zampini for (j=0;j<graph->nvtxs;j++) { 442674ae819SStefano Zampini if (graph->subset[j]==i+1) { 443674ae819SStefano Zampini graph->subset[j] = graph->n_subsets+add_to_subset[temp_buffer_size]; 444674ae819SStefano Zampini temp_buffer_size++; 445674ae819SStefano Zampini } 446674ae819SStefano Zampini } 447674ae819SStefano Zampini ierr = PetscFree(temp_buffer[0]);CHKERRQ(ierr); 448674ae819SStefano Zampini ierr = PetscFree(temp_buffer);CHKERRQ(ierr); 449674ae819SStefano Zampini ierr = PetscFree(add_to_subset);CHKERRQ(ierr); 450674ae819SStefano Zampini } 451674ae819SStefano Zampini } 452674ae819SStefano Zampini ierr = PetscFree(nodes_to_temp_buffer_indices);CHKERRQ(ierr); 453674ae819SStefano Zampini ierr = PetscFree(sizes_of_sends);CHKERRQ(ierr); 454674ae819SStefano Zampini ierr = PetscFree(send_requests);CHKERRQ(ierr); 455674ae819SStefano Zampini ierr = PetscFree(recv_requests);CHKERRQ(ierr); 456674ae819SStefano Zampini ierr = PetscFree(recv_buffer);CHKERRQ(ierr); 457674ae819SStefano Zampini ierr = PetscFree(recv_buffer_subset);CHKERRQ(ierr); 458674ae819SStefano Zampini ierr = PetscFree(send_buffer);CHKERRQ(ierr); 459674ae819SStefano Zampini ierr = PetscFree(cum_recv_counts);CHKERRQ(ierr); 460674ae819SStefano Zampini ierr = PetscFree(subset_to_nodes_indices);CHKERRQ(ierr); 461674ae819SStefano Zampini ierr = PetscFree(subset_cc_adapt);CHKERRQ(ierr); 462674ae819SStefano Zampini /* We are ready to find for connected components consistent among neighbouring subdomains */ 463674ae819SStefano Zampini if (global_subset_counter) { 464674ae819SStefano Zampini ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(*graph->touched));CHKERRQ(ierr); 465674ae819SStefano Zampini global_subset_counter = 0; 466674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 467674ae819SStefano Zampini if (graph->subset[i] && !graph->touched[i]) { 468674ae819SStefano Zampini global_subset_counter++; 469674ae819SStefano Zampini for (j=i+1;j<graph->nvtxs;j++) { 470674ae819SStefano Zampini if (!graph->touched[j] && graph->subset[j]==graph->subset[i]) { 471674ae819SStefano Zampini graph->subset[j] = global_subset_counter; 472674ae819SStefano Zampini graph->touched[j] = PETSC_TRUE; 473674ae819SStefano Zampini } 474674ae819SStefano Zampini } 475674ae819SStefano Zampini graph->subset[i] = global_subset_counter; 476674ae819SStefano Zampini graph->touched[i] = PETSC_TRUE; 477674ae819SStefano Zampini } 478674ae819SStefano Zampini } 479674ae819SStefano Zampini /* refine connected components locally */ 480674ae819SStefano Zampini ierr = PCBDDCGraphComputeConnectedComponentsLocal(graph);CHKERRQ(ierr); 481674ae819SStefano Zampini } 482674ae819SStefano Zampini ierr = PetscFree(queue_in_global_numbering);CHKERRQ(ierr); 483674ae819SStefano Zampini } 484674ae819SStefano Zampini PetscFunctionReturn(0); 485674ae819SStefano Zampini } 486674ae819SStefano Zampini 487674ae819SStefano Zampini /* The following code has been adapted from function IsConnectedSubdomain contained 488674ae819SStefano Zampini in source file contig.c of METIS library (version 5.0.1) 489674ae819SStefano Zampini It finds connected components for each subset */ 490674ae819SStefano Zampini #undef __FUNCT__ 491674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphComputeConnectedComponentsLocal" 492674ae819SStefano Zampini PetscErrorCode PCBDDCGraphComputeConnectedComponentsLocal(PCBDDCGraph graph) 493674ae819SStefano Zampini { 494674ae819SStefano Zampini PetscInt i,j,k,first,last,nleft,ncc,pid,cum_queue,n,ncc_pid; 495674ae819SStefano Zampini PetscInt *queue_global; 496674ae819SStefano Zampini PetscErrorCode ierr; 497674ae819SStefano Zampini 498674ae819SStefano Zampini PetscFunctionBegin; 499674ae819SStefano Zampini /* quiet return if no csr info is available */ 500674ae819SStefano Zampini if (!graph->xadj || !graph->adjncy) { 501674ae819SStefano Zampini PetscFunctionReturn(0); 502674ae819SStefano Zampini } 503674ae819SStefano Zampini 504674ae819SStefano Zampini /* reset any previous search of connected components */ 505674ae819SStefano Zampini ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(*graph->touched));CHKERRQ(ierr); 506674ae819SStefano Zampini graph->n_subsets = 0; 507674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 508674ae819SStefano Zampini if (graph->which_dof[i] == DIRICHLET_MARK || !graph->count[i]) { 509674ae819SStefano Zampini graph->touched[i] = PETSC_TRUE; 510674ae819SStefano Zampini graph->subset[i] = 0; 511674ae819SStefano Zampini } 512674ae819SStefano Zampini graph->n_subsets = PetscMax(graph->n_subsets,graph->subset[i]); 513674ae819SStefano Zampini } 514674ae819SStefano Zampini ierr = PetscFree(graph->subset_ncc);CHKERRQ(ierr); 515674ae819SStefano Zampini ierr = PetscMalloc(graph->n_subsets*sizeof(*graph->subset_ncc),&graph->subset_ncc);CHKERRQ(ierr); 516674ae819SStefano Zampini ierr = PetscMemzero(graph->subset_ncc,graph->n_subsets*sizeof(*graph->subset_ncc));CHKERRQ(ierr); 517674ae819SStefano Zampini ierr = PetscMemzero(graph->cptr,(graph->nvtxs+1)*sizeof(*graph->cptr));CHKERRQ(ierr); 518674ae819SStefano Zampini ierr = PetscMemzero(graph->queue,graph->nvtxs*sizeof(*graph->queue));CHKERRQ(ierr); 519674ae819SStefano Zampini 520674ae819SStefano Zampini /* begin search for connected components */ 521674ae819SStefano Zampini cum_queue = 0; 522674ae819SStefano Zampini ncc = 0; 523674ae819SStefano Zampini for (n=0;n<graph->n_subsets;n++) { 524674ae819SStefano Zampini pid = n+1; /* partition labeled by 0 is discarded */ 525674ae819SStefano Zampini nleft = 0; 526674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 527674ae819SStefano Zampini if (graph->subset[i] == pid) { 528674ae819SStefano Zampini nleft++; 529674ae819SStefano Zampini } 530674ae819SStefano Zampini } 531674ae819SStefano Zampini for (i=0; i<graph->nvtxs; i++) { 532674ae819SStefano Zampini if (graph->subset[i] == pid) { 533674ae819SStefano Zampini break; 534674ae819SStefano Zampini } 535674ae819SStefano Zampini } 536674ae819SStefano Zampini graph->touched[i] = PETSC_TRUE; 537674ae819SStefano Zampini graph->queue[cum_queue] = i; 538674ae819SStefano Zampini first = 0; 539674ae819SStefano Zampini last = 1; 540674ae819SStefano Zampini graph->cptr[ncc] = cum_queue; 541674ae819SStefano Zampini ncc_pid = 0; 542674ae819SStefano Zampini while (first != nleft) { 543674ae819SStefano Zampini if (first == last) { 544674ae819SStefano Zampini graph->cptr[++ncc] = first+cum_queue; 545674ae819SStefano Zampini ncc_pid++; 546674ae819SStefano Zampini for (i=0; i<graph->nvtxs; i++) { 547674ae819SStefano Zampini if (graph->subset[i] == pid && !graph->touched[i]) { 548674ae819SStefano Zampini break; 549674ae819SStefano Zampini } 550674ae819SStefano Zampini } 551674ae819SStefano Zampini graph->queue[cum_queue+last] = i; 552674ae819SStefano Zampini last++; 553674ae819SStefano Zampini graph->touched[i] = PETSC_TRUE; 554674ae819SStefano Zampini } 555674ae819SStefano Zampini i = graph->queue[cum_queue+first]; 556674ae819SStefano Zampini first++; 557674ae819SStefano Zampini for (j=graph->xadj[i];j<graph->xadj[i+1];j++) { 558674ae819SStefano Zampini k = graph->adjncy[j]; 559674ae819SStefano Zampini if (graph->subset[k] == pid && !graph->touched[k]) { 560674ae819SStefano Zampini graph->queue[cum_queue+last] = k; 561674ae819SStefano Zampini last++; 562674ae819SStefano Zampini graph->touched[k] = PETSC_TRUE; 563674ae819SStefano Zampini } 564674ae819SStefano Zampini } 565674ae819SStefano Zampini } 566674ae819SStefano Zampini graph->cptr[++ncc] = first+cum_queue; 567674ae819SStefano Zampini ncc_pid++; 568674ae819SStefano Zampini cum_queue = graph->cptr[ncc]; 569674ae819SStefano Zampini graph->subset_ncc[n] = ncc_pid; 570674ae819SStefano Zampini } 571674ae819SStefano Zampini graph->ncc = ncc; 572674ae819SStefano Zampini /* For consistency among neighbours, I need to sort (by global ordering) each connected component */ 573674ae819SStefano Zampini ierr = PetscMalloc(graph->cptr[graph->ncc]*sizeof(*queue_global),&queue_global);CHKERRQ(ierr); 574674ae819SStefano Zampini ierr = ISLocalToGlobalMappingApply(graph->l2gmap,graph->cptr[graph->ncc],graph->queue,queue_global);CHKERRQ(ierr); 575674ae819SStefano Zampini for (i=0;i<graph->ncc;i++) { 576674ae819SStefano Zampini ierr = PetscSortIntWithArray(graph->cptr[i+1]-graph->cptr[i],&queue_global[graph->cptr[i]],&graph->queue[graph->cptr[i]]);CHKERRQ(ierr); 577674ae819SStefano Zampini } 578674ae819SStefano Zampini ierr = PetscFree(queue_global);CHKERRQ(ierr); 579674ae819SStefano Zampini PetscFunctionReturn(0); 580674ae819SStefano Zampini } 581674ae819SStefano Zampini 582674ae819SStefano Zampini #undef __FUNCT__ 583674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphSetUp" 584674ae819SStefano Zampini PetscErrorCode PCBDDCGraphSetUp(PCBDDCGraph graph, PetscInt custom_minimal_size, IS neumann_is, IS dirichlet_is, PetscInt n_ISForDofs, IS ISForDofs[], IS custom_primal_vertices) 585674ae819SStefano Zampini { 586674ae819SStefano Zampini VecScatter scatter_ctx; 587674ae819SStefano Zampini Vec local_vec,local_vec2,global_vec; 588674ae819SStefano Zampini IS to,from; 589674ae819SStefano Zampini MPI_Comm comm; 590674ae819SStefano Zampini PetscScalar *array,*array2; 591674ae819SStefano Zampini const PetscInt *is_indices; 592674ae819SStefano Zampini PetscInt n_neigh,*neigh,*n_shared,**shared; 593674ae819SStefano Zampini PetscInt i,j,k,s,total_counts,nodes_touched,is_size; 594674ae819SStefano Zampini PetscErrorCode ierr; 595674ae819SStefano Zampini PetscBool same_set; 596674ae819SStefano Zampini 597674ae819SStefano Zampini PetscFunctionBegin; 598674ae819SStefano Zampini ierr = PetscObjectGetComm((PetscObject)(graph->l2gmap),&comm);CHKERRQ(ierr); 599674ae819SStefano Zampini /* custom_minimal_size */ 600674ae819SStefano Zampini graph->custom_minimal_size = PetscMax(graph->custom_minimal_size,custom_minimal_size); 601674ae819SStefano Zampini /* get info l2gmap and allocate work vectors */ 602674ae819SStefano Zampini ierr = ISLocalToGlobalMappingGetInfo(graph->l2gmap,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr); 603674ae819SStefano Zampini ierr = ISLocalToGlobalMappingGetIndices(graph->l2gmap,&is_indices);CHKERRQ(ierr); 604674ae819SStefano Zampini j = 0; 605674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 606674ae819SStefano Zampini j = PetscMax(j,is_indices[i]); 607674ae819SStefano Zampini } 608674ae819SStefano Zampini ierr = MPI_Allreduce(&j,&i,1,MPIU_INT,MPI_MAX,comm);CHKERRQ(ierr); 609674ae819SStefano Zampini i++; 610674ae819SStefano Zampini ierr = VecCreate(PETSC_COMM_SELF,&local_vec);CHKERRQ(ierr); 611674ae819SStefano Zampini ierr = VecSetSizes(local_vec,PETSC_DECIDE,graph->nvtxs);CHKERRQ(ierr); 612674ae819SStefano Zampini ierr = VecSetType(local_vec,VECSTANDARD);CHKERRQ(ierr); 613674ae819SStefano Zampini ierr = VecDuplicate(local_vec,&local_vec2);CHKERRQ(ierr); 614674ae819SStefano Zampini ierr = VecCreate(comm,&global_vec);CHKERRQ(ierr); 615674ae819SStefano Zampini ierr = VecSetSizes(global_vec,PETSC_DECIDE,i);CHKERRQ(ierr); 616674ae819SStefano Zampini ierr = VecSetType(global_vec,VECSTANDARD);CHKERRQ(ierr); 617674ae819SStefano Zampini ierr = ISCreateStride(PETSC_COMM_SELF,graph->nvtxs,0,1,&to);CHKERRQ(ierr); 618674ae819SStefano Zampini ierr = ISLocalToGlobalMappingApplyIS(graph->l2gmap,to,&from);CHKERRQ(ierr); 619674ae819SStefano Zampini ierr = VecScatterCreate(global_vec,from,local_vec,to,&scatter_ctx);CHKERRQ(ierr); 620674ae819SStefano Zampini ierr = ISDestroy(&to);CHKERRQ(ierr); 621674ae819SStefano Zampini ierr = ISDestroy(&from);CHKERRQ(ierr); 622674ae819SStefano Zampini /* Count total number of neigh per node */ 623674ae819SStefano Zampini k=0; 624674ae819SStefano Zampini for (i=1;i<n_neigh;i++){ 625674ae819SStefano Zampini k += n_shared[i]; 626674ae819SStefano Zampini for (j=0;j<n_shared[i];j++){ 627674ae819SStefano Zampini graph->count[shared[i][j]] += 1; 628674ae819SStefano Zampini } 629674ae819SStefano Zampini } 630674ae819SStefano Zampini /* Allocate space for storing the set of neighbours for each node */ 631674ae819SStefano Zampini if (graph->nvtxs) { 632674ae819SStefano Zampini ierr = PetscMalloc(k*sizeof(PetscInt),&graph->neighbours_set[0]);CHKERRQ(ierr); 633674ae819SStefano Zampini } 634674ae819SStefano Zampini for (i=1;i<graph->nvtxs;i++) { /* dont count myself */ 635674ae819SStefano Zampini graph->neighbours_set[i]=graph->neighbours_set[i-1]+graph->count[i-1]; 636674ae819SStefano Zampini } 637674ae819SStefano Zampini /* Get information for sharing subdomains */ 638674ae819SStefano Zampini ierr = PetscMemzero(graph->count,graph->nvtxs*sizeof(*graph->count));CHKERRQ(ierr); 639674ae819SStefano Zampini for (i=1;i<n_neigh;i++) { /* dont count myself */ 640674ae819SStefano Zampini s = n_shared[i]; 641674ae819SStefano Zampini for (j=0;j<s;j++) { 642674ae819SStefano Zampini k = shared[i][j]; 643674ae819SStefano Zampini graph->neighbours_set[k][graph->count[k]] = neigh[i]; 644674ae819SStefano Zampini graph->count[k] += 1; 645674ae819SStefano Zampini } 646674ae819SStefano Zampini } 647674ae819SStefano Zampini /* sort set of sharing subdomains */ 648674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 649674ae819SStefano Zampini ierr = PetscSortInt(graph->count[i],graph->neighbours_set[i]);CHKERRQ(ierr); 650674ae819SStefano Zampini } 651674ae819SStefano Zampini /* Get info for dofs splitting */ 652674ae819SStefano Zampini for (i=0;i<n_ISForDofs;i++) { 653674ae819SStefano Zampini ierr = ISGetSize(ISForDofs[i],&is_size);CHKERRQ(ierr); 654674ae819SStefano Zampini ierr = ISGetIndices(ISForDofs[i],(const PetscInt**)&is_indices);CHKERRQ(ierr); 655674ae819SStefano Zampini for (j=0;j<is_size;j++) { 656674ae819SStefano Zampini graph->which_dof[is_indices[j]]=i; 657674ae819SStefano Zampini } 658674ae819SStefano Zampini ierr = ISRestoreIndices(ISForDofs[i],(const PetscInt**)&is_indices);CHKERRQ(ierr); 659674ae819SStefano Zampini } 660674ae819SStefano Zampini /* Take into account Neumann nodes */ 661674ae819SStefano Zampini ierr = VecSet(local_vec,0.0);CHKERRQ(ierr); 662674ae819SStefano Zampini ierr = VecSet(local_vec2,0.0);CHKERRQ(ierr); 663674ae819SStefano Zampini if (neumann_is) { 664674ae819SStefano Zampini ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr); 665674ae819SStefano Zampini ierr = ISGetSize(neumann_is,&is_size);CHKERRQ(ierr); 666674ae819SStefano Zampini ierr = ISGetIndices(neumann_is,(const PetscInt**)&is_indices);CHKERRQ(ierr); 667674ae819SStefano Zampini for (i=0;i<is_size;i++) { 668674ae819SStefano Zampini array[is_indices[i]] = 1.0; 669674ae819SStefano Zampini } 670674ae819SStefano Zampini ierr = ISRestoreIndices(neumann_is,(const PetscInt**)&is_indices);CHKERRQ(ierr); 671674ae819SStefano Zampini ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr); 672674ae819SStefano Zampini } 673674ae819SStefano Zampini /* Neumann nodes: impose consistency among neighbours */ 674674ae819SStefano Zampini ierr = VecSet(global_vec,0.0);CHKERRQ(ierr); 675674ae819SStefano Zampini ierr = VecScatterBegin(scatter_ctx,local_vec,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 676674ae819SStefano Zampini ierr = VecScatterEnd(scatter_ctx,local_vec,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 677674ae819SStefano Zampini ierr = VecScatterBegin(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 678674ae819SStefano Zampini ierr = VecScatterEnd(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 679674ae819SStefano Zampini ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr); 680674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 681674ae819SStefano Zampini if (array[i] > 0.0) { 682674ae819SStefano Zampini graph->which_dof[i] = NEUMANN_MARK; 683674ae819SStefano Zampini } 684674ae819SStefano Zampini } 685674ae819SStefano Zampini ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr); 686674ae819SStefano Zampini /* Take into account Dirichlet nodes */ 687674ae819SStefano Zampini ierr = VecSet(local_vec2,0.0);CHKERRQ(ierr); 688674ae819SStefano Zampini if (dirichlet_is) { 689674ae819SStefano Zampini ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr); 690674ae819SStefano Zampini ierr = VecGetArray(local_vec2,&array2);CHKERRQ(ierr); 691674ae819SStefano Zampini ierr = ISGetSize(dirichlet_is,&is_size);CHKERRQ(ierr); 692674ae819SStefano Zampini ierr = ISGetIndices(dirichlet_is,(const PetscInt**)&is_indices);CHKERRQ(ierr); 693674ae819SStefano Zampini for (i=0;i<is_size;i++){ 694674ae819SStefano Zampini k = is_indices[i]; 695674ae819SStefano Zampini if (graph->count[k] && !graph->touched[k]) { 696674ae819SStefano Zampini if (array[k] > 0.0) { 697674ae819SStefano Zampini SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_USER,"BDDC cannot have boundary nodes which are marked Neumann and Dirichlet at the same time! Local node %d is wrong!\n",k); 698674ae819SStefano Zampini } 699674ae819SStefano Zampini array2[k] = 1.0; 700674ae819SStefano Zampini } 701674ae819SStefano Zampini } 702674ae819SStefano Zampini ierr = ISRestoreIndices(dirichlet_is,(const PetscInt**)&is_indices);CHKERRQ(ierr); 703674ae819SStefano Zampini ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr); 704674ae819SStefano Zampini ierr = VecRestoreArray(local_vec2,&array2);CHKERRQ(ierr); 705674ae819SStefano Zampini } 706674ae819SStefano Zampini /* Dirichlet nodes: impose consistency among neighbours */ 707674ae819SStefano Zampini ierr = VecSet(global_vec,0.0);CHKERRQ(ierr); 708674ae819SStefano Zampini ierr = VecScatterBegin(scatter_ctx,local_vec2,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 709674ae819SStefano Zampini ierr = VecScatterEnd(scatter_ctx,local_vec2,global_vec,ADD_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 710674ae819SStefano Zampini ierr = VecScatterBegin(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 711674ae819SStefano Zampini ierr = VecScatterEnd(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 712674ae819SStefano Zampini ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr); 713674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 714674ae819SStefano Zampini if (array[i] > 0.0) { 715674ae819SStefano Zampini graph->touched[i] = PETSC_TRUE; 716674ae819SStefano Zampini graph->subset[i] = 0; /* dirichlet nodes treated as internal -> is it ok? */ 717674ae819SStefano Zampini graph->which_dof[i] = DIRICHLET_MARK; 718674ae819SStefano Zampini } 719674ae819SStefano Zampini } 720674ae819SStefano Zampini ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr); 721674ae819SStefano Zampini /* mark special nodes -> each will become a single node equivalence class */ 722674ae819SStefano Zampini if (custom_primal_vertices) { 723674ae819SStefano Zampini ierr = ISGetSize(custom_primal_vertices,&is_size);CHKERRQ(ierr); 724674ae819SStefano Zampini ierr = ISGetIndices(custom_primal_vertices,(const PetscInt**)&is_indices);CHKERRQ(ierr); 725674ae819SStefano Zampini for (i=0;i<is_size;i++) { 726674ae819SStefano Zampini graph->which_dof[is_indices[i]] = SPECIAL_MARK-i; 727674ae819SStefano Zampini } 728674ae819SStefano Zampini ierr = ISRestoreIndices(custom_primal_vertices,(const PetscInt**)&is_indices);CHKERRQ(ierr); 729674ae819SStefano Zampini } 730674ae819SStefano Zampini /* mark interior nodes as touched and belonging to partition number 0 */ 731674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 732674ae819SStefano Zampini if (!graph->count[i]) { 733674ae819SStefano Zampini graph->touched[i] = PETSC_TRUE; 734674ae819SStefano Zampini graph->subset[i] = 0; 735674ae819SStefano Zampini } 736674ae819SStefano Zampini } 737674ae819SStefano Zampini /* init graph structure and compute default subsets */ 738674ae819SStefano Zampini nodes_touched=0; 739674ae819SStefano Zampini for (i=0;i<graph->nvtxs;i++) { 740674ae819SStefano Zampini if (graph->touched[i]) { 741674ae819SStefano Zampini nodes_touched++; 742674ae819SStefano Zampini } 743674ae819SStefano Zampini } 744674ae819SStefano Zampini i = 0; 745674ae819SStefano Zampini graph->ncc = 0; 746674ae819SStefano Zampini total_counts = 0; 747674ae819SStefano Zampini while (nodes_touched<graph->nvtxs) { 748674ae819SStefano Zampini /* find first untouched node in local ordering */ 749674ae819SStefano Zampini while (graph->touched[i]) { 750674ae819SStefano Zampini i++; 751674ae819SStefano Zampini } 752674ae819SStefano Zampini graph->touched[i] = PETSC_TRUE; 753674ae819SStefano Zampini graph->subset[i] = graph->ncc+1; 754674ae819SStefano Zampini graph->cptr[graph->ncc] = total_counts; 755674ae819SStefano Zampini graph->queue[total_counts] = i; 756674ae819SStefano Zampini total_counts++; 757674ae819SStefano Zampini nodes_touched++; 758674ae819SStefano Zampini /* now find all other nodes having the same set of sharing subdomains */ 759674ae819SStefano Zampini for (j=i+1;j<graph->nvtxs;j++){ 760674ae819SStefano Zampini /* check for same number of sharing subdomains and dof number */ 761674ae819SStefano Zampini if (!graph->touched[j] && graph->count[i]==graph->count[j] && graph->which_dof[i] == graph->which_dof[j] ){ 762674ae819SStefano Zampini /* check for same set of sharing subdomains */ 763674ae819SStefano Zampini same_set=PETSC_TRUE; 764674ae819SStefano Zampini for (k=0;k<graph->count[j];k++){ 765674ae819SStefano Zampini if (graph->neighbours_set[i][k]!=graph->neighbours_set[j][k]) { 766674ae819SStefano Zampini same_set=PETSC_FALSE; 767674ae819SStefano Zampini } 768674ae819SStefano Zampini } 769674ae819SStefano Zampini /* I found a friend of mine */ 770674ae819SStefano Zampini if (same_set) { 771674ae819SStefano Zampini graph->subset[j]=graph->ncc+1; 772674ae819SStefano Zampini graph->touched[j]=PETSC_TRUE; 773674ae819SStefano Zampini nodes_touched++; 774674ae819SStefano Zampini graph->queue[total_counts] = j; 775674ae819SStefano Zampini total_counts++; 776674ae819SStefano Zampini } 777674ae819SStefano Zampini } 778674ae819SStefano Zampini } 779674ae819SStefano Zampini graph->ncc++; 780674ae819SStefano Zampini } 781674ae819SStefano Zampini /* set default number of subsets (at this point no info on csr graph has been taken into account, so n_subsets = ncc */ 782674ae819SStefano Zampini graph->n_subsets = graph->ncc; 783674ae819SStefano Zampini ierr = PetscMalloc(graph->n_subsets*sizeof(*graph->subset_ncc),&graph->subset_ncc);CHKERRQ(ierr); 784674ae819SStefano Zampini for (i=0;i<graph->n_subsets;i++) { 785674ae819SStefano Zampini graph->subset_ncc[i] = 1; 786674ae819SStefano Zampini } 787674ae819SStefano Zampini /* final pointer */ 788674ae819SStefano Zampini graph->cptr[graph->ncc] = total_counts; 789674ae819SStefano Zampini /* free memory allocated by ISLocalToGlobalMappingGetInfo */ 790674ae819SStefano Zampini ierr = ISLocalToGlobalMappingRestoreInfo(graph->l2gmap,&n_neigh,&neigh,&n_shared,&shared);CHKERRQ(ierr); 791674ae819SStefano Zampini /* free objects */ 792674ae819SStefano Zampini ierr = VecDestroy(&local_vec);CHKERRQ(ierr); 793674ae819SStefano Zampini ierr = VecDestroy(&local_vec2);CHKERRQ(ierr); 794674ae819SStefano Zampini ierr = VecDestroy(&global_vec);CHKERRQ(ierr); 795674ae819SStefano Zampini ierr = VecScatterDestroy(&scatter_ctx);CHKERRQ(ierr); 796674ae819SStefano Zampini PetscFunctionReturn(0); 797674ae819SStefano Zampini } 798674ae819SStefano Zampini 799674ae819SStefano Zampini #undef __FUNCT__ 800674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphResetCSR" 801674ae819SStefano Zampini PetscErrorCode PCBDDCGraphResetCSR(PCBDDCGraph graph) 802674ae819SStefano Zampini { 803674ae819SStefano Zampini PetscErrorCode ierr; 804674ae819SStefano Zampini 805674ae819SStefano Zampini PetscFunctionBegin; 806674ae819SStefano Zampini ierr = PetscFree(graph->xadj);CHKERRQ(ierr); 807674ae819SStefano Zampini ierr = PetscFree(graph->adjncy);CHKERRQ(ierr); 808674ae819SStefano Zampini PetscFunctionReturn(0); 809674ae819SStefano Zampini } 810674ae819SStefano Zampini 811674ae819SStefano Zampini #undef __FUNCT__ 812674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphReset" 813674ae819SStefano Zampini PetscErrorCode PCBDDCGraphReset(PCBDDCGraph graph) 814674ae819SStefano Zampini { 815674ae819SStefano Zampini PetscErrorCode ierr; 816674ae819SStefano Zampini 817674ae819SStefano Zampini PetscFunctionBegin; 818674ae819SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&graph->l2gmap);CHKERRQ(ierr); 819674ae819SStefano Zampini ierr = PetscFree(graph->subset_ncc);CHKERRQ(ierr); 820674ae819SStefano Zampini if (graph->nvtxs) { 821674ae819SStefano Zampini ierr = PetscFree(graph->neighbours_set[0]);CHKERRQ(ierr); 822674ae819SStefano Zampini } 823674ae819SStefano Zampini ierr = PetscFree7(graph->touched, 824674ae819SStefano Zampini graph->count, 825674ae819SStefano Zampini graph->neighbours_set, 826674ae819SStefano Zampini graph->subset, 827674ae819SStefano Zampini graph->which_dof, 828674ae819SStefano Zampini graph->cptr, 829674ae819SStefano Zampini graph->queue);CHKERRQ(ierr); 830674ae819SStefano Zampini graph->nvtxs = 0; 831674ae819SStefano Zampini graph->n_subsets = 0; 832674ae819SStefano Zampini graph->custom_minimal_size = 1; 833674ae819SStefano Zampini PetscFunctionReturn(0); 834674ae819SStefano Zampini } 835674ae819SStefano Zampini 836674ae819SStefano Zampini #undef __FUNCT__ 837674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphInit" 838674ae819SStefano Zampini PetscErrorCode PCBDDCGraphInit(PCBDDCGraph graph, ISLocalToGlobalMapping l2gmap) 839674ae819SStefano Zampini { 840674ae819SStefano Zampini PetscInt n; 841674ae819SStefano Zampini PetscErrorCode ierr; 842674ae819SStefano Zampini 843674ae819SStefano Zampini PetscFunctionBegin; 844674ae819SStefano Zampini PetscValidPointer(graph,1); 845674ae819SStefano Zampini PetscValidHeaderSpecific(l2gmap,IS_LTOGM_CLASSID,2); 846674ae819SStefano Zampini /* reset graph if already allocated */ 847674ae819SStefano Zampini if (graph->nvtxs) { 848674ae819SStefano Zampini ierr = PCBDDCGraphReset(graph);CHKERRQ(ierr); 849674ae819SStefano Zampini } 850674ae819SStefano Zampini /* set number of vertices */ 851674ae819SStefano Zampini ierr = PetscObjectReference((PetscObject)l2gmap);CHKERRQ(ierr); 852674ae819SStefano Zampini graph->l2gmap = l2gmap; 853674ae819SStefano Zampini ierr = ISLocalToGlobalMappingGetSize(l2gmap,&n);CHKERRQ(ierr); 854674ae819SStefano Zampini graph->nvtxs = n; 855674ae819SStefano Zampini /* allocate used space */ 856674ae819SStefano Zampini ierr = PetscMalloc7(graph->nvtxs,PetscBool,&graph->touched, 857674ae819SStefano Zampini graph->nvtxs,PetscInt,&graph->count, 858674ae819SStefano Zampini graph->nvtxs,PetscInt*,&graph->neighbours_set, 859674ae819SStefano Zampini graph->nvtxs,PetscInt,&graph->subset, 860674ae819SStefano Zampini graph->nvtxs,PetscInt,&graph->which_dof, 861674ae819SStefano Zampini graph->nvtxs+1,PetscInt,&graph->cptr, 862674ae819SStefano Zampini graph->nvtxs,PetscInt,&graph->queue);CHKERRQ(ierr); 863674ae819SStefano Zampini /* zeroes memory */ 864674ae819SStefano Zampini ierr = PetscMemzero(graph->touched,graph->nvtxs*sizeof(PetscBool));CHKERRQ(ierr); 865674ae819SStefano Zampini ierr = PetscMemzero(graph->count,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr); 866674ae819SStefano Zampini ierr = PetscMemzero(graph->subset,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr); 867674ae819SStefano Zampini ierr = PetscMemzero(graph->which_dof,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr); 868674ae819SStefano Zampini ierr = PetscMemzero(graph->cptr,(graph->nvtxs+1)*sizeof(PetscInt));CHKERRQ(ierr); 869674ae819SStefano Zampini ierr = PetscMemzero(graph->queue,graph->nvtxs*sizeof(PetscInt));CHKERRQ(ierr); 870674ae819SStefano Zampini /* zeroes first pointer to neighbour set */ 871674ae819SStefano Zampini if (graph->nvtxs) { 872674ae819SStefano Zampini graph->neighbours_set[0] = 0; 873674ae819SStefano Zampini } 874674ae819SStefano Zampini /* zeroes workspace for values of ncc */ 875674ae819SStefano Zampini graph->subset_ncc = 0; 876674ae819SStefano Zampini PetscFunctionReturn(0); 877674ae819SStefano Zampini } 878674ae819SStefano Zampini 879674ae819SStefano Zampini #undef __FUNCT__ 880674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphDestroy" 881674ae819SStefano Zampini PetscErrorCode PCBDDCGraphDestroy(PCBDDCGraph* graph) 882674ae819SStefano Zampini { 883674ae819SStefano Zampini PetscErrorCode ierr; 884674ae819SStefano Zampini 885674ae819SStefano Zampini PetscFunctionBegin; 886674ae819SStefano Zampini ierr = PCBDDCGraphReset(*graph);CHKERRQ(ierr); 887674ae819SStefano Zampini ierr = PetscFree(*graph);CHKERRQ(ierr); 888674ae819SStefano Zampini PetscFunctionReturn(0); 889674ae819SStefano Zampini } 890674ae819SStefano Zampini 891674ae819SStefano Zampini #undef __FUNCT__ 892674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGraphCreate" 893674ae819SStefano Zampini PetscErrorCode PCBDDCGraphCreate(PCBDDCGraph *graph) 894674ae819SStefano Zampini { 895674ae819SStefano Zampini PCBDDCGraph new_graph; 896674ae819SStefano Zampini PetscErrorCode ierr; 897674ae819SStefano Zampini 898674ae819SStefano Zampini PetscFunctionBegin; 899674ae819SStefano Zampini ierr = PetscMalloc(sizeof(*new_graph),&new_graph);CHKERRQ(ierr); 900674ae819SStefano Zampini /* local to global mapping of dofs */ 901674ae819SStefano Zampini new_graph->l2gmap = 0; 902674ae819SStefano Zampini /* vertex size */ 903674ae819SStefano Zampini new_graph->nvtxs = 0; 904674ae819SStefano Zampini new_graph->n_subsets = 0; 905674ae819SStefano Zampini new_graph->custom_minimal_size = 1; 906674ae819SStefano Zampini /* zeroes ponters */ 907674ae819SStefano Zampini new_graph->neighbours_set = 0; 908674ae819SStefano Zampini new_graph->subset = 0; 909674ae819SStefano Zampini new_graph->which_dof = 0; 910674ae819SStefano Zampini new_graph->cptr = 0; 911674ae819SStefano Zampini new_graph->queue = 0; 912674ae819SStefano Zampini new_graph->count = 0; 913674ae819SStefano Zampini new_graph->subset_ncc = 0; 914674ae819SStefano Zampini new_graph->touched = 0; 915674ae819SStefano Zampini /* zeroes pointers to csr graph of local nodes connectivity (optional data) */ 916674ae819SStefano Zampini new_graph->xadj = 0; 917674ae819SStefano Zampini new_graph->adjncy = 0; 918674ae819SStefano Zampini *graph = new_graph; 919674ae819SStefano Zampini PetscFunctionReturn(0); 920674ae819SStefano Zampini } 921