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