1674ae819SStefano Zampini #include "bddc.h" 2674ae819SStefano Zampini #include "bddcprivate.h" 3674ae819SStefano Zampini #include <petscblaslapack.h> 4674ae819SStefano Zampini 5674ae819SStefano Zampini #undef __FUNCT__ 6674ae819SStefano Zampini #define __FUNCT__ "PCBDDCResetCustomization" 7674ae819SStefano Zampini PetscErrorCode PCBDDCResetCustomization(PC pc) 8674ae819SStefano Zampini { 9674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 10674ae819SStefano Zampini PetscInt i; 11674ae819SStefano Zampini PetscErrorCode ierr; 12674ae819SStefano Zampini 13674ae819SStefano Zampini PetscFunctionBegin; 14674ae819SStefano Zampini ierr = PCBDDCGraphResetCSR(pcbddc->mat_graph);CHKERRQ(ierr); 15674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->user_primal_vertices);CHKERRQ(ierr); 16674ae819SStefano Zampini ierr = MatNullSpaceDestroy(&pcbddc->NullSpace);CHKERRQ(ierr); 17674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->NeumannBoundaries);CHKERRQ(ierr); 18674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->DirichletBoundaries);CHKERRQ(ierr); 19674ae819SStefano Zampini for (i=0;i<pcbddc->n_ISForDofs;i++) { 20674ae819SStefano Zampini ierr = ISDestroy(&pcbddc->ISForDofs[i]);CHKERRQ(ierr); 21674ae819SStefano Zampini } 22674ae819SStefano Zampini ierr = PetscFree(pcbddc->ISForDofs);CHKERRQ(ierr); 23674ae819SStefano Zampini PetscFunctionReturn(0); 24674ae819SStefano Zampini } 25674ae819SStefano Zampini 26674ae819SStefano Zampini #undef __FUNCT__ 27674ae819SStefano Zampini #define __FUNCT__ "PCBDDCResetTopography" 28674ae819SStefano Zampini PetscErrorCode PCBDDCResetTopography(PC pc) 29674ae819SStefano Zampini { 30674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 31674ae819SStefano Zampini PetscErrorCode ierr; 32674ae819SStefano Zampini 33674ae819SStefano Zampini PetscFunctionBegin; 34674ae819SStefano Zampini ierr = MatDestroy(&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 35674ae819SStefano Zampini ierr = MatDestroy(&pcbddc->ConstraintMatrix);CHKERRQ(ierr); 36674ae819SStefano Zampini ierr = PCBDDCGraphReset(pcbddc->mat_graph);CHKERRQ(ierr); 37674ae819SStefano Zampini PetscFunctionReturn(0); 38674ae819SStefano Zampini } 39674ae819SStefano Zampini 40674ae819SStefano Zampini #undef __FUNCT__ 41674ae819SStefano Zampini #define __FUNCT__ "PCBDDCResetSolvers" 42674ae819SStefano Zampini PetscErrorCode PCBDDCResetSolvers(PC pc) 43674ae819SStefano Zampini { 44674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 45674ae819SStefano Zampini PetscErrorCode ierr; 46674ae819SStefano Zampini 47674ae819SStefano Zampini PetscFunctionBegin; 48674ae819SStefano Zampini ierr = VecDestroy(&pcbddc->coarse_vec);CHKERRQ(ierr); 49674ae819SStefano Zampini ierr = VecDestroy(&pcbddc->coarse_rhs);CHKERRQ(ierr); 50674ae819SStefano Zampini ierr = KSPDestroy(&pcbddc->coarse_ksp);CHKERRQ(ierr); 51674ae819SStefano Zampini ierr = MatDestroy(&pcbddc->coarse_mat);CHKERRQ(ierr); 52674ae819SStefano Zampini ierr = MatDestroy(&pcbddc->coarse_phi_B);CHKERRQ(ierr); 53674ae819SStefano Zampini ierr = MatDestroy(&pcbddc->coarse_phi_D);CHKERRQ(ierr); 5415aaf578SStefano Zampini ierr = MatDestroy(&pcbddc->coarse_psi_B);CHKERRQ(ierr); 5515aaf578SStefano Zampini ierr = MatDestroy(&pcbddc->coarse_psi_D);CHKERRQ(ierr); 56674ae819SStefano Zampini ierr = VecDestroy(&pcbddc->vec1_P);CHKERRQ(ierr); 57674ae819SStefano Zampini ierr = VecDestroy(&pcbddc->vec1_C);CHKERRQ(ierr); 58674ae819SStefano Zampini ierr = MatDestroy(&pcbddc->local_auxmat1);CHKERRQ(ierr); 59674ae819SStefano Zampini ierr = MatDestroy(&pcbddc->local_auxmat2);CHKERRQ(ierr); 60674ae819SStefano Zampini ierr = VecDestroy(&pcbddc->vec1_R);CHKERRQ(ierr); 61674ae819SStefano Zampini ierr = VecDestroy(&pcbddc->vec2_R);CHKERRQ(ierr); 62674ae819SStefano Zampini ierr = VecDestroy(&pcbddc->vec4_D);CHKERRQ(ierr); 63674ae819SStefano Zampini ierr = VecScatterDestroy(&pcbddc->R_to_B);CHKERRQ(ierr); 64674ae819SStefano Zampini ierr = VecScatterDestroy(&pcbddc->R_to_D);CHKERRQ(ierr); 65674ae819SStefano Zampini ierr = VecScatterDestroy(&pcbddc->coarse_loc_to_glob);CHKERRQ(ierr); 66674ae819SStefano Zampini ierr = PetscFree(pcbddc->local_primal_indices);CHKERRQ(ierr); 67674ae819SStefano Zampini ierr = PetscFree(pcbddc->replicated_local_primal_indices);CHKERRQ(ierr); 68674ae819SStefano Zampini ierr = PetscFree(pcbddc->replicated_local_primal_values);CHKERRQ(ierr); 69674ae819SStefano Zampini ierr = PetscFree(pcbddc->local_primal_displacements);CHKERRQ(ierr); 70674ae819SStefano Zampini ierr = PetscFree(pcbddc->local_primal_sizes);CHKERRQ(ierr); 71674ae819SStefano Zampini PetscFunctionReturn(0); 72674ae819SStefano Zampini } 73674ae819SStefano Zampini 74674ae819SStefano Zampini #undef __FUNCT__ 756bfb1811SStefano Zampini #define __FUNCT__ "PCBDDCCreateWorkVectors" 766bfb1811SStefano Zampini PetscErrorCode PCBDDCCreateWorkVectors(PC pc) 776bfb1811SStefano Zampini { 786bfb1811SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 796bfb1811SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 806bfb1811SStefano Zampini VecType impVecType; 816bfb1811SStefano Zampini PetscInt n_vertices,n_constraints,local_primal_size,n_R; 826bfb1811SStefano Zampini PetscErrorCode ierr; 836bfb1811SStefano Zampini 846bfb1811SStefano Zampini PetscFunctionBegin; 856bfb1811SStefano Zampini ierr = PCBDDCGetPrimalVerticesLocalIdx(pc,&n_vertices,NULL);CHKERRQ(ierr); 866bfb1811SStefano Zampini ierr = PCBDDCGetPrimalConstraintsLocalIdx(pc,&n_constraints,NULL);CHKERRQ(ierr); 876bfb1811SStefano Zampini local_primal_size = n_constraints+n_vertices; 886bfb1811SStefano Zampini n_R = pcis->n-n_vertices; 896bfb1811SStefano Zampini /* local work vectors */ 906bfb1811SStefano Zampini ierr = VecGetType(pcis->vec1_N,&impVecType);CHKERRQ(ierr); 916bfb1811SStefano Zampini ierr = VecDuplicate(pcis->vec1_D,&pcbddc->vec4_D);CHKERRQ(ierr); 926bfb1811SStefano Zampini ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_R);CHKERRQ(ierr); 936bfb1811SStefano Zampini ierr = VecSetSizes(pcbddc->vec1_R,PETSC_DECIDE,n_R);CHKERRQ(ierr); 946bfb1811SStefano Zampini ierr = VecSetType(pcbddc->vec1_R,impVecType);CHKERRQ(ierr); 956bfb1811SStefano Zampini ierr = VecDuplicate(pcbddc->vec1_R,&pcbddc->vec2_R);CHKERRQ(ierr); 9683b7ccabSStefano Zampini ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_P);CHKERRQ(ierr); 976bfb1811SStefano Zampini ierr = VecSetSizes(pcbddc->vec1_P,PETSC_DECIDE,local_primal_size);CHKERRQ(ierr); 986bfb1811SStefano Zampini ierr = VecSetType(pcbddc->vec1_P,impVecType);CHKERRQ(ierr); 9983b7ccabSStefano Zampini if (n_constraints) { 10083b7ccabSStefano Zampini ierr = VecCreate(PetscObjectComm((PetscObject)pcis->vec1_N),&pcbddc->vec1_C);CHKERRQ(ierr); 10183b7ccabSStefano Zampini ierr = VecSetSizes(pcbddc->vec1_C,PETSC_DECIDE,n_constraints);CHKERRQ(ierr); 10283b7ccabSStefano Zampini ierr = VecSetType(pcbddc->vec1_C,impVecType);CHKERRQ(ierr); 10383b7ccabSStefano Zampini } 1046bfb1811SStefano Zampini PetscFunctionReturn(0); 1056bfb1811SStefano Zampini } 1066bfb1811SStefano Zampini 1076bfb1811SStefano Zampini #undef __FUNCT__ 108*88ebb749SStefano Zampini #define __FUNCT__ "PCBDDCSetUpCorrectionAndBasis" 109*88ebb749SStefano Zampini PetscErrorCode PCBDDCSetUpCorrectionAndBasis(PC pc, IS is_R_local) 110*88ebb749SStefano Zampini { 111*88ebb749SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 112*88ebb749SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 113*88ebb749SStefano Zampini PetscErrorCode ierr; 114*88ebb749SStefano Zampini Mat A_RV,A_VR,A_VV; 115*88ebb749SStefano Zampini Mat M1; 116*88ebb749SStefano Zampini Mat C_CR; 117*88ebb749SStefano Zampini Mat AUXMAT; 118*88ebb749SStefano Zampini Vec vec1_C; 119*88ebb749SStefano Zampini Vec vec2_C; 120*88ebb749SStefano Zampini Vec vec1_V; 121*88ebb749SStefano Zampini Vec vec2_V; 122*88ebb749SStefano Zampini IS is_C_local,is_V_local,is_aux1; 123*88ebb749SStefano Zampini ISLocalToGlobalMapping BtoNmap; 124*88ebb749SStefano Zampini PetscInt *nnz; 125*88ebb749SStefano Zampini PetscInt *idx_V_B; 126*88ebb749SStefano Zampini PetscInt *auxindices; 127*88ebb749SStefano Zampini PetscInt index; 128*88ebb749SStefano Zampini PetscScalar* array2; 129*88ebb749SStefano Zampini MatFactorInfo matinfo; 130*88ebb749SStefano Zampini PetscBool setsym=PETSC_FALSE,issym=PETSC_FALSE; 131*88ebb749SStefano Zampini /* old variables */ 132*88ebb749SStefano Zampini VecType impVecType; 133*88ebb749SStefano Zampini MatType impMatType; 134*88ebb749SStefano Zampini PetscInt n_R=0; 135*88ebb749SStefano Zampini PetscInt n_D=0; 136*88ebb749SStefano Zampini PetscInt n_B=0; 137*88ebb749SStefano Zampini PetscScalar zero=0.0; 138*88ebb749SStefano Zampini PetscScalar one=1.0; 139*88ebb749SStefano Zampini PetscScalar m_one=-1.0; 140*88ebb749SStefano Zampini PetscScalar* array; 141*88ebb749SStefano Zampini PetscScalar *coarse_submat_vals; 142*88ebb749SStefano Zampini PetscInt *idx_R_local; 143*88ebb749SStefano Zampini PetscReal *coarsefunctions_errors,*constraints_errors; 144*88ebb749SStefano Zampini /* auxiliary indices */ 145*88ebb749SStefano Zampini PetscInt i,j; 146*88ebb749SStefano Zampini /* for verbose output of bddc */ 147*88ebb749SStefano Zampini PetscViewer viewer=pcbddc->dbg_viewer; 148*88ebb749SStefano Zampini PetscInt dbg_flag=pcbddc->dbg_flag; 149*88ebb749SStefano Zampini /* for counting coarse dofs */ 150*88ebb749SStefano Zampini PetscInt n_vertices,n_constraints; 151*88ebb749SStefano Zampini PetscInt size_of_constraint; 152*88ebb749SStefano Zampini PetscInt *row_cmat_indices; 153*88ebb749SStefano Zampini PetscScalar *row_cmat_values; 154*88ebb749SStefano Zampini PetscInt *vertices; 155*88ebb749SStefano Zampini 156*88ebb749SStefano Zampini PetscFunctionBegin; 157*88ebb749SStefano Zampini /* get number of vertices */ 158*88ebb749SStefano Zampini ierr = PCBDDCGetPrimalVerticesLocalIdx(pc,&n_vertices,NULL);CHKERRQ(ierr); 159*88ebb749SStefano Zampini n_constraints = pcbddc->local_primal_size-n_vertices; 160*88ebb749SStefano Zampini /* Set Non-overlapping dimensions */ 161*88ebb749SStefano Zampini n_B = pcis->n_B; n_D = pcis->n - n_B; 162*88ebb749SStefano Zampini n_R = pcis->n-n_vertices; 163*88ebb749SStefano Zampini 164*88ebb749SStefano Zampini /* Set types for local objects needed by BDDC precondtioner */ 165*88ebb749SStefano Zampini impMatType = MATSEQDENSE; 166*88ebb749SStefano Zampini impVecType = VECSEQ; 167*88ebb749SStefano Zampini 168*88ebb749SStefano Zampini /* Allocating some extra storage just to be safe */ 169*88ebb749SStefano Zampini ierr = PetscMalloc (pcis->n*sizeof(PetscInt),&nnz);CHKERRQ(ierr); 170*88ebb749SStefano Zampini ierr = PetscMalloc (pcis->n*sizeof(PetscInt),&auxindices);CHKERRQ(ierr); 171*88ebb749SStefano Zampini for (i=0;i<pcis->n;i++) auxindices[i]=i; 172*88ebb749SStefano Zampini 173*88ebb749SStefano Zampini ierr = PCBDDCGetPrimalVerticesLocalIdx(pc,&n_vertices,&vertices);CHKERRQ(ierr); 174*88ebb749SStefano Zampini /* vertices in boundary numbering */ 175*88ebb749SStefano Zampini ierr = PetscMalloc(n_vertices*sizeof(PetscInt),&idx_V_B);CHKERRQ(ierr); 176*88ebb749SStefano Zampini ierr = ISLocalToGlobalMappingCreateIS(pcis->is_B_local,&BtoNmap);CHKERRQ(ierr); 177*88ebb749SStefano Zampini ierr = ISGlobalToLocalMappingApply(BtoNmap,IS_GTOLM_DROP,n_vertices,vertices,&i,idx_V_B);CHKERRQ(ierr); 178*88ebb749SStefano Zampini ierr = ISLocalToGlobalMappingDestroy(&BtoNmap);CHKERRQ(ierr); 179*88ebb749SStefano Zampini if (i != n_vertices) { 180*88ebb749SStefano Zampini SETERRQ2(PETSC_COMM_SELF,PETSC_ERR_SUP,"Error in boundary numbering for BDDC vertices! %d != %d\n",n_vertices,i); 181*88ebb749SStefano Zampini } 182*88ebb749SStefano Zampini 183*88ebb749SStefano Zampini /* some work vectors on vertices and/or constraints */ 184*88ebb749SStefano Zampini if (n_vertices) { 185*88ebb749SStefano Zampini ierr = VecCreate(PETSC_COMM_SELF,&vec1_V);CHKERRQ(ierr); 186*88ebb749SStefano Zampini ierr = VecSetSizes(vec1_V,n_vertices,n_vertices);CHKERRQ(ierr); 187*88ebb749SStefano Zampini ierr = VecSetType(vec1_V,impVecType);CHKERRQ(ierr); 188*88ebb749SStefano Zampini ierr = VecDuplicate(vec1_V,&vec2_V);CHKERRQ(ierr); 189*88ebb749SStefano Zampini } 190*88ebb749SStefano Zampini if (n_constraints) { 191*88ebb749SStefano Zampini ierr = VecDuplicate(pcbddc->vec1_C,&vec1_C);CHKERRQ(ierr); 192*88ebb749SStefano Zampini ierr = VecDuplicate(pcbddc->vec1_C,&vec2_C);CHKERRQ(ierr); 193*88ebb749SStefano Zampini } 194*88ebb749SStefano Zampini /* Precompute stuffs needed for preprocessing and application of BDDC*/ 195*88ebb749SStefano Zampini if (n_constraints) { 196*88ebb749SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->local_auxmat2);CHKERRQ(ierr); 197*88ebb749SStefano Zampini ierr = MatSetSizes(pcbddc->local_auxmat2,n_R,n_constraints,n_R,n_constraints);CHKERRQ(ierr); 198*88ebb749SStefano Zampini ierr = MatSetType(pcbddc->local_auxmat2,impMatType);CHKERRQ(ierr); 199*88ebb749SStefano Zampini ierr = MatSeqDenseSetPreallocation(pcbddc->local_auxmat2,NULL);CHKERRQ(ierr); 200*88ebb749SStefano Zampini 201*88ebb749SStefano Zampini /* Create Constraint matrix on R nodes: C_{CR} */ 202*88ebb749SStefano Zampini ierr = ISCreateStride(PETSC_COMM_SELF,n_constraints,n_vertices,1,&is_C_local);CHKERRQ(ierr); 203*88ebb749SStefano Zampini ierr = MatGetSubMatrix(pcbddc->ConstraintMatrix,is_C_local,is_R_local,MAT_INITIAL_MATRIX,&C_CR);CHKERRQ(ierr); 204*88ebb749SStefano Zampini ierr = ISDestroy(&is_C_local);CHKERRQ(ierr); 205*88ebb749SStefano Zampini 206*88ebb749SStefano Zampini /* Assemble local_auxmat2 = - A_{RR}^{-1} C^T_{CR} needed by BDDC application */ 207*88ebb749SStefano Zampini for (i=0;i<n_constraints;i++) { 208*88ebb749SStefano Zampini ierr = VecSet(pcbddc->vec1_R,zero);CHKERRQ(ierr); 209*88ebb749SStefano Zampini /* Get row of constraint matrix in R numbering */ 210*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_R,&array);CHKERRQ(ierr); 211*88ebb749SStefano Zampini ierr = MatGetRow(C_CR,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,(const PetscScalar**)&row_cmat_values);CHKERRQ(ierr); 212*88ebb749SStefano Zampini for (j=0;j<size_of_constraint;j++) array[row_cmat_indices[j]] = -row_cmat_values[j]; 213*88ebb749SStefano Zampini ierr = MatRestoreRow(C_CR,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,(const PetscScalar**)&row_cmat_values);CHKERRQ(ierr); 214*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_R,&array);CHKERRQ(ierr); 215*88ebb749SStefano Zampini 216*88ebb749SStefano Zampini /* Solve for row of constraint matrix in R numbering */ 217*88ebb749SStefano Zampini ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr); 218*88ebb749SStefano Zampini 219*88ebb749SStefano Zampini /* Set values */ 220*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec2_R,&array);CHKERRQ(ierr); 221*88ebb749SStefano Zampini ierr = MatSetValues(pcbddc->local_auxmat2,n_R,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr); 222*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec2_R,&array);CHKERRQ(ierr); 223*88ebb749SStefano Zampini } 224*88ebb749SStefano Zampini ierr = MatAssemblyBegin(pcbddc->local_auxmat2,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 225*88ebb749SStefano Zampini ierr = MatAssemblyEnd(pcbddc->local_auxmat2,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 226*88ebb749SStefano Zampini 227*88ebb749SStefano Zampini /* Assemble AUXMAT = ( LUFactor )( -C_{CR} A_{RR}^{-1} C^T_{CR} )^{-1} */ 228*88ebb749SStefano Zampini ierr = MatMatMult(C_CR,pcbddc->local_auxmat2,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&AUXMAT);CHKERRQ(ierr); 229*88ebb749SStefano Zampini ierr = MatFactorInfoInitialize(&matinfo);CHKERRQ(ierr); 230*88ebb749SStefano Zampini ierr = ISCreateStride(PETSC_COMM_SELF,n_constraints,0,1,&is_aux1);CHKERRQ(ierr); 231*88ebb749SStefano Zampini ierr = MatLUFactor(AUXMAT,is_aux1,is_aux1,&matinfo);CHKERRQ(ierr); 232*88ebb749SStefano Zampini ierr = ISDestroy(&is_aux1);CHKERRQ(ierr); 233*88ebb749SStefano Zampini 234*88ebb749SStefano Zampini /* Assemble explicitly M1 = ( C_{CR} A_{RR}^{-1} C^T_{CR} )^{-1} needed in preproc */ 235*88ebb749SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&M1);CHKERRQ(ierr); 236*88ebb749SStefano Zampini ierr = MatSetSizes(M1,n_constraints,n_constraints,n_constraints,n_constraints);CHKERRQ(ierr); 237*88ebb749SStefano Zampini ierr = MatSetType(M1,impMatType);CHKERRQ(ierr); 238*88ebb749SStefano Zampini ierr = MatSeqDenseSetPreallocation(M1,NULL);CHKERRQ(ierr); 239*88ebb749SStefano Zampini for (i=0;i<n_constraints;i++) { 240*88ebb749SStefano Zampini ierr = VecSet(vec1_C,zero);CHKERRQ(ierr); 241*88ebb749SStefano Zampini ierr = VecSetValue(vec1_C,i,one,INSERT_VALUES);CHKERRQ(ierr); 242*88ebb749SStefano Zampini ierr = VecAssemblyBegin(vec1_C);CHKERRQ(ierr); 243*88ebb749SStefano Zampini ierr = VecAssemblyEnd(vec1_C);CHKERRQ(ierr); 244*88ebb749SStefano Zampini ierr = MatSolve(AUXMAT,vec1_C,vec2_C);CHKERRQ(ierr); 245*88ebb749SStefano Zampini ierr = VecScale(vec2_C,m_one);CHKERRQ(ierr); 246*88ebb749SStefano Zampini ierr = VecGetArray(vec2_C,&array);CHKERRQ(ierr); 247*88ebb749SStefano Zampini ierr = MatSetValues(M1,n_constraints,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr); 248*88ebb749SStefano Zampini ierr = VecRestoreArray(vec2_C,&array);CHKERRQ(ierr); 249*88ebb749SStefano Zampini } 250*88ebb749SStefano Zampini ierr = MatAssemblyBegin(M1,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 251*88ebb749SStefano Zampini ierr = MatAssemblyEnd(M1,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 252*88ebb749SStefano Zampini ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); 253*88ebb749SStefano Zampini /* Assemble local_auxmat1 = M1*C_{CR} needed by BDDC application in KSP and in preproc */ 254*88ebb749SStefano Zampini ierr = MatMatMult(M1,C_CR,MAT_INITIAL_MATRIX,PETSC_DEFAULT,&pcbddc->local_auxmat1);CHKERRQ(ierr); 255*88ebb749SStefano Zampini 256*88ebb749SStefano Zampini } 257*88ebb749SStefano Zampini 258*88ebb749SStefano Zampini /* Get submatrices from subdomain matrix */ 259*88ebb749SStefano Zampini if (n_vertices) { 260*88ebb749SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,n_vertices,vertices,PETSC_COPY_VALUES,&is_V_local);CHKERRQ(ierr); 261*88ebb749SStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,is_R_local,is_V_local,MAT_INITIAL_MATRIX,&A_RV);CHKERRQ(ierr); 262*88ebb749SStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,is_V_local,is_R_local,MAT_INITIAL_MATRIX,&A_VR);CHKERRQ(ierr); 263*88ebb749SStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,is_V_local,is_V_local,MAT_INITIAL_MATRIX,&A_VV);CHKERRQ(ierr); 264*88ebb749SStefano Zampini ierr = ISDestroy(&is_V_local);CHKERRQ(ierr); 265*88ebb749SStefano Zampini } 266*88ebb749SStefano Zampini 267*88ebb749SStefano Zampini /* Matrix of coarse basis functions (local) */ 268*88ebb749SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->coarse_phi_B);CHKERRQ(ierr); 269*88ebb749SStefano Zampini ierr = MatSetSizes(pcbddc->coarse_phi_B,n_B,pcbddc->local_primal_size,n_B,pcbddc->local_primal_size);CHKERRQ(ierr); 270*88ebb749SStefano Zampini ierr = MatSetType(pcbddc->coarse_phi_B,impMatType);CHKERRQ(ierr); 271*88ebb749SStefano Zampini ierr = MatSeqDenseSetPreallocation(pcbddc->coarse_phi_B,NULL);CHKERRQ(ierr); 272*88ebb749SStefano Zampini if (pcbddc->inexact_prec_type || dbg_flag ) { 273*88ebb749SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->coarse_phi_D);CHKERRQ(ierr); 274*88ebb749SStefano Zampini ierr = MatSetSizes(pcbddc->coarse_phi_D,n_D,pcbddc->local_primal_size,n_D,pcbddc->local_primal_size);CHKERRQ(ierr); 275*88ebb749SStefano Zampini ierr = MatSetType(pcbddc->coarse_phi_D,impMatType);CHKERRQ(ierr); 276*88ebb749SStefano Zampini ierr = MatSeqDenseSetPreallocation(pcbddc->coarse_phi_D,NULL);CHKERRQ(ierr); 277*88ebb749SStefano Zampini } 278*88ebb749SStefano Zampini 279*88ebb749SStefano Zampini if (dbg_flag) { 280*88ebb749SStefano Zampini ierr = ISGetIndices(is_R_local,(const PetscInt**)&idx_R_local);CHKERRQ(ierr); 281*88ebb749SStefano Zampini ierr = PetscMalloc(2*pcbddc->local_primal_size*sizeof(*coarsefunctions_errors),&coarsefunctions_errors);CHKERRQ(ierr); 282*88ebb749SStefano Zampini ierr = PetscMalloc(2*pcbddc->local_primal_size*sizeof(*constraints_errors),&constraints_errors);CHKERRQ(ierr); 283*88ebb749SStefano Zampini } 284*88ebb749SStefano Zampini /* Subdomain contribution (Non-overlapping) to coarse matrix */ 285*88ebb749SStefano Zampini ierr = PetscMalloc ((pcbddc->local_primal_size)*(pcbddc->local_primal_size)*sizeof(PetscScalar),&coarse_submat_vals);CHKERRQ(ierr); 286*88ebb749SStefano Zampini 287*88ebb749SStefano Zampini /* We are now ready to evaluate coarse basis functions and subdomain contribution to coarse problem */ 288*88ebb749SStefano Zampini for (i=0;i<n_vertices;i++){ 289*88ebb749SStefano Zampini ierr = VecSet(vec1_V,zero);CHKERRQ(ierr); 290*88ebb749SStefano Zampini ierr = VecSetValue(vec1_V,i,one,INSERT_VALUES);CHKERRQ(ierr); 291*88ebb749SStefano Zampini ierr = VecAssemblyBegin(vec1_V);CHKERRQ(ierr); 292*88ebb749SStefano Zampini ierr = VecAssemblyEnd(vec1_V);CHKERRQ(ierr); 293*88ebb749SStefano Zampini /* solution of saddle point problem */ 294*88ebb749SStefano Zampini ierr = MatMult(A_RV,vec1_V,pcbddc->vec1_R);CHKERRQ(ierr); 295*88ebb749SStefano Zampini ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr); 296*88ebb749SStefano Zampini ierr = VecScale(pcbddc->vec1_R,m_one);CHKERRQ(ierr); 297*88ebb749SStefano Zampini if (n_constraints) { 298*88ebb749SStefano Zampini ierr = MatMult(pcbddc->local_auxmat1,pcbddc->vec1_R,vec1_C);CHKERRQ(ierr); 299*88ebb749SStefano Zampini ierr = MatMultAdd(pcbddc->local_auxmat2,vec1_C,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr); 300*88ebb749SStefano Zampini ierr = VecScale(vec1_C,m_one);CHKERRQ(ierr); 301*88ebb749SStefano Zampini } 302*88ebb749SStefano Zampini ierr = MatMult(A_VR,pcbddc->vec1_R,vec2_V);CHKERRQ(ierr); 303*88ebb749SStefano Zampini ierr = MatMultAdd(A_VV,vec1_V,vec2_V,vec2_V);CHKERRQ(ierr); 304*88ebb749SStefano Zampini 305*88ebb749SStefano Zampini /* Set values in coarse basis function and subdomain part of coarse_mat */ 306*88ebb749SStefano Zampini /* coarse basis functions */ 307*88ebb749SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 308*88ebb749SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 309*88ebb749SStefano Zampini ierr = VecScatterEnd (pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 310*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_B,&array);CHKERRQ(ierr); 311*88ebb749SStefano Zampini ierr = MatSetValues(pcbddc->coarse_phi_B,n_B,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr); 312*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_B,&array);CHKERRQ(ierr); 313*88ebb749SStefano Zampini ierr = MatSetValue(pcbddc->coarse_phi_B,idx_V_B[i],i,one,INSERT_VALUES);CHKERRQ(ierr); 314*88ebb749SStefano Zampini if ( pcbddc->inexact_prec_type || dbg_flag ) { 315*88ebb749SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 316*88ebb749SStefano Zampini ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 317*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_D,&array);CHKERRQ(ierr); 318*88ebb749SStefano Zampini ierr = MatSetValues(pcbddc->coarse_phi_D,n_D,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr); 319*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_D,&array);CHKERRQ(ierr); 320*88ebb749SStefano Zampini } 321*88ebb749SStefano Zampini /* subdomain contribution to coarse matrix */ 322*88ebb749SStefano Zampini ierr = VecGetArray(vec2_V,&array);CHKERRQ(ierr); 323*88ebb749SStefano Zampini for (j=0; j<n_vertices; j++) coarse_submat_vals[i*pcbddc->local_primal_size+j] = array[j]; /* WARNING -> column major ordering */ 324*88ebb749SStefano Zampini ierr = VecRestoreArray(vec2_V,&array);CHKERRQ(ierr); 325*88ebb749SStefano Zampini if (n_constraints) { 326*88ebb749SStefano Zampini ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr); 327*88ebb749SStefano Zampini for (j=0; j<n_constraints; j++) coarse_submat_vals[i*pcbddc->local_primal_size+j+n_vertices] = array[j]; /* WARNING -> column major ordering */ 328*88ebb749SStefano Zampini ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr); 329*88ebb749SStefano Zampini } 330*88ebb749SStefano Zampini 331*88ebb749SStefano Zampini if ( dbg_flag ) { 332*88ebb749SStefano Zampini /* assemble subdomain vector on nodes */ 333*88ebb749SStefano Zampini ierr = VecSet(pcis->vec1_N,zero);CHKERRQ(ierr); 334*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr); 335*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr); 336*88ebb749SStefano Zampini for (j=0;j<n_R;j++) array[idx_R_local[j]] = array2[j]; 337*88ebb749SStefano Zampini array[ vertices[i] ] = one; 338*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr); 339*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr); 340*88ebb749SStefano Zampini /* assemble subdomain vector of lagrange multipliers (i.e. primal nodes) */ 341*88ebb749SStefano Zampini ierr = VecSet(pcbddc->vec1_P,zero);CHKERRQ(ierr); 342*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr); 343*88ebb749SStefano Zampini ierr = VecGetArray(vec2_V,&array);CHKERRQ(ierr); 344*88ebb749SStefano Zampini for (j=0;j<n_vertices;j++) array2[j]=array[j]; 345*88ebb749SStefano Zampini ierr = VecRestoreArray(vec2_V,&array);CHKERRQ(ierr); 346*88ebb749SStefano Zampini if (n_constraints) { 347*88ebb749SStefano Zampini ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr); 348*88ebb749SStefano Zampini for (j=0;j<n_constraints;j++) array2[j+n_vertices]=array[j]; 349*88ebb749SStefano Zampini ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr); 350*88ebb749SStefano Zampini } 351*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr); 352*88ebb749SStefano Zampini ierr = VecScale(pcbddc->vec1_P,m_one);CHKERRQ(ierr); 353*88ebb749SStefano Zampini /* check saddle point solution */ 354*88ebb749SStefano Zampini ierr = MatMult(pcbddc->local_mat,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 355*88ebb749SStefano Zampini ierr = MatMultTransposeAdd(pcbddc->ConstraintMatrix,pcbddc->vec1_P,pcis->vec2_N,pcis->vec2_N);CHKERRQ(ierr); 356*88ebb749SStefano Zampini ierr = VecNorm(pcis->vec2_N,NORM_INFINITY,&coarsefunctions_errors[i]);CHKERRQ(ierr); 357*88ebb749SStefano Zampini ierr = MatMult(pcbddc->ConstraintMatrix,pcis->vec1_N,pcbddc->vec1_P);CHKERRQ(ierr); 358*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 359*88ebb749SStefano Zampini array[i]=array[i]+m_one; /* shift by the identity matrix */ 360*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 361*88ebb749SStefano Zampini ierr = VecNorm(pcbddc->vec1_P,NORM_INFINITY,&constraints_errors[i]);CHKERRQ(ierr); 362*88ebb749SStefano Zampini } 363*88ebb749SStefano Zampini } 364*88ebb749SStefano Zampini 365*88ebb749SStefano Zampini for (i=0;i<n_constraints;i++){ 366*88ebb749SStefano Zampini ierr = VecSet(vec2_C,zero);CHKERRQ(ierr); 367*88ebb749SStefano Zampini ierr = VecSetValue(vec2_C,i,m_one,INSERT_VALUES);CHKERRQ(ierr); 368*88ebb749SStefano Zampini ierr = VecAssemblyBegin(vec2_C);CHKERRQ(ierr); 369*88ebb749SStefano Zampini ierr = VecAssemblyEnd(vec2_C);CHKERRQ(ierr); 370*88ebb749SStefano Zampini /* solution of saddle point problem */ 371*88ebb749SStefano Zampini ierr = MatMult(M1,vec2_C,vec1_C);CHKERRQ(ierr); 372*88ebb749SStefano Zampini ierr = MatMult(pcbddc->local_auxmat2,vec1_C,pcbddc->vec1_R);CHKERRQ(ierr); 373*88ebb749SStefano Zampini ierr = VecScale(vec1_C,m_one);CHKERRQ(ierr); 374*88ebb749SStefano Zampini if (n_vertices) { ierr = MatMult(A_VR,pcbddc->vec1_R,vec2_V);CHKERRQ(ierr); } 375*88ebb749SStefano Zampini /* Set values in coarse basis function and subdomain part of coarse_mat */ 376*88ebb749SStefano Zampini /* coarse basis functions */ 377*88ebb749SStefano Zampini index=i+n_vertices; 378*88ebb749SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 379*88ebb749SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 380*88ebb749SStefano Zampini ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 381*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_B,&array);CHKERRQ(ierr); 382*88ebb749SStefano Zampini ierr = MatSetValues(pcbddc->coarse_phi_B,n_B,auxindices,1,&index,array,INSERT_VALUES);CHKERRQ(ierr); 383*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_B,&array);CHKERRQ(ierr); 384*88ebb749SStefano Zampini if ( pcbddc->inexact_prec_type || dbg_flag ) { 385*88ebb749SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 386*88ebb749SStefano Zampini ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 387*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_D,&array);CHKERRQ(ierr); 388*88ebb749SStefano Zampini ierr = MatSetValues(pcbddc->coarse_phi_D,n_D,auxindices,1,&index,array,INSERT_VALUES);CHKERRQ(ierr); 389*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_D,&array);CHKERRQ(ierr); 390*88ebb749SStefano Zampini } 391*88ebb749SStefano Zampini /* subdomain contribution to coarse matrix */ 392*88ebb749SStefano Zampini if (n_vertices) { 393*88ebb749SStefano Zampini ierr = VecGetArray(vec2_V,&array);CHKERRQ(ierr); 394*88ebb749SStefano Zampini for (j=0; j<n_vertices; j++) coarse_submat_vals[index*pcbddc->local_primal_size+j]=array[j]; /* WARNING -> column major ordering */ 395*88ebb749SStefano Zampini ierr = VecRestoreArray(vec2_V,&array);CHKERRQ(ierr); 396*88ebb749SStefano Zampini } 397*88ebb749SStefano Zampini ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr); 398*88ebb749SStefano Zampini for (j=0; j<n_constraints; j++) coarse_submat_vals[index*pcbddc->local_primal_size+j+n_vertices]=array[j]; /* WARNING -> column major ordering */ 399*88ebb749SStefano Zampini ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr); 400*88ebb749SStefano Zampini 401*88ebb749SStefano Zampini if ( dbg_flag ) { 402*88ebb749SStefano Zampini /* assemble subdomain vector on nodes */ 403*88ebb749SStefano Zampini ierr = VecSet(pcis->vec1_N,zero);CHKERRQ(ierr); 404*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr); 405*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr); 406*88ebb749SStefano Zampini for (j=0;j<n_R;j++) array[idx_R_local[j]] = array2[j]; 407*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr); 408*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr); 409*88ebb749SStefano Zampini /* assemble subdomain vector of lagrange multipliers */ 410*88ebb749SStefano Zampini ierr = VecSet(pcbddc->vec1_P,zero);CHKERRQ(ierr); 411*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr); 412*88ebb749SStefano Zampini if ( n_vertices) { 413*88ebb749SStefano Zampini ierr = VecGetArray(vec2_V,&array);CHKERRQ(ierr); 414*88ebb749SStefano Zampini for (j=0;j<n_vertices;j++) array2[j]=-array[j]; 415*88ebb749SStefano Zampini ierr = VecRestoreArray(vec2_V,&array);CHKERRQ(ierr); 416*88ebb749SStefano Zampini } 417*88ebb749SStefano Zampini ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr); 418*88ebb749SStefano Zampini for (j=0;j<n_constraints;j++) {array2[j+n_vertices]=-array[j];} 419*88ebb749SStefano Zampini ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr); 420*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array2);CHKERRQ(ierr); 421*88ebb749SStefano Zampini /* check saddle point solution */ 422*88ebb749SStefano Zampini ierr = MatMult(pcbddc->local_mat,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 423*88ebb749SStefano Zampini ierr = MatMultTransposeAdd(pcbddc->ConstraintMatrix,pcbddc->vec1_P,pcis->vec2_N,pcis->vec2_N);CHKERRQ(ierr); 424*88ebb749SStefano Zampini ierr = VecNorm(pcis->vec2_N,NORM_INFINITY,&coarsefunctions_errors[index]);CHKERRQ(ierr); 425*88ebb749SStefano Zampini ierr = MatMult(pcbddc->ConstraintMatrix,pcis->vec1_N,pcbddc->vec1_P);CHKERRQ(ierr); 426*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 427*88ebb749SStefano Zampini array[index]=array[index]+m_one; /* shift by the identity matrix */ 428*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 429*88ebb749SStefano Zampini ierr = VecNorm(pcbddc->vec1_P,NORM_INFINITY,&constraints_errors[index]);CHKERRQ(ierr); 430*88ebb749SStefano Zampini } 431*88ebb749SStefano Zampini } 432*88ebb749SStefano Zampini ierr = MatAssemblyBegin(pcbddc->coarse_phi_B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 433*88ebb749SStefano Zampini ierr = MatAssemblyEnd(pcbddc->coarse_phi_B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 434*88ebb749SStefano Zampini if ( pcbddc->inexact_prec_type || dbg_flag ) { 435*88ebb749SStefano Zampini ierr = MatAssemblyBegin(pcbddc->coarse_phi_D,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 436*88ebb749SStefano Zampini ierr = MatAssemblyEnd(pcbddc->coarse_phi_D,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 437*88ebb749SStefano Zampini } 438*88ebb749SStefano Zampini /* compute other basis functions for non-symmetric problems */ 439*88ebb749SStefano Zampini ierr = MatIsSymmetricKnown(pc->pmat,&setsym,&issym);CHKERRQ(ierr); 440*88ebb749SStefano Zampini if ( !setsym || (setsym && !issym) ) { 441*88ebb749SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->coarse_psi_B);CHKERRQ(ierr); 442*88ebb749SStefano Zampini ierr = MatSetSizes(pcbddc->coarse_psi_B,n_B,pcbddc->local_primal_size,n_B,pcbddc->local_primal_size);CHKERRQ(ierr); 443*88ebb749SStefano Zampini ierr = MatSetType(pcbddc->coarse_psi_B,impMatType);CHKERRQ(ierr); 444*88ebb749SStefano Zampini ierr = MatSeqDenseSetPreallocation(pcbddc->coarse_psi_B,NULL);CHKERRQ(ierr); 445*88ebb749SStefano Zampini if (pcbddc->inexact_prec_type || dbg_flag ) { 446*88ebb749SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->coarse_psi_D);CHKERRQ(ierr); 447*88ebb749SStefano Zampini ierr = MatSetSizes(pcbddc->coarse_psi_D,n_D,pcbddc->local_primal_size,n_D,pcbddc->local_primal_size);CHKERRQ(ierr); 448*88ebb749SStefano Zampini ierr = MatSetType(pcbddc->coarse_psi_D,impMatType);CHKERRQ(ierr); 449*88ebb749SStefano Zampini ierr = MatSeqDenseSetPreallocation(pcbddc->coarse_psi_D,NULL);CHKERRQ(ierr); 450*88ebb749SStefano Zampini } 451*88ebb749SStefano Zampini for (i=0;i<pcbddc->local_primal_size;i++) { 452*88ebb749SStefano Zampini if (n_constraints) { 453*88ebb749SStefano Zampini ierr = VecSet(vec1_C,zero);CHKERRQ(ierr); 454*88ebb749SStefano Zampini ierr = VecGetArray(vec1_C,&array);CHKERRQ(ierr); 455*88ebb749SStefano Zampini for (j=0;j<n_constraints;j++) { 456*88ebb749SStefano Zampini array[j]=coarse_submat_vals[(j+n_vertices)*pcbddc->local_primal_size+i]; 457*88ebb749SStefano Zampini } 458*88ebb749SStefano Zampini ierr = VecRestoreArray(vec1_C,&array);CHKERRQ(ierr); 459*88ebb749SStefano Zampini } 460*88ebb749SStefano Zampini if (i<n_vertices) { 461*88ebb749SStefano Zampini ierr = VecSet(vec1_V,zero);CHKERRQ(ierr); 462*88ebb749SStefano Zampini ierr = VecSetValue(vec1_V,i,m_one,INSERT_VALUES);CHKERRQ(ierr); 463*88ebb749SStefano Zampini ierr = VecAssemblyBegin(vec1_V);CHKERRQ(ierr); 464*88ebb749SStefano Zampini ierr = VecAssemblyEnd(vec1_V);CHKERRQ(ierr); 465*88ebb749SStefano Zampini ierr = MatMultTranspose(A_VR,vec1_V,pcbddc->vec1_R);CHKERRQ(ierr); 466*88ebb749SStefano Zampini if (n_constraints) { 467*88ebb749SStefano Zampini ierr = MatMultTransposeAdd(C_CR,vec1_C,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr); 468*88ebb749SStefano Zampini } 469*88ebb749SStefano Zampini } else { 470*88ebb749SStefano Zampini ierr = MatMultTranspose(C_CR,vec1_C,pcbddc->vec1_R);CHKERRQ(ierr); 471*88ebb749SStefano Zampini } 472*88ebb749SStefano Zampini ierr = KSPSolveTranspose(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec1_R);CHKERRQ(ierr); 473*88ebb749SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 474*88ebb749SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 475*88ebb749SStefano Zampini ierr = VecScatterEnd(pcbddc->R_to_B,pcbddc->vec1_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 476*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_B,&array);CHKERRQ(ierr); 477*88ebb749SStefano Zampini ierr = MatSetValues(pcbddc->coarse_psi_B,n_B,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr); 478*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_B,&array);CHKERRQ(ierr); 479*88ebb749SStefano Zampini if (i<n_vertices) { 480*88ebb749SStefano Zampini ierr = MatSetValue(pcbddc->coarse_psi_B,idx_V_B[i],i,one,INSERT_VALUES);CHKERRQ(ierr); 481*88ebb749SStefano Zampini } 482*88ebb749SStefano Zampini if ( pcbddc->inexact_prec_type || dbg_flag ) { 483*88ebb749SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 484*88ebb749SStefano Zampini ierr = VecScatterEnd(pcbddc->R_to_D,pcbddc->vec1_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 485*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_D,&array);CHKERRQ(ierr); 486*88ebb749SStefano Zampini ierr = MatSetValues(pcbddc->coarse_psi_D,n_D,auxindices,1,&i,array,INSERT_VALUES);CHKERRQ(ierr); 487*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_D,&array);CHKERRQ(ierr); 488*88ebb749SStefano Zampini } 489*88ebb749SStefano Zampini 490*88ebb749SStefano Zampini if ( dbg_flag ) { 491*88ebb749SStefano Zampini /* assemble subdomain vector on nodes */ 492*88ebb749SStefano Zampini ierr = VecSet(pcis->vec1_N,zero);CHKERRQ(ierr); 493*88ebb749SStefano Zampini ierr = VecGetArray(pcis->vec1_N,&array);CHKERRQ(ierr); 494*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr); 495*88ebb749SStefano Zampini for (j=0;j<n_R;j++) array[idx_R_local[j]] = array2[j]; 496*88ebb749SStefano Zampini if (i<n_vertices) array[vertices[i]] = one; 497*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_R,&array2);CHKERRQ(ierr); 498*88ebb749SStefano Zampini ierr = VecRestoreArray(pcis->vec1_N,&array);CHKERRQ(ierr); 499*88ebb749SStefano Zampini /* assemble subdomain vector of lagrange multipliers */ 500*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 501*88ebb749SStefano Zampini for (j=0;j<pcbddc->local_primal_size;j++) { 502*88ebb749SStefano Zampini array[j]=-coarse_submat_vals[j*pcbddc->local_primal_size+i]; 503*88ebb749SStefano Zampini } 504*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 505*88ebb749SStefano Zampini /* check saddle point solution */ 506*88ebb749SStefano Zampini ierr = MatMultTranspose(pcbddc->local_mat,pcis->vec1_N,pcis->vec2_N);CHKERRQ(ierr); 507*88ebb749SStefano Zampini ierr = MatMultTransposeAdd(pcbddc->ConstraintMatrix,pcbddc->vec1_P,pcis->vec2_N,pcis->vec2_N);CHKERRQ(ierr); 508*88ebb749SStefano Zampini ierr = VecNorm(pcis->vec2_N,NORM_INFINITY,&coarsefunctions_errors[i+pcbddc->local_primal_size]);CHKERRQ(ierr); 509*88ebb749SStefano Zampini ierr = MatMult(pcbddc->ConstraintMatrix,pcis->vec1_N,pcbddc->vec1_P);CHKERRQ(ierr); 510*88ebb749SStefano Zampini ierr = VecGetArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 511*88ebb749SStefano Zampini array[i]=array[i]+m_one; /* shift by the identity matrix */ 512*88ebb749SStefano Zampini ierr = VecRestoreArray(pcbddc->vec1_P,&array);CHKERRQ(ierr); 513*88ebb749SStefano Zampini ierr = VecNorm(pcbddc->vec1_P,NORM_INFINITY,&constraints_errors[i+pcbddc->local_primal_size]);CHKERRQ(ierr); 514*88ebb749SStefano Zampini } 515*88ebb749SStefano Zampini } 516*88ebb749SStefano Zampini ierr = MatAssemblyBegin(pcbddc->coarse_psi_B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 517*88ebb749SStefano Zampini ierr = MatAssemblyEnd(pcbddc->coarse_psi_B,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 518*88ebb749SStefano Zampini if ( pcbddc->inexact_prec_type || dbg_flag ) { 519*88ebb749SStefano Zampini ierr = MatAssemblyBegin(pcbddc->coarse_psi_D,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 520*88ebb749SStefano Zampini ierr = MatAssemblyEnd(pcbddc->coarse_psi_D,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 521*88ebb749SStefano Zampini } 522*88ebb749SStefano Zampini } 523*88ebb749SStefano Zampini ierr = PetscFree(idx_V_B);CHKERRQ(ierr); 524*88ebb749SStefano Zampini /* Checking coarse_sub_mat and coarse basis functios */ 525*88ebb749SStefano Zampini /* Symmetric case : It should be \Phi^{(j)^T} A^{(j)} \Phi^{(j)}=coarse_sub_mat */ 526*88ebb749SStefano Zampini /* Non-symmetric case : It should be \Psi^{(j)^T} A^{(j)} \Phi^{(j)}=coarse_sub_mat */ 527*88ebb749SStefano Zampini if (dbg_flag) { 528*88ebb749SStefano Zampini Mat coarse_sub_mat; 529*88ebb749SStefano Zampini Mat TM1,TM2,TM3,TM4; 530*88ebb749SStefano Zampini Mat coarse_phi_D,coarse_phi_B; 531*88ebb749SStefano Zampini Mat coarse_psi_D,coarse_psi_B; 532*88ebb749SStefano Zampini Mat A_II,A_BB,A_IB,A_BI; 533*88ebb749SStefano Zampini MatType checkmattype=MATSEQAIJ; 534*88ebb749SStefano Zampini PetscReal real_value; 535*88ebb749SStefano Zampini 536*88ebb749SStefano Zampini ierr = MatConvert(pcis->A_II,checkmattype,MAT_INITIAL_MATRIX,&A_II);CHKERRQ(ierr); 537*88ebb749SStefano Zampini ierr = MatConvert(pcis->A_IB,checkmattype,MAT_INITIAL_MATRIX,&A_IB);CHKERRQ(ierr); 538*88ebb749SStefano Zampini ierr = MatConvert(pcis->A_BI,checkmattype,MAT_INITIAL_MATRIX,&A_BI);CHKERRQ(ierr); 539*88ebb749SStefano Zampini ierr = MatConvert(pcis->A_BB,checkmattype,MAT_INITIAL_MATRIX,&A_BB);CHKERRQ(ierr); 540*88ebb749SStefano Zampini ierr = MatConvert(pcbddc->coarse_phi_D,checkmattype,MAT_INITIAL_MATRIX,&coarse_phi_D);CHKERRQ(ierr); 541*88ebb749SStefano Zampini ierr = MatConvert(pcbddc->coarse_phi_B,checkmattype,MAT_INITIAL_MATRIX,&coarse_phi_B);CHKERRQ(ierr); 542*88ebb749SStefano Zampini if (pcbddc->coarse_psi_B) { 543*88ebb749SStefano Zampini ierr = MatConvert(pcbddc->coarse_psi_D,checkmattype,MAT_INITIAL_MATRIX,&coarse_psi_D);CHKERRQ(ierr); 544*88ebb749SStefano Zampini ierr = MatConvert(pcbddc->coarse_psi_B,checkmattype,MAT_INITIAL_MATRIX,&coarse_psi_B);CHKERRQ(ierr); 545*88ebb749SStefano Zampini } 546*88ebb749SStefano Zampini ierr = MatCreateSeqDense(PETSC_COMM_SELF,pcbddc->local_primal_size,pcbddc->local_primal_size,coarse_submat_vals,&coarse_sub_mat);CHKERRQ(ierr); 547*88ebb749SStefano Zampini 548*88ebb749SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer,"--------------------------------------------------\n");CHKERRQ(ierr); 549*88ebb749SStefano Zampini ierr = PetscViewerASCIIPrintf(viewer,"Check coarse sub mat and local basis functions\n");CHKERRQ(ierr); 550*88ebb749SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 551*88ebb749SStefano Zampini if (pcbddc->coarse_psi_B) { 552*88ebb749SStefano Zampini ierr = MatMatMult(A_II,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); 553*88ebb749SStefano Zampini ierr = MatTransposeMatMult(coarse_psi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM1);CHKERRQ(ierr); 554*88ebb749SStefano Zampini ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); 555*88ebb749SStefano Zampini ierr = MatMatMult(A_BB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); 556*88ebb749SStefano Zampini ierr = MatTransposeMatMult(coarse_psi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM2);CHKERRQ(ierr); 557*88ebb749SStefano Zampini ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); 558*88ebb749SStefano Zampini ierr = MatMatMult(A_IB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); 559*88ebb749SStefano Zampini ierr = MatTransposeMatMult(coarse_psi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM3);CHKERRQ(ierr); 560*88ebb749SStefano Zampini ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); 561*88ebb749SStefano Zampini ierr = MatMatMult(A_BI,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); 562*88ebb749SStefano Zampini ierr = MatTransposeMatMult(coarse_psi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM4);CHKERRQ(ierr); 563*88ebb749SStefano Zampini ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); 564*88ebb749SStefano Zampini } else { 565*88ebb749SStefano Zampini ierr = MatPtAP(A_II,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&TM1);CHKERRQ(ierr); 566*88ebb749SStefano Zampini ierr = MatPtAP(A_BB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&TM2);CHKERRQ(ierr); 567*88ebb749SStefano Zampini ierr = MatMatMult(A_IB,coarse_phi_B,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); 568*88ebb749SStefano Zampini ierr = MatTransposeMatMult(coarse_phi_D,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM3);CHKERRQ(ierr); 569*88ebb749SStefano Zampini ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); 570*88ebb749SStefano Zampini ierr = MatMatMult(A_BI,coarse_phi_D,MAT_INITIAL_MATRIX,1.0,&AUXMAT);CHKERRQ(ierr); 571*88ebb749SStefano Zampini ierr = MatTransposeMatMult(coarse_phi_B,AUXMAT,MAT_INITIAL_MATRIX,1.0,&TM4);CHKERRQ(ierr); 572*88ebb749SStefano Zampini ierr = MatDestroy(&AUXMAT);CHKERRQ(ierr); 573*88ebb749SStefano Zampini } 574*88ebb749SStefano Zampini ierr = MatAXPY(TM1,one,TM2,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 575*88ebb749SStefano Zampini ierr = MatAXPY(TM1,one,TM3,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 576*88ebb749SStefano Zampini ierr = MatAXPY(TM1,one,TM4,DIFFERENT_NONZERO_PATTERN);CHKERRQ(ierr); 577*88ebb749SStefano Zampini ierr = MatConvert(TM1,MATSEQDENSE,MAT_REUSE_MATRIX,&TM1);CHKERRQ(ierr); 578*88ebb749SStefano Zampini ierr = MatAXPY(TM1,m_one,coarse_sub_mat,SAME_NONZERO_PATTERN);CHKERRQ(ierr); 579*88ebb749SStefano Zampini ierr = MatNorm(TM1,NORM_INFINITY,&real_value);CHKERRQ(ierr); 580*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"----------------------------------\n");CHKERRQ(ierr); 581*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"Subdomain %04d \n",PetscGlobalRank);CHKERRQ(ierr); 582*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"matrix error = % 1.14e\n",real_value);CHKERRQ(ierr); 583*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"coarse functions (phi) errors\n");CHKERRQ(ierr); 584*88ebb749SStefano Zampini for (i=0;i<pcbddc->local_primal_size;i++) { 585*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"local %02d-th function error = % 1.14e\n",i,coarsefunctions_errors[i]);CHKERRQ(ierr); 586*88ebb749SStefano Zampini } 587*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"constraints (phi) errors\n");CHKERRQ(ierr); 588*88ebb749SStefano Zampini for (i=0;i<pcbddc->local_primal_size;i++) { 589*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"local %02d-th function error = % 1.14e\n",i,constraints_errors[i]);CHKERRQ(ierr); 590*88ebb749SStefano Zampini } 591*88ebb749SStefano Zampini if (pcbddc->coarse_psi_B) { 592*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"coarse functions (psi) errors\n");CHKERRQ(ierr); 593*88ebb749SStefano Zampini for (i=pcbddc->local_primal_size;i<2*pcbddc->local_primal_size;i++) { 594*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"local %02d-th function error = % 1.14e\n",i-pcbddc->local_primal_size,coarsefunctions_errors[i]);CHKERRQ(ierr); 595*88ebb749SStefano Zampini } 596*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"constraints (psi) errors\n");CHKERRQ(ierr); 597*88ebb749SStefano Zampini for (i=pcbddc->local_primal_size;i<2*pcbddc->local_primal_size;i++) { 598*88ebb749SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(viewer,"local %02d-th function error = % 1.14e\n",i-pcbddc->local_primal_size,constraints_errors[i]);CHKERRQ(ierr); 599*88ebb749SStefano Zampini } 600*88ebb749SStefano Zampini } 601*88ebb749SStefano Zampini ierr = PetscViewerFlush(viewer);CHKERRQ(ierr); 602*88ebb749SStefano Zampini ierr = MatDestroy(&A_II);CHKERRQ(ierr); 603*88ebb749SStefano Zampini ierr = MatDestroy(&A_BB);CHKERRQ(ierr); 604*88ebb749SStefano Zampini ierr = MatDestroy(&A_IB);CHKERRQ(ierr); 605*88ebb749SStefano Zampini ierr = MatDestroy(&A_BI);CHKERRQ(ierr); 606*88ebb749SStefano Zampini ierr = MatDestroy(&TM1);CHKERRQ(ierr); 607*88ebb749SStefano Zampini ierr = MatDestroy(&TM2);CHKERRQ(ierr); 608*88ebb749SStefano Zampini ierr = MatDestroy(&TM3);CHKERRQ(ierr); 609*88ebb749SStefano Zampini ierr = MatDestroy(&TM4);CHKERRQ(ierr); 610*88ebb749SStefano Zampini ierr = MatDestroy(&coarse_phi_D);CHKERRQ(ierr); 611*88ebb749SStefano Zampini ierr = MatDestroy(&coarse_phi_B);CHKERRQ(ierr); 612*88ebb749SStefano Zampini if (pcbddc->coarse_psi_B) { 613*88ebb749SStefano Zampini ierr = MatDestroy(&coarse_psi_D);CHKERRQ(ierr); 614*88ebb749SStefano Zampini ierr = MatDestroy(&coarse_psi_B);CHKERRQ(ierr); 615*88ebb749SStefano Zampini } 616*88ebb749SStefano Zampini ierr = MatDestroy(&coarse_sub_mat);CHKERRQ(ierr); 617*88ebb749SStefano Zampini ierr = ISRestoreIndices(is_R_local,(const PetscInt**)&idx_R_local);CHKERRQ(ierr); 618*88ebb749SStefano Zampini ierr = PetscFree(coarsefunctions_errors);CHKERRQ(ierr); 619*88ebb749SStefano Zampini ierr = PetscFree(constraints_errors);CHKERRQ(ierr); 620*88ebb749SStefano Zampini } 621*88ebb749SStefano Zampini /* free memory */ 622*88ebb749SStefano Zampini if (n_vertices) { 623*88ebb749SStefano Zampini ierr = PetscFree(vertices);CHKERRQ(ierr); 624*88ebb749SStefano Zampini ierr = VecDestroy(&vec1_V);CHKERRQ(ierr); 625*88ebb749SStefano Zampini ierr = VecDestroy(&vec2_V);CHKERRQ(ierr); 626*88ebb749SStefano Zampini ierr = MatDestroy(&A_RV);CHKERRQ(ierr); 627*88ebb749SStefano Zampini ierr = MatDestroy(&A_VR);CHKERRQ(ierr); 628*88ebb749SStefano Zampini ierr = MatDestroy(&A_VV);CHKERRQ(ierr); 629*88ebb749SStefano Zampini } 630*88ebb749SStefano Zampini if (n_constraints) { 631*88ebb749SStefano Zampini ierr = VecDestroy(&vec1_C);CHKERRQ(ierr); 632*88ebb749SStefano Zampini ierr = VecDestroy(&vec2_C);CHKERRQ(ierr); 633*88ebb749SStefano Zampini ierr = MatDestroy(&M1);CHKERRQ(ierr); 634*88ebb749SStefano Zampini ierr = MatDestroy(&C_CR);CHKERRQ(ierr); 635*88ebb749SStefano Zampini } 636*88ebb749SStefano Zampini ierr = PetscFree(auxindices);CHKERRQ(ierr); 637*88ebb749SStefano Zampini ierr = PetscFree(nnz);CHKERRQ(ierr); 638*88ebb749SStefano Zampini /* create coarse matrix and data structures for message passing associated actual choice of coarse problem type */ 639*88ebb749SStefano Zampini ierr = PCBDDCSetUpCoarseEnvironment(pc,coarse_submat_vals);CHKERRQ(ierr); 640*88ebb749SStefano Zampini ierr = PetscFree(coarse_submat_vals);CHKERRQ(ierr); 641*88ebb749SStefano Zampini PetscFunctionReturn(0); 642*88ebb749SStefano Zampini } 643*88ebb749SStefano Zampini 644*88ebb749SStefano Zampini #undef __FUNCT__ 645aa0d41d4SStefano Zampini #define __FUNCT__ "PCBDDCSetUpLocalMatrices" 646aa0d41d4SStefano Zampini PetscErrorCode PCBDDCSetUpLocalMatrices(PC pc) 647aa0d41d4SStefano Zampini { 648aa0d41d4SStefano Zampini PC_IS* pcis = (PC_IS*)(pc->data); 649aa0d41d4SStefano Zampini PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 650aa0d41d4SStefano Zampini Mat_IS* matis = (Mat_IS*)pc->pmat->data; 651aa0d41d4SStefano Zampini /* manage repeated solves */ 652aa0d41d4SStefano Zampini MatReuse reuse; 653aa0d41d4SStefano Zampini MatStructure matstruct; 654aa0d41d4SStefano Zampini PetscErrorCode ierr; 655aa0d41d4SStefano Zampini 656aa0d41d4SStefano Zampini PetscFunctionBegin; 657aa0d41d4SStefano Zampini /* get mat flags */ 658aa0d41d4SStefano Zampini ierr = PCGetOperators(pc,NULL,NULL,&matstruct);CHKERRQ(ierr); 659aa0d41d4SStefano Zampini reuse = MAT_INITIAL_MATRIX; 660aa0d41d4SStefano Zampini if (pc->setupcalled) { 661aa0d41d4SStefano Zampini /* when matstruct is SAME_PRECONDITIONER, we shouldn't be here */ 662aa0d41d4SStefano Zampini if (matstruct == SAME_PRECONDITIONER) SETERRQ(PetscObjectComm((PetscObject)pc),PETSC_ERR_PLIB,"This should not happen"); 663aa0d41d4SStefano Zampini if (matstruct == SAME_NONZERO_PATTERN) { 664aa0d41d4SStefano Zampini reuse = MAT_REUSE_MATRIX; 665aa0d41d4SStefano Zampini } else { 666aa0d41d4SStefano Zampini reuse = MAT_INITIAL_MATRIX; 667aa0d41d4SStefano Zampini } 668aa0d41d4SStefano Zampini } 669aa0d41d4SStefano Zampini if (reuse == MAT_INITIAL_MATRIX) { 670aa0d41d4SStefano Zampini ierr = MatDestroy(&pcis->A_II);CHKERRQ(ierr); 671aa0d41d4SStefano Zampini ierr = MatDestroy(&pcis->A_IB);CHKERRQ(ierr); 672aa0d41d4SStefano Zampini ierr = MatDestroy(&pcis->A_BI);CHKERRQ(ierr); 673aa0d41d4SStefano Zampini ierr = MatDestroy(&pcis->A_BB);CHKERRQ(ierr); 674aa0d41d4SStefano Zampini ierr = MatDestroy(&pcbddc->local_mat);CHKERRQ(ierr); 675aa0d41d4SStefano Zampini } 676aa0d41d4SStefano Zampini 677aa0d41d4SStefano Zampini /* transform local matrices if needed */ 678aa0d41d4SStefano Zampini if (pcbddc->use_change_of_basis) { 679aa0d41d4SStefano Zampini Mat change_mat_all; 680aa0d41d4SStefano Zampini PetscScalar *row_cmat_values; 681aa0d41d4SStefano Zampini PetscInt *row_cmat_indices; 682aa0d41d4SStefano Zampini PetscInt *nnz,*is_indices,*temp_indices; 683aa0d41d4SStefano Zampini PetscInt i,j,k,n_D,n_B; 684aa0d41d4SStefano Zampini 685aa0d41d4SStefano Zampini /* Get Non-overlapping dimensions */ 686aa0d41d4SStefano Zampini n_B = pcis->n_B; 687aa0d41d4SStefano Zampini n_D = pcis->n-n_B; 688aa0d41d4SStefano Zampini 689aa0d41d4SStefano Zampini /* compute nonzero structure of change of basis on all local nodes */ 690aa0d41d4SStefano Zampini ierr = PetscMalloc(pcis->n*sizeof(PetscInt),&nnz);CHKERRQ(ierr); 691aa0d41d4SStefano Zampini ierr = ISGetIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 692aa0d41d4SStefano Zampini for (i=0;i<n_D;i++) nnz[is_indices[i]] = 1; 693aa0d41d4SStefano Zampini ierr = ISRestoreIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 694aa0d41d4SStefano Zampini ierr = ISGetIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 695aa0d41d4SStefano Zampini k=1; 696aa0d41d4SStefano Zampini for (i=0;i<n_B;i++) { 697aa0d41d4SStefano Zampini ierr = MatGetRow(pcbddc->ChangeOfBasisMatrix,i,&j,NULL,NULL);CHKERRQ(ierr); 698aa0d41d4SStefano Zampini nnz[is_indices[i]]=j; 699aa0d41d4SStefano Zampini if (k < j) k = j; 700aa0d41d4SStefano Zampini ierr = MatRestoreRow(pcbddc->ChangeOfBasisMatrix,i,&j,NULL,NULL);CHKERRQ(ierr); 701aa0d41d4SStefano Zampini } 702aa0d41d4SStefano Zampini ierr = ISRestoreIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 703aa0d41d4SStefano Zampini /* assemble change of basis matrix on the whole set of local dofs */ 704aa0d41d4SStefano Zampini ierr = PetscMalloc(k*sizeof(PetscInt),&temp_indices);CHKERRQ(ierr); 705aa0d41d4SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&change_mat_all);CHKERRQ(ierr); 706aa0d41d4SStefano Zampini ierr = MatSetSizes(change_mat_all,pcis->n,pcis->n,pcis->n,pcis->n);CHKERRQ(ierr); 707aa0d41d4SStefano Zampini ierr = MatSetType(change_mat_all,MATSEQAIJ);CHKERRQ(ierr); 708aa0d41d4SStefano Zampini ierr = MatSeqAIJSetPreallocation(change_mat_all,0,nnz);CHKERRQ(ierr); 709aa0d41d4SStefano Zampini ierr = ISGetIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 710aa0d41d4SStefano Zampini for (i=0;i<n_D;i++) { 711aa0d41d4SStefano Zampini ierr = MatSetValue(change_mat_all,is_indices[i],is_indices[i],1.0,INSERT_VALUES);CHKERRQ(ierr); 712aa0d41d4SStefano Zampini } 713aa0d41d4SStefano Zampini ierr = ISRestoreIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 714aa0d41d4SStefano Zampini ierr = ISGetIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 715aa0d41d4SStefano Zampini for (i=0;i<n_B;i++) { 716aa0d41d4SStefano Zampini ierr = MatGetRow(pcbddc->ChangeOfBasisMatrix,i,&j,(const PetscInt**)&row_cmat_indices,(const PetscScalar**)&row_cmat_values);CHKERRQ(ierr); 717aa0d41d4SStefano Zampini for (k=0; k<j; k++) temp_indices[k]=is_indices[row_cmat_indices[k]]; 718aa0d41d4SStefano Zampini ierr = MatSetValues(change_mat_all,1,&is_indices[i],j,temp_indices,row_cmat_values,INSERT_VALUES);CHKERRQ(ierr); 719aa0d41d4SStefano Zampini ierr = MatRestoreRow(pcbddc->ChangeOfBasisMatrix,i,&j,(const PetscInt**)&row_cmat_indices,(const PetscScalar**)&row_cmat_values);CHKERRQ(ierr); 720aa0d41d4SStefano Zampini } 721aa0d41d4SStefano Zampini ierr = MatAssemblyBegin(change_mat_all,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 722aa0d41d4SStefano Zampini ierr = MatAssemblyEnd(change_mat_all,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 723aa0d41d4SStefano Zampini /* TODO: HOW TO WORK WITH BAIJ? PtAP not provided */ 724aa0d41d4SStefano Zampini ierr = MatGetBlockSize(matis->A,&i);CHKERRQ(ierr); 725aa0d41d4SStefano Zampini if (i==1) { 726aa0d41d4SStefano Zampini ierr = MatPtAP(matis->A,change_mat_all,reuse,2.0,&pcbddc->local_mat);CHKERRQ(ierr); 727aa0d41d4SStefano Zampini } else { 728aa0d41d4SStefano Zampini Mat work_mat; 729aa0d41d4SStefano Zampini ierr = MatConvert(matis->A,MATSEQAIJ,MAT_INITIAL_MATRIX,&work_mat);CHKERRQ(ierr); 730aa0d41d4SStefano Zampini ierr = MatPtAP(work_mat,change_mat_all,reuse,2.0,&pcbddc->local_mat);CHKERRQ(ierr); 731aa0d41d4SStefano Zampini ierr = MatDestroy(&work_mat);CHKERRQ(ierr); 732aa0d41d4SStefano Zampini } 733aa0d41d4SStefano Zampini ierr = MatDestroy(&change_mat_all);CHKERRQ(ierr); 734aa0d41d4SStefano Zampini ierr = PetscFree(nnz);CHKERRQ(ierr); 735aa0d41d4SStefano Zampini ierr = PetscFree(temp_indices);CHKERRQ(ierr); 736aa0d41d4SStefano Zampini } else { 737aa0d41d4SStefano Zampini /* without change of basis, the local matrix is unchanged */ 738aa0d41d4SStefano Zampini if (!pcbddc->local_mat) { 739aa0d41d4SStefano Zampini ierr = PetscObjectReference((PetscObject)matis->A);CHKERRQ(ierr); 740aa0d41d4SStefano Zampini pcbddc->local_mat = matis->A; 741aa0d41d4SStefano Zampini } 742aa0d41d4SStefano Zampini } 743aa0d41d4SStefano Zampini 744aa0d41d4SStefano Zampini /* get submatrices */ 745aa0d41d4SStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_I_local,reuse,&pcis->A_II);CHKERRQ(ierr); 746aa0d41d4SStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_I_local,pcis->is_B_local,reuse,&pcis->A_IB);CHKERRQ(ierr); 747aa0d41d4SStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_I_local,reuse,&pcis->A_BI);CHKERRQ(ierr); 748aa0d41d4SStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,pcis->is_B_local,pcis->is_B_local,reuse,&pcis->A_BB);CHKERRQ(ierr); 749aa0d41d4SStefano Zampini PetscFunctionReturn(0); 750aa0d41d4SStefano Zampini } 751aa0d41d4SStefano Zampini 752aa0d41d4SStefano Zampini #undef __FUNCT__ 753a64d13efSStefano Zampini #define __FUNCT__ "PCBDDCSetUpLocalScatters" 754a64d13efSStefano Zampini PetscErrorCode PCBDDCSetUpLocalScatters(PC pc,IS* is_R_local_n) 755a64d13efSStefano Zampini { 756a64d13efSStefano Zampini PC_IS* pcis = (PC_IS*)(pc->data); 757a64d13efSStefano Zampini PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 758a64d13efSStefano Zampini IS is_R_local,is_aux1,is_aux2; 759a64d13efSStefano Zampini PetscInt *vertices,*aux_array1,*aux_array2,*is_indices,*idx_R_local; 760a64d13efSStefano Zampini PetscInt n_vertices,n_constraints,i,j,n_R,n_D,n_B; 761a64d13efSStefano Zampini PetscBool *array_bool; 762a64d13efSStefano Zampini PetscErrorCode ierr; 763a64d13efSStefano Zampini 764a64d13efSStefano Zampini PetscFunctionBegin; 765a64d13efSStefano Zampini /* Set Non-overlapping dimensions */ 766a64d13efSStefano Zampini n_B = pcis->n_B; n_D = pcis->n - n_B; 767a64d13efSStefano Zampini /* get vertex indices from constraint matrix */ 768a64d13efSStefano Zampini ierr = PCBDDCGetPrimalVerticesLocalIdx(pc,&n_vertices,&vertices);CHKERRQ(ierr); 769a64d13efSStefano Zampini /* Set number of constraints */ 770a64d13efSStefano Zampini n_constraints = pcbddc->local_primal_size-n_vertices; 771a64d13efSStefano Zampini /* Dohrmann's notation: dofs splitted in R (Remaining: all dofs but the vertices) and V (Vertices) */ 772a64d13efSStefano Zampini ierr = PetscMalloc(pcis->n*sizeof(PetscInt),&array_bool);CHKERRQ(ierr); 773a64d13efSStefano Zampini for (i=0;i<pcis->n;i++) array_bool[i] = PETSC_TRUE; 774a64d13efSStefano Zampini for (i=0;i<n_vertices;i++) array_bool[vertices[i]] = PETSC_FALSE; 775a64d13efSStefano Zampini ierr = PetscMalloc((pcis->n-n_vertices)*sizeof(PetscInt),&idx_R_local);CHKERRQ(ierr); 776a64d13efSStefano Zampini for (i=0, n_R=0; i<pcis->n; i++) { 777a64d13efSStefano Zampini if (array_bool[i]) { 778a64d13efSStefano Zampini idx_R_local[n_R] = i; 779a64d13efSStefano Zampini n_R++; 780a64d13efSStefano Zampini } 781a64d13efSStefano Zampini } 782a64d13efSStefano Zampini ierr = PetscFree(vertices);CHKERRQ(ierr); 783a64d13efSStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,n_R,idx_R_local,PETSC_OWN_POINTER,&is_R_local);CHKERRQ(ierr); 784a64d13efSStefano Zampini 785a64d13efSStefano Zampini /* print some info if requested */ 786a64d13efSStefano Zampini if (pcbddc->dbg_flag) { 787a64d13efSStefano Zampini ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr); 788a64d13efSStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 789a64d13efSStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d local dimensions\n",PetscGlobalRank);CHKERRQ(ierr); 790a64d13efSStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"local_size = %d, dirichlet_size = %d, boundary_size = %d\n",pcis->n,n_D,n_B);CHKERRQ(ierr); 791a64d13efSStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"r_size = %d, v_size = %d, constraints = %d, local_primal_size = %d\n",n_R,n_vertices,n_constraints,pcbddc->local_primal_size);CHKERRQ(ierr); 792a64d13efSStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"pcbddc->n_vertices = %d, pcbddc->n_constraints = %d\n",pcbddc->n_vertices,pcbddc->n_constraints);CHKERRQ(ierr); 793a64d13efSStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 794a64d13efSStefano Zampini } 795a64d13efSStefano Zampini 796a64d13efSStefano Zampini /* VecScatters pcbddc->R_to_B and (optionally) pcbddc->R_to_D */ 797a64d13efSStefano Zampini ierr = PetscMalloc((pcis->n_B-n_vertices)*sizeof(PetscInt),&aux_array1);CHKERRQ(ierr); 798a64d13efSStefano Zampini ierr = PetscMalloc((pcis->n_B-n_vertices)*sizeof(PetscInt),&aux_array2);CHKERRQ(ierr); 799a64d13efSStefano Zampini ierr = ISGetIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 800a64d13efSStefano Zampini for (i=0; i<n_D; i++) array_bool[is_indices[i]] = PETSC_FALSE; 801a64d13efSStefano Zampini ierr = ISRestoreIndices(pcis->is_I_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 802a64d13efSStefano Zampini for (i=0, j=0; i<n_R; i++) { 803a64d13efSStefano Zampini if (array_bool[idx_R_local[i]]) { 804a64d13efSStefano Zampini aux_array1[j] = i; 805a64d13efSStefano Zampini j++; 806a64d13efSStefano Zampini } 807a64d13efSStefano Zampini } 808a64d13efSStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,j,aux_array1,PETSC_OWN_POINTER,&is_aux1);CHKERRQ(ierr); 809a64d13efSStefano Zampini ierr = ISGetIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 810a64d13efSStefano Zampini for (i=0, j=0; i<n_B; i++) { 811a64d13efSStefano Zampini if (array_bool[is_indices[i]]) { 812a64d13efSStefano Zampini aux_array2[j] = i; j++; 813a64d13efSStefano Zampini } 814a64d13efSStefano Zampini } 815a64d13efSStefano Zampini ierr = ISRestoreIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 816a64d13efSStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,j,aux_array2,PETSC_OWN_POINTER,&is_aux2);CHKERRQ(ierr); 817a64d13efSStefano Zampini ierr = VecScatterCreate(pcbddc->vec1_R,is_aux1,pcis->vec1_B,is_aux2,&pcbddc->R_to_B);CHKERRQ(ierr); 818a64d13efSStefano Zampini ierr = ISDestroy(&is_aux1);CHKERRQ(ierr); 819a64d13efSStefano Zampini ierr = ISDestroy(&is_aux2);CHKERRQ(ierr); 820a64d13efSStefano Zampini 821a64d13efSStefano Zampini if (pcbddc->inexact_prec_type || pcbddc->dbg_flag ) { 822a64d13efSStefano Zampini ierr = PetscMalloc(n_D*sizeof(PetscInt),&aux_array1);CHKERRQ(ierr); 823a64d13efSStefano Zampini for (i=0, j=0; i<n_R; i++) { 824a64d13efSStefano Zampini if (!array_bool[idx_R_local[i]]) { 825a64d13efSStefano Zampini aux_array1[j] = i; 826a64d13efSStefano Zampini j++; 827a64d13efSStefano Zampini } 828a64d13efSStefano Zampini } 829a64d13efSStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,j,aux_array1,PETSC_OWN_POINTER,&is_aux1);CHKERRQ(ierr); 830a64d13efSStefano Zampini ierr = VecScatterCreate(pcbddc->vec1_R,is_aux1,pcis->vec1_D,(IS)0,&pcbddc->R_to_D);CHKERRQ(ierr); 831a64d13efSStefano Zampini ierr = ISDestroy(&is_aux1);CHKERRQ(ierr); 832a64d13efSStefano Zampini } 833a64d13efSStefano Zampini ierr = PetscFree(array_bool);CHKERRQ(ierr); 834a64d13efSStefano Zampini *is_R_local_n = is_R_local; 835a64d13efSStefano Zampini PetscFunctionReturn(0); 836a64d13efSStefano Zampini } 837a64d13efSStefano Zampini 838a64d13efSStefano Zampini #undef __FUNCT__ 839304d26faSStefano Zampini #define __FUNCT__ "PCBDDCSetUseExactDirichlet" 840304d26faSStefano Zampini PetscErrorCode PCBDDCSetUseExactDirichlet(PC pc,PetscBool use) 841304d26faSStefano Zampini { 842304d26faSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 843304d26faSStefano Zampini 844304d26faSStefano Zampini PetscFunctionBegin; 845304d26faSStefano Zampini pcbddc->use_exact_dirichlet=use; 846304d26faSStefano Zampini PetscFunctionReturn(0); 847304d26faSStefano Zampini } 848304d26faSStefano Zampini 849304d26faSStefano Zampini #undef __FUNCT__ 850304d26faSStefano Zampini #define __FUNCT__ "PCBDDCSetUpLocalSolvers" 851304d26faSStefano Zampini PetscErrorCode PCBDDCSetUpLocalSolvers(PC pc, IS is_I_local, IS is_R_local) 852304d26faSStefano Zampini { 853304d26faSStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 854304d26faSStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 855304d26faSStefano Zampini PC pc_temp; 856304d26faSStefano Zampini Mat A_RR; 857304d26faSStefano Zampini Vec vec1,vec2,vec3; 858304d26faSStefano Zampini MatStructure matstruct; 859304d26faSStefano Zampini PetscScalar m_one = -1.0; 860304d26faSStefano Zampini PetscReal value; 861304d26faSStefano Zampini PetscInt n_D,n_R,use_exact,use_exact_reduced; 862304d26faSStefano Zampini PetscErrorCode ierr; 863304d26faSStefano Zampini 864304d26faSStefano Zampini PetscFunctionBegin; 865304d26faSStefano Zampini /* Creating PC contexts for local Dirichlet and Neumann problems */ 866304d26faSStefano Zampini ierr = PCGetOperators(pc,NULL,NULL,&matstruct);CHKERRQ(ierr); 867304d26faSStefano Zampini 868304d26faSStefano Zampini /* DIRICHLET PROBLEM */ 869ac78edfcSStefano Zampini /* Matrix for Dirichlet problem is pcis->A_II */ 870304d26faSStefano Zampini ierr = ISGetSize(is_I_local,&n_D);CHKERRQ(ierr); 871304d26faSStefano Zampini if (!pcbddc->ksp_D) { /* create object if not yet build */ 872304d26faSStefano Zampini ierr = KSPCreate(PETSC_COMM_SELF,&pcbddc->ksp_D);CHKERRQ(ierr); 873304d26faSStefano Zampini ierr = PetscObjectIncrementTabLevel((PetscObject)pcbddc->ksp_D,(PetscObject)pc,1);CHKERRQ(ierr); 874304d26faSStefano Zampini /* default */ 875304d26faSStefano Zampini ierr = KSPSetType(pcbddc->ksp_D,KSPPREONLY);CHKERRQ(ierr); 876304d26faSStefano Zampini ierr = KSPSetOptionsPrefix(pcbddc->ksp_D,"dirichlet_");CHKERRQ(ierr); 877304d26faSStefano Zampini ierr = KSPGetPC(pcbddc->ksp_D,&pc_temp);CHKERRQ(ierr); 878304d26faSStefano Zampini ierr = PCSetType(pc_temp,PCLU);CHKERRQ(ierr); 879304d26faSStefano Zampini ierr = PCFactorSetReuseFill(pc_temp,PETSC_TRUE);CHKERRQ(ierr); 880304d26faSStefano Zampini } 881304d26faSStefano Zampini ierr = KSPSetOperators(pcbddc->ksp_D,pcis->A_II,pcis->A_II,matstruct);CHKERRQ(ierr); 882304d26faSStefano Zampini /* Allow user's customization */ 883304d26faSStefano Zampini ierr = KSPSetFromOptions(pcbddc->ksp_D);CHKERRQ(ierr); 884304d26faSStefano Zampini /* umfpack interface has a bug when matrix dimension is zero. TODO solve from umfpack interface */ 885304d26faSStefano Zampini if (!n_D) { 886304d26faSStefano Zampini ierr = KSPGetPC(pcbddc->ksp_D,&pc_temp);CHKERRQ(ierr); 887304d26faSStefano Zampini ierr = PCSetType(pc_temp,PCNONE);CHKERRQ(ierr); 888304d26faSStefano Zampini } 889304d26faSStefano Zampini /* Set Up KSP for Dirichlet problem of BDDC */ 890304d26faSStefano Zampini ierr = KSPSetUp(pcbddc->ksp_D);CHKERRQ(ierr); 891304d26faSStefano Zampini /* set ksp_D into pcis data */ 892304d26faSStefano Zampini ierr = KSPDestroy(&pcis->ksp_D);CHKERRQ(ierr); 893304d26faSStefano Zampini ierr = PetscObjectReference((PetscObject)pcbddc->ksp_D);CHKERRQ(ierr); 894304d26faSStefano Zampini pcis->ksp_D = pcbddc->ksp_D; 895304d26faSStefano Zampini 896304d26faSStefano Zampini /* NEUMANN PROBLEM */ 897304d26faSStefano Zampini /* Matrix for Neumann problem is A_RR -> we need to create it */ 898304d26faSStefano Zampini ierr = ISGetSize(is_R_local,&n_R);CHKERRQ(ierr); 899304d26faSStefano Zampini ierr = MatGetSubMatrix(pcbddc->local_mat,is_R_local,is_R_local,MAT_INITIAL_MATRIX,&A_RR);CHKERRQ(ierr); 900304d26faSStefano Zampini if (!pcbddc->ksp_R) { /* create object if not yet build */ 901304d26faSStefano Zampini ierr = KSPCreate(PETSC_COMM_SELF,&pcbddc->ksp_R);CHKERRQ(ierr); 902304d26faSStefano Zampini ierr = PetscObjectIncrementTabLevel((PetscObject)pcbddc->ksp_R,(PetscObject)pc,1);CHKERRQ(ierr); 903304d26faSStefano Zampini /* default */ 904304d26faSStefano Zampini ierr = KSPSetType(pcbddc->ksp_R,KSPPREONLY);CHKERRQ(ierr); 905304d26faSStefano Zampini ierr = KSPSetOptionsPrefix(pcbddc->ksp_R,"neumann_");CHKERRQ(ierr); 906304d26faSStefano Zampini ierr = KSPGetPC(pcbddc->ksp_R,&pc_temp);CHKERRQ(ierr); 907304d26faSStefano Zampini ierr = PCSetType(pc_temp,PCLU);CHKERRQ(ierr); 908304d26faSStefano Zampini ierr = PCFactorSetReuseFill(pc_temp,PETSC_TRUE);CHKERRQ(ierr); 909304d26faSStefano Zampini } 910304d26faSStefano Zampini ierr = KSPSetOperators(pcbddc->ksp_R,A_RR,A_RR,matstruct);CHKERRQ(ierr); 911304d26faSStefano Zampini /* Allow user's customization */ 912304d26faSStefano Zampini ierr = KSPSetFromOptions(pcbddc->ksp_R);CHKERRQ(ierr); 913304d26faSStefano Zampini /* umfpack interface has a bug when matrix dimension is zero. TODO solve from umfpack interface */ 914304d26faSStefano Zampini if (!n_R) { 915304d26faSStefano Zampini ierr = KSPGetPC(pcbddc->ksp_R,&pc_temp);CHKERRQ(ierr); 916304d26faSStefano Zampini ierr = PCSetType(pc_temp,PCNONE);CHKERRQ(ierr); 917304d26faSStefano Zampini } 918304d26faSStefano Zampini /* Set Up KSP for Neumann problem of BDDC */ 919304d26faSStefano Zampini ierr = KSPSetUp(pcbddc->ksp_R);CHKERRQ(ierr); 920304d26faSStefano Zampini 921304d26faSStefano Zampini /* check Dirichlet and Neumann solvers and adapt them if a nullspace correction is needed */ 922304d26faSStefano Zampini 923304d26faSStefano Zampini /* Dirichlet */ 924304d26faSStefano Zampini ierr = MatGetVecs(pcis->A_II,&vec1,&vec2);CHKERRQ(ierr); 925304d26faSStefano Zampini ierr = VecDuplicate(vec1,&vec3);CHKERRQ(ierr); 926304d26faSStefano Zampini ierr = VecSetRandom(vec1,NULL);CHKERRQ(ierr); 927304d26faSStefano Zampini ierr = MatMult(pcis->A_II,vec1,vec2);CHKERRQ(ierr); 928304d26faSStefano Zampini ierr = KSPSolve(pcbddc->ksp_D,vec2,vec3);CHKERRQ(ierr); 929304d26faSStefano Zampini ierr = VecAXPY(vec3,m_one,vec1);CHKERRQ(ierr); 930304d26faSStefano Zampini ierr = VecNorm(vec3,NORM_INFINITY,&value);CHKERRQ(ierr); 931304d26faSStefano Zampini ierr = VecDestroy(&vec1);CHKERRQ(ierr); 932304d26faSStefano Zampini ierr = VecDestroy(&vec2);CHKERRQ(ierr); 933304d26faSStefano Zampini ierr = VecDestroy(&vec3);CHKERRQ(ierr); 934304d26faSStefano Zampini /* need to be adapted? */ 935304d26faSStefano Zampini use_exact = (PetscAbsReal(value) > 1.e-4 ? 0 : 1); 936304d26faSStefano Zampini ierr = MPI_Allreduce(&use_exact,&use_exact_reduced,1,MPIU_INT,MPI_LAND,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 937304d26faSStefano Zampini ierr = PCBDDCSetUseExactDirichlet(pc,(PetscBool)use_exact_reduced);CHKERRQ(ierr); 938304d26faSStefano Zampini /* print info */ 939304d26faSStefano Zampini if (pcbddc->dbg_flag) { 940304d26faSStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 941304d26faSStefano Zampini ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"--------------------------------------------------\n");CHKERRQ(ierr); 942304d26faSStefano Zampini ierr = PetscViewerASCIIPrintf(pcbddc->dbg_viewer,"Checking solution of Dirichlet and Neumann problems\n");CHKERRQ(ierr); 943304d26faSStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d infinity error for Dirichlet solve = % 1.14e \n",PetscGlobalRank,value);CHKERRQ(ierr); 944304d26faSStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 945304d26faSStefano Zampini } 946304d26faSStefano Zampini if (n_D && pcbddc->NullSpace && !use_exact_reduced && !pcbddc->inexact_prec_type) { 947304d26faSStefano Zampini ierr = PCBDDCNullSpaceAssembleCorrection(pc,is_I_local);CHKERRQ(ierr); 948304d26faSStefano Zampini } 949304d26faSStefano Zampini 950304d26faSStefano Zampini /* Neumann */ 951304d26faSStefano Zampini ierr = MatGetVecs(A_RR,&vec1,&vec2);CHKERRQ(ierr); 952304d26faSStefano Zampini ierr = VecDuplicate(vec1,&vec3);CHKERRQ(ierr); 953304d26faSStefano Zampini ierr = VecSetRandom(vec1,NULL);CHKERRQ(ierr); 954304d26faSStefano Zampini ierr = MatMult(A_RR,vec1,vec2);CHKERRQ(ierr); 955304d26faSStefano Zampini ierr = KSPSolve(pcbddc->ksp_R,vec2,vec3);CHKERRQ(ierr); 956304d26faSStefano Zampini ierr = VecAXPY(vec3,m_one,vec1);CHKERRQ(ierr); 957304d26faSStefano Zampini ierr = VecNorm(vec3,NORM_INFINITY,&value);CHKERRQ(ierr); 958304d26faSStefano Zampini ierr = VecDestroy(&vec1);CHKERRQ(ierr); 959304d26faSStefano Zampini ierr = VecDestroy(&vec2);CHKERRQ(ierr); 960304d26faSStefano Zampini ierr = VecDestroy(&vec3);CHKERRQ(ierr); 961304d26faSStefano Zampini /* need to be adapted? */ 962304d26faSStefano Zampini use_exact = (PetscAbsReal(value) > 1.e-4 ? 0 : 1); 963304d26faSStefano Zampini if (PetscAbsReal(value) > 1.e-4) use_exact = 0; 964304d26faSStefano Zampini ierr = MPI_Allreduce(&use_exact,&use_exact_reduced,1,MPIU_INT,MPI_LAND,PetscObjectComm((PetscObject)pc));CHKERRQ(ierr); 965304d26faSStefano Zampini /* print info */ 966304d26faSStefano Zampini if (pcbddc->dbg_flag) { 967304d26faSStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d infinity error for Neumann solve = % 1.14e \n",PetscGlobalRank,value);CHKERRQ(ierr); 968304d26faSStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 969304d26faSStefano Zampini } 970304d26faSStefano Zampini if (n_R && pcbddc->NullSpace && !use_exact_reduced) { /* is it the right logic? */ 971304d26faSStefano Zampini ierr = PCBDDCNullSpaceAssembleCorrection(pc,is_R_local);CHKERRQ(ierr); 972304d26faSStefano Zampini } 973304d26faSStefano Zampini 974304d26faSStefano Zampini /* free Neumann problem's matrix */ 975304d26faSStefano Zampini ierr = MatDestroy(&A_RR);CHKERRQ(ierr); 976304d26faSStefano Zampini PetscFunctionReturn(0); 977304d26faSStefano Zampini } 978304d26faSStefano Zampini 979304d26faSStefano Zampini #undef __FUNCT__ 980674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSolveSaddlePoint" 981674ae819SStefano Zampini static PetscErrorCode PCBDDCSolveSaddlePoint(PC pc) 982674ae819SStefano Zampini { 983674ae819SStefano Zampini PetscErrorCode ierr; 984674ae819SStefano Zampini PC_BDDC* pcbddc = (PC_BDDC*)(pc->data); 985674ae819SStefano Zampini 986674ae819SStefano Zampini PetscFunctionBegin; 987674ae819SStefano Zampini ierr = KSPSolve(pcbddc->ksp_R,pcbddc->vec1_R,pcbddc->vec2_R);CHKERRQ(ierr); 988674ae819SStefano Zampini if (pcbddc->local_auxmat1) { 989674ae819SStefano Zampini ierr = MatMult(pcbddc->local_auxmat1,pcbddc->vec2_R,pcbddc->vec1_C);CHKERRQ(ierr); 990674ae819SStefano Zampini ierr = MatMultAdd(pcbddc->local_auxmat2,pcbddc->vec1_C,pcbddc->vec2_R,pcbddc->vec2_R);CHKERRQ(ierr); 991674ae819SStefano Zampini } 992674ae819SStefano Zampini PetscFunctionReturn(0); 993674ae819SStefano Zampini } 994674ae819SStefano Zampini 995674ae819SStefano Zampini #undef __FUNCT__ 996674ae819SStefano Zampini #define __FUNCT__ "PCBDDCApplyInterfacePreconditioner" 997674ae819SStefano Zampini PetscErrorCode PCBDDCApplyInterfacePreconditioner(PC pc) 998674ae819SStefano Zampini { 999674ae819SStefano Zampini PetscErrorCode ierr; 1000674ae819SStefano Zampini PC_BDDC* pcbddc = (PC_BDDC*)(pc->data); 1001674ae819SStefano Zampini PC_IS* pcis = (PC_IS*) (pc->data); 1002674ae819SStefano Zampini const PetscScalar zero = 0.0; 1003674ae819SStefano Zampini 1004674ae819SStefano Zampini PetscFunctionBegin; 100515aaf578SStefano Zampini /* Application of PHI^T (or PSI^T) */ 100615aaf578SStefano Zampini if (pcbddc->coarse_psi_B) { 100715aaf578SStefano Zampini ierr = MatMultTranspose(pcbddc->coarse_psi_B,pcis->vec1_B,pcbddc->vec1_P);CHKERRQ(ierr); 100815aaf578SStefano Zampini if (pcbddc->inexact_prec_type) { ierr = MatMultTransposeAdd(pcbddc->coarse_psi_D,pcis->vec1_D,pcbddc->vec1_P,pcbddc->vec1_P);CHKERRQ(ierr); } 100915aaf578SStefano Zampini } else { 1010674ae819SStefano Zampini ierr = MatMultTranspose(pcbddc->coarse_phi_B,pcis->vec1_B,pcbddc->vec1_P);CHKERRQ(ierr); 1011674ae819SStefano Zampini if (pcbddc->inexact_prec_type) { ierr = MatMultTransposeAdd(pcbddc->coarse_phi_D,pcis->vec1_D,pcbddc->vec1_P,pcbddc->vec1_P);CHKERRQ(ierr); } 101215aaf578SStefano Zampini } 1013674ae819SStefano Zampini /* Scatter data of coarse_rhs */ 1014674ae819SStefano Zampini if (pcbddc->coarse_rhs) { ierr = VecSet(pcbddc->coarse_rhs,zero);CHKERRQ(ierr); } 1015674ae819SStefano Zampini ierr = PCBDDCScatterCoarseDataBegin(pc,pcbddc->vec1_P,pcbddc->coarse_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1016674ae819SStefano Zampini 1017674ae819SStefano Zampini /* Local solution on R nodes */ 1018674ae819SStefano Zampini ierr = VecSet(pcbddc->vec1_R,zero);CHKERRQ(ierr); 1019674ae819SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_B,pcis->vec1_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1020674ae819SStefano Zampini ierr = VecScatterEnd (pcbddc->R_to_B,pcis->vec1_B,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1021674ae819SStefano Zampini if (pcbddc->inexact_prec_type) { 1022674ae819SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_D,pcis->vec1_D,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1023674ae819SStefano Zampini ierr = VecScatterEnd (pcbddc->R_to_D,pcis->vec1_D,pcbddc->vec1_R,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1024674ae819SStefano Zampini } 1025674ae819SStefano Zampini ierr = PCBDDCSolveSaddlePoint(pc);CHKERRQ(ierr); 1026674ae819SStefano Zampini ierr = VecSet(pcis->vec1_B,zero);CHKERRQ(ierr); 1027674ae819SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_B,pcbddc->vec2_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1028674ae819SStefano Zampini ierr = VecScatterEnd (pcbddc->R_to_B,pcbddc->vec2_R,pcis->vec1_B,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1029674ae819SStefano Zampini if (pcbddc->inexact_prec_type) { 1030674ae819SStefano Zampini ierr = VecScatterBegin(pcbddc->R_to_D,pcbddc->vec2_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1031674ae819SStefano Zampini ierr = VecScatterEnd (pcbddc->R_to_D,pcbddc->vec2_R,pcis->vec1_D,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1032674ae819SStefano Zampini } 1033674ae819SStefano Zampini 1034674ae819SStefano Zampini /* Coarse solution */ 1035674ae819SStefano Zampini ierr = PCBDDCScatterCoarseDataEnd(pc,pcbddc->vec1_P,pcbddc->coarse_rhs,ADD_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1036674ae819SStefano Zampini if (pcbddc->coarse_rhs) { /* TODO remove null space when doing multilevel */ 1037674ae819SStefano Zampini ierr = KSPSolve(pcbddc->coarse_ksp,pcbddc->coarse_rhs,pcbddc->coarse_vec);CHKERRQ(ierr); 1038674ae819SStefano Zampini } 1039674ae819SStefano Zampini ierr = PCBDDCScatterCoarseDataBegin(pc,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1040674ae819SStefano Zampini ierr = PCBDDCScatterCoarseDataEnd (pc,pcbddc->coarse_vec,pcbddc->vec1_P,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 1041674ae819SStefano Zampini 1042674ae819SStefano Zampini /* Sum contributions from two levels */ 1043674ae819SStefano Zampini ierr = MatMultAdd(pcbddc->coarse_phi_B,pcbddc->vec1_P,pcis->vec1_B,pcis->vec1_B);CHKERRQ(ierr); 1044674ae819SStefano Zampini if (pcbddc->inexact_prec_type) { ierr = MatMultAdd(pcbddc->coarse_phi_D,pcbddc->vec1_P,pcis->vec1_D,pcis->vec1_D);CHKERRQ(ierr); } 1045674ae819SStefano Zampini PetscFunctionReturn(0); 1046674ae819SStefano Zampini } 1047674ae819SStefano Zampini 1048674ae819SStefano Zampini #undef __FUNCT__ 1049674ae819SStefano Zampini #define __FUNCT__ "PCBDDCScatterCoarseDataBegin" 1050674ae819SStefano Zampini PetscErrorCode PCBDDCScatterCoarseDataBegin(PC pc,Vec vec_from, Vec vec_to, InsertMode imode, ScatterMode smode) 1051674ae819SStefano Zampini { 1052674ae819SStefano Zampini PetscErrorCode ierr; 1053674ae819SStefano Zampini PC_BDDC* pcbddc = (PC_BDDC*)(pc->data); 1054674ae819SStefano Zampini 1055674ae819SStefano Zampini PetscFunctionBegin; 1056674ae819SStefano Zampini switch (pcbddc->coarse_communications_type) { 1057674ae819SStefano Zampini case SCATTERS_BDDC: 1058674ae819SStefano Zampini ierr = VecScatterBegin(pcbddc->coarse_loc_to_glob,vec_from,vec_to,imode,smode);CHKERRQ(ierr); 1059674ae819SStefano Zampini break; 1060674ae819SStefano Zampini case GATHERS_BDDC: 1061674ae819SStefano Zampini break; 1062674ae819SStefano Zampini } 1063674ae819SStefano Zampini PetscFunctionReturn(0); 1064674ae819SStefano Zampini } 1065674ae819SStefano Zampini 1066674ae819SStefano Zampini #undef __FUNCT__ 1067674ae819SStefano Zampini #define __FUNCT__ "PCBDDCScatterCoarseDataEnd" 1068674ae819SStefano Zampini PetscErrorCode PCBDDCScatterCoarseDataEnd(PC pc,Vec vec_from, Vec vec_to, InsertMode imode, ScatterMode smode) 1069674ae819SStefano Zampini { 1070674ae819SStefano Zampini PetscErrorCode ierr; 1071674ae819SStefano Zampini PC_BDDC* pcbddc = (PC_BDDC*)(pc->data); 1072674ae819SStefano Zampini PetscScalar* array_to; 1073674ae819SStefano Zampini PetscScalar* array_from; 1074674ae819SStefano Zampini MPI_Comm comm; 1075674ae819SStefano Zampini PetscInt i; 1076674ae819SStefano Zampini 1077674ae819SStefano Zampini PetscFunctionBegin; 1078674ae819SStefano Zampini ierr = PetscObjectGetComm((PetscObject)pc,&comm);CHKERRQ(ierr); 1079674ae819SStefano Zampini switch (pcbddc->coarse_communications_type) { 1080674ae819SStefano Zampini case SCATTERS_BDDC: 1081674ae819SStefano Zampini ierr = VecScatterEnd(pcbddc->coarse_loc_to_glob,vec_from,vec_to,imode,smode);CHKERRQ(ierr); 1082674ae819SStefano Zampini break; 1083674ae819SStefano Zampini case GATHERS_BDDC: 1084674ae819SStefano Zampini if (vec_from) { 1085674ae819SStefano Zampini ierr = VecGetArray(vec_from,&array_from);CHKERRQ(ierr); 1086674ae819SStefano Zampini } 1087674ae819SStefano Zampini if (vec_to) { 1088674ae819SStefano Zampini ierr = VecGetArray(vec_to,&array_to);CHKERRQ(ierr); 1089674ae819SStefano Zampini } 1090674ae819SStefano Zampini switch(pcbddc->coarse_problem_type){ 1091674ae819SStefano Zampini case SEQUENTIAL_BDDC: 1092674ae819SStefano Zampini if (smode == SCATTER_FORWARD) { 1093674ae819SStefano Zampini ierr = MPI_Gatherv(&array_from[0],pcbddc->local_primal_size,MPIU_SCALAR,&pcbddc->replicated_local_primal_values[0],pcbddc->local_primal_sizes,pcbddc->local_primal_displacements,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 1094674ae819SStefano Zampini if (vec_to) { 1095674ae819SStefano Zampini if (imode == ADD_VALUES) { 1096674ae819SStefano Zampini for (i=0;i<pcbddc->replicated_primal_size;i++) { 1097674ae819SStefano Zampini array_to[pcbddc->replicated_local_primal_indices[i]]+=pcbddc->replicated_local_primal_values[i]; 1098674ae819SStefano Zampini } 1099674ae819SStefano Zampini } else { 1100674ae819SStefano Zampini for (i=0;i<pcbddc->replicated_primal_size;i++) { 1101674ae819SStefano Zampini array_to[pcbddc->replicated_local_primal_indices[i]]=pcbddc->replicated_local_primal_values[i]; 1102674ae819SStefano Zampini } 1103674ae819SStefano Zampini } 1104674ae819SStefano Zampini } 1105674ae819SStefano Zampini } else { 1106674ae819SStefano Zampini if (vec_from) { 1107674ae819SStefano Zampini if (imode == ADD_VALUES) { 1108674ae819SStefano Zampini MPI_Comm vec_from_comm; 1109674ae819SStefano Zampini ierr = PetscObjectGetComm((PetscObject)(vec_from),&vec_from_comm);CHKERRQ(ierr); 1110674ae819SStefano Zampini SETERRQ2(vec_from_comm,PETSC_ERR_SUP,"Unsupported insert mode ADD_VALUES for SCATTER_REVERSE in %s for case %d\n",__FUNCT__,pcbddc->coarse_problem_type); 1111674ae819SStefano Zampini } 1112674ae819SStefano Zampini for (i=0;i<pcbddc->replicated_primal_size;i++) { 1113674ae819SStefano Zampini pcbddc->replicated_local_primal_values[i]=array_from[pcbddc->replicated_local_primal_indices[i]]; 1114674ae819SStefano Zampini } 1115674ae819SStefano Zampini } 1116674ae819SStefano Zampini ierr = MPI_Scatterv(&pcbddc->replicated_local_primal_values[0],pcbddc->local_primal_sizes,pcbddc->local_primal_displacements,MPIU_SCALAR,&array_to[0],pcbddc->local_primal_size,MPIU_SCALAR,0,comm);CHKERRQ(ierr); 1117674ae819SStefano Zampini } 1118674ae819SStefano Zampini break; 1119674ae819SStefano Zampini case REPLICATED_BDDC: 1120674ae819SStefano Zampini if (smode == SCATTER_FORWARD) { 1121674ae819SStefano Zampini ierr = MPI_Allgatherv(&array_from[0],pcbddc->local_primal_size,MPIU_SCALAR,&pcbddc->replicated_local_primal_values[0],pcbddc->local_primal_sizes,pcbddc->local_primal_displacements,MPIU_SCALAR,comm);CHKERRQ(ierr); 1122674ae819SStefano Zampini if (imode == ADD_VALUES) { 1123674ae819SStefano Zampini for (i=0;i<pcbddc->replicated_primal_size;i++) { 1124674ae819SStefano Zampini array_to[pcbddc->replicated_local_primal_indices[i]]+=pcbddc->replicated_local_primal_values[i]; 1125674ae819SStefano Zampini } 1126674ae819SStefano Zampini } else { 1127674ae819SStefano Zampini for (i=0;i<pcbddc->replicated_primal_size;i++) { 1128674ae819SStefano Zampini array_to[pcbddc->replicated_local_primal_indices[i]]=pcbddc->replicated_local_primal_values[i]; 1129674ae819SStefano Zampini } 1130674ae819SStefano Zampini } 1131674ae819SStefano Zampini } else { /* no communications needed for SCATTER_REVERSE since needed data is already present */ 1132674ae819SStefano Zampini if (imode == ADD_VALUES) { 1133674ae819SStefano Zampini for (i=0;i<pcbddc->local_primal_size;i++) { 1134674ae819SStefano Zampini array_to[i]+=array_from[pcbddc->local_primal_indices[i]]; 1135674ae819SStefano Zampini } 1136674ae819SStefano Zampini } else { 1137674ae819SStefano Zampini for (i=0;i<pcbddc->local_primal_size;i++) { 1138674ae819SStefano Zampini array_to[i]=array_from[pcbddc->local_primal_indices[i]]; 1139674ae819SStefano Zampini } 1140674ae819SStefano Zampini } 1141674ae819SStefano Zampini } 1142674ae819SStefano Zampini break; 1143674ae819SStefano Zampini case MULTILEVEL_BDDC: 1144674ae819SStefano Zampini break; 1145674ae819SStefano Zampini case PARALLEL_BDDC: 1146674ae819SStefano Zampini break; 1147674ae819SStefano Zampini } 1148674ae819SStefano Zampini if (vec_from) { 1149674ae819SStefano Zampini ierr = VecRestoreArray(vec_from,&array_from);CHKERRQ(ierr); 1150674ae819SStefano Zampini } 1151674ae819SStefano Zampini if (vec_to) { 1152674ae819SStefano Zampini ierr = VecRestoreArray(vec_to,&array_to);CHKERRQ(ierr); 1153674ae819SStefano Zampini } 1154674ae819SStefano Zampini break; 1155674ae819SStefano Zampini } 1156674ae819SStefano Zampini PetscFunctionReturn(0); 1157674ae819SStefano Zampini } 1158674ae819SStefano Zampini 1159984c4197SStefano Zampini /* uncomment for testing purposes */ 1160984c4197SStefano Zampini /* #define PETSC_MISSING_LAPACK_GESVD 1 */ 1161674ae819SStefano Zampini #undef __FUNCT__ 1162674ae819SStefano Zampini #define __FUNCT__ "PCBDDCConstraintsSetUp" 1163674ae819SStefano Zampini PetscErrorCode PCBDDCConstraintsSetUp(PC pc) 1164674ae819SStefano Zampini { 1165674ae819SStefano Zampini PetscErrorCode ierr; 1166674ae819SStefano Zampini PC_IS* pcis = (PC_IS*)(pc->data); 1167674ae819SStefano Zampini PC_BDDC* pcbddc = (PC_BDDC*)pc->data; 1168674ae819SStefano Zampini Mat_IS* matis = (Mat_IS*)pc->pmat->data; 1169984c4197SStefano Zampini /* constraint and (optionally) change of basis matrix implemented as SeqAIJ */ 1170674ae819SStefano Zampini MatType impMatType=MATSEQAIJ; 1171984c4197SStefano Zampini /* one and zero */ 1172984c4197SStefano Zampini PetscScalar one=1.0,zero=0.0; 1173984c4197SStefano Zampini /* space to store constraints and their local indices */ 1174984c4197SStefano Zampini PetscScalar *temp_quadrature_constraint; 1175984c4197SStefano Zampini PetscInt *temp_indices,*temp_indices_to_constraint,*temp_indices_to_constraint_B; 1176984c4197SStefano Zampini /* iterators */ 1177984c4197SStefano Zampini PetscInt i,j,k,total_counts,temp_start_ptr; 1178984c4197SStefano Zampini /* stuff to store connected components stored in pcbddc->mat_graph */ 1179984c4197SStefano Zampini IS ISForVertices,*ISForFaces,*ISForEdges,*used_IS; 1180984c4197SStefano Zampini PetscInt n_ISForFaces,n_ISForEdges; 1181984c4197SStefano Zampini PetscBool get_faces,get_edges,get_vertices; 1182984c4197SStefano Zampini /* near null space stuff */ 1183674ae819SStefano Zampini MatNullSpace nearnullsp; 1184674ae819SStefano Zampini const Vec *nearnullvecs; 1185674ae819SStefano Zampini Vec *localnearnullsp; 1186984c4197SStefano Zampini PetscBool nnsp_has_cnst; 1187984c4197SStefano Zampini PetscInt nnsp_size; 1188984c4197SStefano Zampini PetscScalar *array; 1189984c4197SStefano Zampini /* BLAS integers */ 1190e310c8b4SStefano Zampini PetscBLASInt lwork,lierr; 1191e310c8b4SStefano Zampini PetscBLASInt Blas_N,Blas_M,Blas_K,Blas_one=1; 1192c4303822SStefano Zampini PetscBLASInt Blas_LDA,Blas_LDB,Blas_LDC; 1193984c4197SStefano Zampini /* LAPACK working arrays for SVD or POD */ 1194242a89d7SStefano Zampini PetscBool skip_lapack; 1195984c4197SStefano Zampini PetscScalar *work; 1196984c4197SStefano Zampini PetscReal *singular_vals; 1197984c4197SStefano Zampini #if defined(PETSC_USE_COMPLEX) 1198984c4197SStefano Zampini PetscReal *rwork; 1199674ae819SStefano Zampini #endif 1200984c4197SStefano Zampini #if defined(PETSC_MISSING_LAPACK_GESVD) 1201e310c8b4SStefano Zampini PetscBLASInt Blas_one_2=1; 1202984c4197SStefano Zampini PetscScalar *temp_basis,*correlation_mat; 1203b7d8b9f8SStefano Zampini #else 1204b7d8b9f8SStefano Zampini PetscBLASInt dummy_int_1=1,dummy_int_2=1; 1205b7d8b9f8SStefano Zampini PetscScalar dummy_scalar_1=0.0,dummy_scalar_2=0.0; 1206984c4197SStefano Zampini #endif 1207984c4197SStefano Zampini /* change of basis */ 1208984c4197SStefano Zampini PetscInt *aux_primal_numbering,*aux_primal_minloc,*global_indices; 1209984c4197SStefano Zampini PetscBool boolforchange,*change_basis,*touched; 1210984c4197SStefano Zampini /* auxiliary stuff */ 1211984c4197SStefano Zampini PetscInt *nnz,*is_indices,*local_to_B; 1212984c4197SStefano Zampini /* some quantities */ 1213984c4197SStefano Zampini PetscInt n_vertices,total_primal_vertices; 1214984c4197SStefano Zampini PetscInt size_of_constraint,max_size_of_constraint,max_constraints,temp_constraints; 1215984c4197SStefano Zampini 1216674ae819SStefano Zampini 1217674ae819SStefano Zampini PetscFunctionBegin; 1218674ae819SStefano Zampini /* Get index sets for faces, edges and vertices from graph */ 1219674ae819SStefano Zampini get_faces = PETSC_TRUE; 1220674ae819SStefano Zampini get_edges = PETSC_TRUE; 1221674ae819SStefano Zampini get_vertices = PETSC_TRUE; 1222674ae819SStefano Zampini if (pcbddc->vertices_flag) { 1223674ae819SStefano Zampini get_faces = PETSC_FALSE; 1224674ae819SStefano Zampini get_edges = PETSC_FALSE; 1225674ae819SStefano Zampini } 1226674ae819SStefano Zampini if (pcbddc->constraints_flag) { 1227674ae819SStefano Zampini get_vertices = PETSC_FALSE; 1228674ae819SStefano Zampini } 1229674ae819SStefano Zampini if (pcbddc->faces_flag) { 1230674ae819SStefano Zampini get_edges = PETSC_FALSE; 1231674ae819SStefano Zampini } 1232674ae819SStefano Zampini if (pcbddc->edges_flag) { 1233674ae819SStefano Zampini get_faces = PETSC_FALSE; 1234674ae819SStefano Zampini } 1235674ae819SStefano Zampini /* default */ 1236674ae819SStefano Zampini if (!get_faces && !get_edges && !get_vertices) { 1237674ae819SStefano Zampini get_vertices = PETSC_TRUE; 1238674ae819SStefano Zampini } 1239674ae819SStefano Zampini ierr = PCBDDCGraphGetCandidatesIS(pcbddc->mat_graph,get_faces,get_edges,get_vertices,&n_ISForFaces,&ISForFaces,&n_ISForEdges,&ISForEdges,&ISForVertices); 1240984c4197SStefano Zampini /* print some info */ 1241674ae819SStefano Zampini if (pcbddc->dbg_flag) { 1242674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"--------------------------------------------------------------\n");CHKERRQ(ierr); 1243674ae819SStefano Zampini i = 0; 1244674ae819SStefano Zampini if (ISForVertices) { 1245674ae819SStefano Zampini ierr = ISGetSize(ISForVertices,&i);CHKERRQ(ierr); 1246674ae819SStefano Zampini } 1247674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate vertices\n",PetscGlobalRank,i);CHKERRQ(ierr); 1248674ae819SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate edges\n",PetscGlobalRank,n_ISForEdges);CHKERRQ(ierr); 124915aaf578SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Subdomain %04d got %02d local candidate faces\n",PetscGlobalRank,n_ISForFaces);CHKERRQ(ierr); 1250674ae819SStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 1251674ae819SStefano Zampini } 1252674ae819SStefano Zampini /* check if near null space is attached to global mat */ 1253674ae819SStefano Zampini ierr = MatGetNearNullSpace(pc->pmat,&nearnullsp);CHKERRQ(ierr); 1254674ae819SStefano Zampini if (nearnullsp) { 1255674ae819SStefano Zampini ierr = MatNullSpaceGetVecs(nearnullsp,&nnsp_has_cnst,&nnsp_size,&nearnullvecs);CHKERRQ(ierr); 1256984c4197SStefano Zampini } else { /* if near null space is not provided BDDC uses constants by default */ 1257984c4197SStefano Zampini nnsp_size = 0; 1258674ae819SStefano Zampini nnsp_has_cnst = PETSC_TRUE; 1259674ae819SStefano Zampini } 1260984c4197SStefano Zampini /* get max number of constraints on a single cc */ 1261984c4197SStefano Zampini max_constraints = nnsp_size; 1262984c4197SStefano Zampini if (nnsp_has_cnst) max_constraints++; 1263984c4197SStefano Zampini 1264674ae819SStefano Zampini /* 1265674ae819SStefano Zampini Evaluate maximum storage size needed by the procedure 1266674ae819SStefano Zampini - temp_indices will contain start index of each constraint stored as follows 1267674ae819SStefano Zampini - temp_indices_to_constraint [temp_indices[i],...,temp[indices[i+1]-1] will contain the indices (in local numbering) on which the constraint acts 1268674ae819SStefano Zampini - temp_indices_to_constraint_B[temp_indices[i],...,temp[indices[i+1]-1] will contain the indices (in boundary numbering) on which the constraint acts 1269674ae819SStefano Zampini - temp_quadrature_constraint [temp_indices[i],...,temp[indices[i+1]-1] will contain the scalars representing the constraint itself 1270674ae819SStefano Zampini */ 1271674ae819SStefano Zampini total_counts = n_ISForFaces+n_ISForEdges; 1272984c4197SStefano Zampini total_counts *= max_constraints; 1273674ae819SStefano Zampini n_vertices = 0; 1274674ae819SStefano Zampini if (ISForVertices) { 1275674ae819SStefano Zampini ierr = ISGetSize(ISForVertices,&n_vertices);CHKERRQ(ierr); 1276674ae819SStefano Zampini } 1277674ae819SStefano Zampini total_counts += n_vertices; 1278674ae819SStefano Zampini ierr = PetscMalloc((total_counts+1)*sizeof(PetscInt),&temp_indices);CHKERRQ(ierr); 1279674ae819SStefano Zampini ierr = PetscMalloc((total_counts+1)*sizeof(PetscBool),&change_basis);CHKERRQ(ierr); 1280674ae819SStefano Zampini total_counts = 0; 1281674ae819SStefano Zampini max_size_of_constraint = 0; 1282674ae819SStefano Zampini for (i=0;i<n_ISForEdges+n_ISForFaces;i++) { 1283674ae819SStefano Zampini if (i<n_ISForEdges) { 1284674ae819SStefano Zampini used_IS = &ISForEdges[i]; 1285674ae819SStefano Zampini } else { 1286674ae819SStefano Zampini used_IS = &ISForFaces[i-n_ISForEdges]; 1287674ae819SStefano Zampini } 1288674ae819SStefano Zampini ierr = ISGetSize(*used_IS,&j);CHKERRQ(ierr); 1289674ae819SStefano Zampini total_counts += j; 1290674ae819SStefano Zampini max_size_of_constraint = PetscMax(j,max_size_of_constraint); 1291674ae819SStefano Zampini } 1292984c4197SStefano Zampini total_counts *= max_constraints; 1293674ae819SStefano Zampini total_counts += n_vertices; 1294674ae819SStefano Zampini ierr = PetscMalloc(total_counts*sizeof(PetscScalar),&temp_quadrature_constraint);CHKERRQ(ierr); 1295674ae819SStefano Zampini ierr = PetscMalloc(total_counts*sizeof(PetscInt),&temp_indices_to_constraint);CHKERRQ(ierr); 1296674ae819SStefano Zampini ierr = PetscMalloc(total_counts*sizeof(PetscInt),&temp_indices_to_constraint_B);CHKERRQ(ierr); 1297984c4197SStefano Zampini /* local to boundary numbering */ 1298674ae819SStefano Zampini ierr = PetscMalloc(pcis->n*sizeof(PetscInt),&local_to_B);CHKERRQ(ierr); 1299674ae819SStefano Zampini ierr = ISGetIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 1300984c4197SStefano Zampini for (i=0;i<pcis->n;i++) local_to_B[i]=-1; 1301984c4197SStefano Zampini for (i=0;i<pcis->n_B;i++) local_to_B[is_indices[i]]=i; 1302674ae819SStefano Zampini ierr = ISRestoreIndices(pcis->is_B_local,(const PetscInt**)&is_indices);CHKERRQ(ierr); 1303984c4197SStefano Zampini /* get local part of global near null space vectors */ 1304984c4197SStefano Zampini ierr = PetscMalloc(nnsp_size*sizeof(Vec),&localnearnullsp);CHKERRQ(ierr); 1305984c4197SStefano Zampini for (k=0;k<nnsp_size;k++) { 1306984c4197SStefano Zampini ierr = VecDuplicate(pcis->vec1_N,&localnearnullsp[k]);CHKERRQ(ierr); 1307984c4197SStefano Zampini ierr = VecScatterBegin(matis->ctx,nearnullvecs[k],localnearnullsp[k],INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1308984c4197SStefano Zampini ierr = VecScatterEnd(matis->ctx,nearnullvecs[k],localnearnullsp[k],INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 1309984c4197SStefano Zampini } 1310674ae819SStefano Zampini 1311242a89d7SStefano Zampini /* whether or not to skip lapack calls */ 1312242a89d7SStefano Zampini skip_lapack = PETSC_TRUE; 1313242a89d7SStefano Zampini if (n_ISForFaces+n_ISForEdges) skip_lapack = PETSC_FALSE; 1314242a89d7SStefano Zampini 1315984c4197SStefano Zampini /* First we issue queries to allocate optimal workspace for LAPACKgesvd (or LAPACKsyev if SVD is missing) */ 1316242a89d7SStefano Zampini if (!pcbddc->use_nnsp_true && !skip_lapack) { 1317674ae819SStefano Zampini PetscScalar temp_work; 1318674ae819SStefano Zampini #if defined(PETSC_MISSING_LAPACK_GESVD) 1319984c4197SStefano Zampini /* Proper Orthogonal Decomposition (POD) using the snapshot method */ 1320984c4197SStefano Zampini ierr = PetscMalloc(max_constraints*max_constraints*sizeof(PetscScalar),&correlation_mat);CHKERRQ(ierr); 1321984c4197SStefano Zampini ierr = PetscMalloc(max_constraints*sizeof(PetscReal),&singular_vals);CHKERRQ(ierr); 1322984c4197SStefano Zampini ierr = PetscMalloc(max_size_of_constraint*max_constraints*sizeof(PetscScalar),&temp_basis);CHKERRQ(ierr); 1323674ae819SStefano Zampini #if defined(PETSC_USE_COMPLEX) 1324984c4197SStefano Zampini ierr = PetscMalloc(3*max_constraints*sizeof(PetscReal),&rwork);CHKERRQ(ierr); 1325674ae819SStefano Zampini #endif 1326674ae819SStefano Zampini /* now we evaluate the optimal workspace using query with lwork=-1 */ 1327c8244a33SStefano Zampini ierr = PetscBLASIntCast(max_constraints,&Blas_N);CHKERRQ(ierr); 1328c8244a33SStefano Zampini ierr = PetscBLASIntCast(max_constraints,&Blas_LDA);CHKERRQ(ierr); 1329674ae819SStefano Zampini lwork = -1; 1330674ae819SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1331674ae819SStefano Zampini #if !defined(PETSC_USE_COMPLEX) 1332c8244a33SStefano Zampini PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,&temp_work,&lwork,&lierr)); 1333674ae819SStefano Zampini #else 1334c8244a33SStefano Zampini PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,&temp_work,&lwork,rwork,&lierr)); 1335674ae819SStefano Zampini #endif 1336674ae819SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1337984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to SYEV Lapack routine %d",(int)lierr); 1338674ae819SStefano Zampini #else /* on missing GESVD */ 1339674ae819SStefano Zampini /* SVD */ 1340674ae819SStefano Zampini PetscInt max_n,min_n; 1341674ae819SStefano Zampini max_n = max_size_of_constraint; 1342984c4197SStefano Zampini min_n = max_constraints; 1343984c4197SStefano Zampini if (max_size_of_constraint < max_constraints) { 1344674ae819SStefano Zampini min_n = max_size_of_constraint; 1345984c4197SStefano Zampini max_n = max_constraints; 1346674ae819SStefano Zampini } 1347674ae819SStefano Zampini ierr = PetscMalloc(min_n*sizeof(PetscReal),&singular_vals);CHKERRQ(ierr); 1348674ae819SStefano Zampini #if defined(PETSC_USE_COMPLEX) 1349674ae819SStefano Zampini ierr = PetscMalloc(5*min_n*sizeof(PetscReal),&rwork);CHKERRQ(ierr); 1350674ae819SStefano Zampini #endif 1351674ae819SStefano Zampini /* now we evaluate the optimal workspace using query with lwork=-1 */ 1352674ae819SStefano Zampini lwork = -1; 1353e310c8b4SStefano Zampini ierr = PetscBLASIntCast(max_n,&Blas_M);CHKERRQ(ierr); 1354e310c8b4SStefano Zampini ierr = PetscBLASIntCast(min_n,&Blas_N);CHKERRQ(ierr); 1355b7d8b9f8SStefano Zampini ierr = PetscBLASIntCast(max_n,&Blas_LDA);CHKERRQ(ierr); 1356674ae819SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1357674ae819SStefano Zampini #if !defined(PETSC_USE_COMPLEX) 1358e310c8b4SStefano Zampini PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&temp_quadrature_constraint[0],&Blas_LDA,singular_vals,&dummy_scalar_1,&dummy_int_1,&dummy_scalar_2,&dummy_int_2,&temp_work,&lwork,&lierr)); 1359674ae819SStefano Zampini #else 1360e310c8b4SStefano Zampini PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&temp_quadrature_constraint[0],&Blas_LDA,singular_vals,&dummy_scalar_1,&dummy_int_1,&dummy_scalar_2,&dummy_int_2,&temp_work,&lwork,rwork,&lierr)); 1361674ae819SStefano Zampini #endif 1362674ae819SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1363984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to GESVD Lapack routine %d",(int)lierr); 1364984c4197SStefano Zampini #endif /* on missing GESVD */ 1365674ae819SStefano Zampini /* Allocate optimal workspace */ 1366674ae819SStefano Zampini ierr = PetscBLASIntCast((PetscInt)PetscRealPart(temp_work),&lwork);CHKERRQ(ierr); 1367984c4197SStefano Zampini ierr = PetscMalloc((PetscInt)lwork*sizeof(PetscScalar),&work);CHKERRQ(ierr); 1368674ae819SStefano Zampini } 1369674ae819SStefano Zampini /* Now we can loop on constraining sets */ 1370674ae819SStefano Zampini total_counts = 0; 1371674ae819SStefano Zampini temp_indices[0] = 0; 1372674ae819SStefano Zampini /* vertices */ 1373674ae819SStefano Zampini if (ISForVertices) { 1374674ae819SStefano Zampini ierr = ISGetIndices(ISForVertices,(const PetscInt**)&is_indices);CHKERRQ(ierr); 1375674ae819SStefano Zampini if (nnsp_has_cnst) { /* consider all vertices */ 1376674ae819SStefano Zampini for (i=0;i<n_vertices;i++) { 1377674ae819SStefano Zampini temp_indices_to_constraint[temp_indices[total_counts]]=is_indices[i]; 1378674ae819SStefano Zampini temp_indices_to_constraint_B[temp_indices[total_counts]]=local_to_B[is_indices[i]]; 1379674ae819SStefano Zampini temp_quadrature_constraint[temp_indices[total_counts]]=1.0; 1380674ae819SStefano Zampini temp_indices[total_counts+1]=temp_indices[total_counts]+1; 1381674ae819SStefano Zampini change_basis[total_counts]=PETSC_FALSE; 1382674ae819SStefano Zampini total_counts++; 1383674ae819SStefano Zampini } 1384674ae819SStefano Zampini } else { /* consider vertices for which exist at least a localnearnullsp which is not null there */ 1385984c4197SStefano Zampini PetscBool used_vertex; 1386674ae819SStefano Zampini for (i=0;i<n_vertices;i++) { 1387674ae819SStefano Zampini used_vertex = PETSC_FALSE; 1388674ae819SStefano Zampini k = 0; 1389674ae819SStefano Zampini while (!used_vertex && k<nnsp_size) { 1390984c4197SStefano Zampini ierr = VecGetArrayRead(localnearnullsp[k],(const PetscScalar**)&array);CHKERRQ(ierr); 1391984c4197SStefano Zampini if (PetscAbsScalar(array[is_indices[i]])>0.0) { 1392674ae819SStefano Zampini temp_indices_to_constraint[temp_indices[total_counts]]=is_indices[i]; 1393674ae819SStefano Zampini temp_indices_to_constraint_B[temp_indices[total_counts]]=local_to_B[is_indices[i]]; 1394674ae819SStefano Zampini temp_quadrature_constraint[temp_indices[total_counts]]=1.0; 1395674ae819SStefano Zampini temp_indices[total_counts+1]=temp_indices[total_counts]+1; 1396674ae819SStefano Zampini change_basis[total_counts]=PETSC_FALSE; 1397674ae819SStefano Zampini total_counts++; 1398674ae819SStefano Zampini used_vertex = PETSC_TRUE; 1399674ae819SStefano Zampini } 1400984c4197SStefano Zampini ierr = VecRestoreArrayRead(localnearnullsp[k],(const PetscScalar**)&array);CHKERRQ(ierr); 1401674ae819SStefano Zampini k++; 1402674ae819SStefano Zampini } 1403674ae819SStefano Zampini } 1404674ae819SStefano Zampini } 1405674ae819SStefano Zampini ierr = ISRestoreIndices(ISForVertices,(const PetscInt**)&is_indices);CHKERRQ(ierr); 1406674ae819SStefano Zampini n_vertices = total_counts; 1407674ae819SStefano Zampini } 1408984c4197SStefano Zampini 1409674ae819SStefano Zampini /* edges and faces */ 1410674ae819SStefano Zampini for (i=0;i<n_ISForEdges+n_ISForFaces;i++) { 1411674ae819SStefano Zampini if (i<n_ISForEdges) { 1412674ae819SStefano Zampini used_IS = &ISForEdges[i]; 1413984c4197SStefano Zampini boolforchange = pcbddc->use_change_of_basis; /* change or not the basis on the edge */ 1414674ae819SStefano Zampini } else { 1415674ae819SStefano Zampini used_IS = &ISForFaces[i-n_ISForEdges]; 1416984c4197SStefano Zampini boolforchange = (PetscBool)(pcbddc->use_change_of_basis && pcbddc->use_change_on_faces); /* change or not the basis on the face */ 1417674ae819SStefano Zampini } 1418674ae819SStefano Zampini temp_constraints = 0; /* zero the number of constraints I have on this conn comp */ 1419674ae819SStefano Zampini temp_start_ptr = total_counts; /* need to know the starting index of constraints stored */ 1420674ae819SStefano Zampini ierr = ISGetSize(*used_IS,&size_of_constraint);CHKERRQ(ierr); 1421674ae819SStefano Zampini ierr = ISGetIndices(*used_IS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 1422984c4197SStefano Zampini /* change of basis should not be performed on local periodic nodes */ 1423984c4197SStefano Zampini if (pcbddc->mat_graph->mirrors && pcbddc->mat_graph->mirrors[is_indices[0]]) boolforchange = PETSC_FALSE; 1424674ae819SStefano Zampini if (nnsp_has_cnst) { 14255b08dc53SStefano Zampini PetscScalar quad_value; 1426674ae819SStefano Zampini temp_constraints++; 1427674ae819SStefano Zampini quad_value = (PetscScalar)(1.0/PetscSqrtReal((PetscReal)size_of_constraint)); 1428674ae819SStefano Zampini for (j=0;j<size_of_constraint;j++) { 1429674ae819SStefano Zampini temp_indices_to_constraint[temp_indices[total_counts]+j]=is_indices[j]; 1430674ae819SStefano Zampini temp_indices_to_constraint_B[temp_indices[total_counts]+j]=local_to_B[is_indices[j]]; 1431674ae819SStefano Zampini temp_quadrature_constraint[temp_indices[total_counts]+j]=quad_value; 1432674ae819SStefano Zampini } 1433674ae819SStefano Zampini temp_indices[total_counts+1]=temp_indices[total_counts]+size_of_constraint; /* store new starting point */ 143451b0f6cfSStefano Zampini change_basis[total_counts]=boolforchange; 1435674ae819SStefano Zampini total_counts++; 1436674ae819SStefano Zampini } 1437674ae819SStefano Zampini for (k=0;k<nnsp_size;k++) { 1438984c4197SStefano Zampini PetscReal real_value; 1439984c4197SStefano Zampini ierr = VecGetArrayRead(localnearnullsp[k],(const PetscScalar**)&array);CHKERRQ(ierr); 1440674ae819SStefano Zampini for (j=0;j<size_of_constraint;j++) { 1441674ae819SStefano Zampini temp_indices_to_constraint[temp_indices[total_counts]+j]=is_indices[j]; 1442674ae819SStefano Zampini temp_indices_to_constraint_B[temp_indices[total_counts]+j]=local_to_B[is_indices[j]]; 1443984c4197SStefano Zampini temp_quadrature_constraint[temp_indices[total_counts]+j]=array[is_indices[j]]; 1444674ae819SStefano Zampini } 1445984c4197SStefano Zampini ierr = VecRestoreArrayRead(localnearnullsp[k],(const PetscScalar**)&array);CHKERRQ(ierr); 1446984c4197SStefano Zampini /* check if array is null on the connected component */ 1447e310c8b4SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr); 1448e310c8b4SStefano Zampini PetscStackCallBLAS("BLASasum",real_value = BLASasum_(&Blas_N,&temp_quadrature_constraint[temp_indices[total_counts]],&Blas_one)); 14495b08dc53SStefano Zampini if (real_value > 0.0) { /* keep indices and values */ 1450674ae819SStefano Zampini temp_constraints++; 1451674ae819SStefano Zampini temp_indices[total_counts+1]=temp_indices[total_counts]+size_of_constraint; /* store new starting point */ 145251b0f6cfSStefano Zampini change_basis[total_counts]=boolforchange; 1453674ae819SStefano Zampini total_counts++; 1454674ae819SStefano Zampini } 1455674ae819SStefano Zampini } 1456674ae819SStefano Zampini ierr = ISRestoreIndices(*used_IS,(const PetscInt**)&is_indices);CHKERRQ(ierr); 1457984c4197SStefano Zampini /* perform SVD on the constraints if use_nnsp_true has not be requested by the user */ 1458984c4197SStefano Zampini if (!pcbddc->use_nnsp_true) { 1459984c4197SStefano Zampini PetscReal tol = 1.0e-8; /* tolerance for retaining eigenmodes */ 1460674ae819SStefano Zampini 1461674ae819SStefano Zampini #if defined(PETSC_MISSING_LAPACK_GESVD) 1462984c4197SStefano Zampini /* SVD: Y = U*S*V^H -> U (eigenvectors of Y*Y^H) = Y*V*(S)^\dag 1463984c4197SStefano Zampini POD: Y^H*Y = V*D*V^H, D = S^H*S -> U = Y*V*D^(-1/2) 1464984c4197SStefano Zampini -> When PETSC_USE_COMPLEX and PETSC_MISSING_LAPACK_GESVD are defined 1465984c4197SStefano Zampini the constraints basis will differ (by a complex factor with absolute value equal to 1) 1466984c4197SStefano Zampini from that computed using LAPACKgesvd 1467984c4197SStefano Zampini -> This is due to a different computation of eigenvectors in LAPACKheev 1468984c4197SStefano Zampini -> The quality of the POD-computed basis will be the same */ 1469984c4197SStefano Zampini ierr = PetscMemzero(correlation_mat,temp_constraints*temp_constraints*sizeof(PetscScalar));CHKERRQ(ierr); 1470674ae819SStefano Zampini /* Store upper triangular part of correlation matrix */ 1471e310c8b4SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr); 1472984c4197SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1473674ae819SStefano Zampini for (j=0;j<temp_constraints;j++) { 1474674ae819SStefano Zampini for (k=0;k<j+1;k++) { 1475e310c8b4SStefano Zampini PetscStackCallBLAS("BLASdot",correlation_mat[j*temp_constraints+k]=BLASdot_(&Blas_N,&temp_quadrature_constraint[temp_indices[temp_start_ptr+k]],&Blas_one,&temp_quadrature_constraint[temp_indices[temp_start_ptr+j]],&Blas_one_2)); 1476674ae819SStefano Zampini } 1477674ae819SStefano Zampini } 1478e310c8b4SStefano Zampini /* compute eigenvalues and eigenvectors of correlation matrix */ 1479e310c8b4SStefano Zampini ierr = PetscBLASIntCast(temp_constraints,&Blas_N);CHKERRQ(ierr); 1480e310c8b4SStefano Zampini ierr = PetscBLASIntCast(temp_constraints,&Blas_LDA);CHKERRQ(ierr); 1481674ae819SStefano Zampini #if !defined(PETSC_USE_COMPLEX) 1482c8244a33SStefano Zampini PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,work,&lwork,&lierr)); 1483674ae819SStefano Zampini #else 1484c8244a33SStefano Zampini PetscStackCallBLAS("LAPACKsyev",LAPACKsyev_("V","U",&Blas_N,correlation_mat,&Blas_LDA,singular_vals,work,&lwork,rwork,&lierr)); 1485674ae819SStefano Zampini #endif 1486674ae819SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1487984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in SYEV Lapack routine %d",(int)lierr); 1488984c4197SStefano Zampini /* retain eigenvalues greater than tol: note that LAPACKsyev gives eigs in ascending order */ 1489674ae819SStefano Zampini j=0; 1490984c4197SStefano Zampini while (j < temp_constraints && singular_vals[j] < tol) j++; 1491674ae819SStefano Zampini total_counts=total_counts-j; 1492e310c8b4SStefano Zampini /* scale and copy POD basis into used quadrature memory */ 1493c4303822SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr); 1494c4303822SStefano Zampini ierr = PetscBLASIntCast(temp_constraints,&Blas_N);CHKERRQ(ierr); 1495c4303822SStefano Zampini ierr = PetscBLASIntCast(temp_constraints,&Blas_K);CHKERRQ(ierr); 1496c4303822SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 1497c4303822SStefano Zampini ierr = PetscBLASIntCast(temp_constraints,&Blas_LDB);CHKERRQ(ierr); 1498c4303822SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDC);CHKERRQ(ierr); 1499674ae819SStefano Zampini if (j<temp_constraints) { 1500984c4197SStefano Zampini PetscInt ii; 1501984c4197SStefano Zampini for (k=j;k<temp_constraints;k++) singular_vals[k]=1.0/PetscSqrtReal(singular_vals[k]); 1502674ae819SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1503c4303822SStefano Zampini PetscStackCallBLAS("BLASgemm",BLASgemm_("N","N",&Blas_M,&Blas_N,&Blas_K,&one,&temp_quadrature_constraint[temp_indices[temp_start_ptr]],&Blas_LDA,correlation_mat,&Blas_LDB,&zero,temp_basis,&Blas_LDC)); 1504674ae819SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1505984c4197SStefano Zampini for (k=0;k<temp_constraints-j;k++) { 1506674ae819SStefano Zampini for (ii=0;ii<size_of_constraint;ii++) { 1507984c4197SStefano Zampini temp_quadrature_constraint[temp_indices[temp_start_ptr+k]+ii]=singular_vals[temp_constraints-1-k]*temp_basis[(temp_constraints-1-k)*size_of_constraint+ii]; 1508674ae819SStefano Zampini } 1509674ae819SStefano Zampini } 1510674ae819SStefano Zampini } 1511674ae819SStefano Zampini #else /* on missing GESVD */ 1512e310c8b4SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr); 1513e310c8b4SStefano Zampini ierr = PetscBLASIntCast(temp_constraints,&Blas_N);CHKERRQ(ierr); 1514b7d8b9f8SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 1515674ae819SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1516674ae819SStefano Zampini #if !defined(PETSC_USE_COMPLEX) 1517e310c8b4SStefano Zampini PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&temp_quadrature_constraint[temp_indices[temp_start_ptr]],&Blas_LDA,singular_vals,&dummy_scalar_1,&dummy_int_1,&dummy_scalar_2,&dummy_int_2,work,&lwork,&lierr)); 1518674ae819SStefano Zampini #else 1519e310c8b4SStefano Zampini PetscStackCallBLAS("LAPACKgesvd",LAPACKgesvd_("O","N",&Blas_M,&Blas_N,&temp_quadrature_constraint[temp_indices[temp_start_ptr]],&Blas_LDA,singular_vals,&dummy_scalar_1,&dummy_int_1,&dummy_scalar_2,&dummy_int_2,work,&lwork,rwork,&lierr)); 1520674ae819SStefano Zampini #endif 1521984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GESVD Lapack routine %d",(int)lierr); 1522674ae819SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1523984c4197SStefano Zampini /* retain eigenvalues greater than tol: note that LAPACKgesvd gives eigs in descending order */ 1524e310c8b4SStefano Zampini k = temp_constraints; 1525e310c8b4SStefano Zampini if (k > size_of_constraint) k = size_of_constraint; 1526674ae819SStefano Zampini j = 0; 1527e310c8b4SStefano Zampini while (j < k && singular_vals[k-j-1] < tol) j++; 1528e310c8b4SStefano Zampini total_counts = total_counts-temp_constraints+k-j; 1529984c4197SStefano Zampini #endif /* on missing GESVD */ 1530674ae819SStefano Zampini } 1531674ae819SStefano Zampini } 1532674ae819SStefano Zampini /* free index sets of faces, edges and vertices */ 1533674ae819SStefano Zampini for (i=0;i<n_ISForFaces;i++) { 1534674ae819SStefano Zampini ierr = ISDestroy(&ISForFaces[i]);CHKERRQ(ierr); 1535674ae819SStefano Zampini } 1536674ae819SStefano Zampini ierr = PetscFree(ISForFaces);CHKERRQ(ierr); 1537674ae819SStefano Zampini for (i=0;i<n_ISForEdges;i++) { 1538674ae819SStefano Zampini ierr = ISDestroy(&ISForEdges[i]);CHKERRQ(ierr); 1539674ae819SStefano Zampini } 1540674ae819SStefano Zampini ierr = PetscFree(ISForEdges);CHKERRQ(ierr); 1541674ae819SStefano Zampini ierr = ISDestroy(&ISForVertices);CHKERRQ(ierr); 1542674ae819SStefano Zampini 1543984c4197SStefano Zampini /* free workspace */ 1544242a89d7SStefano Zampini if (!pcbddc->use_nnsp_true && !skip_lapack) { 1545984c4197SStefano Zampini ierr = PetscFree(work);CHKERRQ(ierr); 1546984c4197SStefano Zampini #if defined(PETSC_USE_COMPLEX) 1547984c4197SStefano Zampini ierr = PetscFree(rwork);CHKERRQ(ierr); 1548984c4197SStefano Zampini #endif 1549984c4197SStefano Zampini ierr = PetscFree(singular_vals);CHKERRQ(ierr); 1550984c4197SStefano Zampini #if defined(PETSC_MISSING_LAPACK_GESVD) 1551984c4197SStefano Zampini ierr = PetscFree(correlation_mat);CHKERRQ(ierr); 1552984c4197SStefano Zampini ierr = PetscFree(temp_basis);CHKERRQ(ierr); 1553984c4197SStefano Zampini #endif 1554984c4197SStefano Zampini } 1555984c4197SStefano Zampini for (k=0;k<nnsp_size;k++) { 1556984c4197SStefano Zampini ierr = VecDestroy(&localnearnullsp[k]);CHKERRQ(ierr); 1557984c4197SStefano Zampini } 1558984c4197SStefano Zampini ierr = PetscFree(localnearnullsp);CHKERRQ(ierr); 1559984c4197SStefano Zampini 1560674ae819SStefano Zampini /* set quantities in pcbddc data structure */ 1561984c4197SStefano Zampini /* n_vertices defines the number of subdomain corners in the primal space */ 1562674ae819SStefano Zampini /* n_constraints defines the number of averages (they can be point primal dofs if change of basis is requested) */ 1563984c4197SStefano Zampini pcbddc->local_primal_size = total_counts; 1564674ae819SStefano Zampini pcbddc->n_vertices = n_vertices; 1565984c4197SStefano Zampini pcbddc->n_constraints = pcbddc->local_primal_size-pcbddc->n_vertices; 1566674ae819SStefano Zampini 1567674ae819SStefano Zampini /* Create constraint matrix */ 1568674ae819SStefano Zampini /* The constraint matrix is used to compute the l2g map of primal dofs */ 1569674ae819SStefano Zampini /* so we need to set it up properly either with or without change of basis */ 1570674ae819SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->ConstraintMatrix);CHKERRQ(ierr); 1571674ae819SStefano Zampini ierr = MatSetType(pcbddc->ConstraintMatrix,impMatType);CHKERRQ(ierr); 1572984c4197SStefano Zampini ierr = MatSetSizes(pcbddc->ConstraintMatrix,pcbddc->local_primal_size,pcis->n,pcbddc->local_primal_size,pcis->n);CHKERRQ(ierr); 1573984c4197SStefano Zampini /* array to compute a local numbering of constraints : vertices first then constraints */ 1574984c4197SStefano Zampini ierr = PetscMalloc(pcbddc->local_primal_size*sizeof(PetscInt),&aux_primal_numbering);CHKERRQ(ierr); 1575984c4197SStefano Zampini /* array to select the proper local node (of minimum index with respect to global ordering) when changing the basis */ 1576984c4197SStefano Zampini /* note: it should not be needed since IS for faces and edges are already sorted by global ordering when analyzing the graph but... just in case */ 1577984c4197SStefano Zampini ierr = PetscMalloc(pcbddc->local_primal_size*sizeof(PetscInt),&aux_primal_minloc);CHKERRQ(ierr); 1578984c4197SStefano Zampini /* auxiliary stuff for basis change */ 1579984c4197SStefano Zampini ierr = PetscMalloc(max_size_of_constraint*sizeof(PetscInt),&global_indices);CHKERRQ(ierr); 1580984c4197SStefano Zampini ierr = PetscMalloc(pcis->n_B*sizeof(PetscBool),&touched);CHKERRQ(ierr); 1581984c4197SStefano Zampini ierr = PetscMemzero(touched,pcis->n_B*sizeof(PetscBool));CHKERRQ(ierr); 1582984c4197SStefano Zampini 1583984c4197SStefano Zampini /* find primal_dofs: subdomain corners plus dofs selected as primal after change of basis */ 1584984c4197SStefano Zampini total_primal_vertices=0; 1585984c4197SStefano Zampini for (i=0;i<pcbddc->local_primal_size;i++) { 1586674ae819SStefano Zampini size_of_constraint=temp_indices[i+1]-temp_indices[i]; 1587984c4197SStefano Zampini if (size_of_constraint == 1) { 1588984c4197SStefano Zampini touched[temp_indices_to_constraint_B[temp_indices[i]]]=PETSC_TRUE; 1589984c4197SStefano Zampini aux_primal_numbering[total_primal_vertices]=temp_indices_to_constraint[temp_indices[i]]; 1590984c4197SStefano Zampini aux_primal_minloc[total_primal_vertices]=0; 1591984c4197SStefano Zampini total_primal_vertices++; 1592984c4197SStefano Zampini } else if (change_basis[i]) { /* Same procedure used in PCBDDCGetPrimalConstraintsLocalIdx */ 1593984c4197SStefano Zampini PetscInt min_loc,min_index; 1594984c4197SStefano Zampini ierr = ISLocalToGlobalMappingApply(pcbddc->mat_graph->l2gmap,size_of_constraint,&temp_indices_to_constraint[temp_indices[i]],global_indices);CHKERRQ(ierr); 1595984c4197SStefano Zampini /* find first untouched local node */ 1596674ae819SStefano Zampini k = 0; 1597984c4197SStefano Zampini while (touched[temp_indices_to_constraint_B[temp_indices[i]+k]]) k++; 1598984c4197SStefano Zampini min_index = global_indices[k]; 1599984c4197SStefano Zampini min_loc = k; 1600984c4197SStefano Zampini /* search the minimum among global nodes already untouched on the cc */ 1601984c4197SStefano Zampini for (k=1;k<size_of_constraint;k++) { 1602984c4197SStefano Zampini /* there can be more than one constraint on a single connected component */ 1603984c4197SStefano Zampini if (min_index > global_indices[k] && !touched[temp_indices_to_constraint_B[temp_indices[i]+k]]) { 1604984c4197SStefano Zampini min_index = global_indices[k]; 1605984c4197SStefano Zampini min_loc = k; 1606674ae819SStefano Zampini } 1607674ae819SStefano Zampini } 1608984c4197SStefano Zampini touched[temp_indices_to_constraint_B[temp_indices[i]+min_loc]] = PETSC_TRUE; 1609984c4197SStefano Zampini aux_primal_numbering[total_primal_vertices]=temp_indices_to_constraint[temp_indices[i]+min_loc]; 1610984c4197SStefano Zampini aux_primal_minloc[total_primal_vertices]=min_loc; 1611984c4197SStefano Zampini total_primal_vertices++; 1612984c4197SStefano Zampini } 1613984c4197SStefano Zampini } 1614984c4197SStefano Zampini /* free workspace */ 1615984c4197SStefano Zampini ierr = PetscFree(global_indices);CHKERRQ(ierr); 1616984c4197SStefano Zampini ierr = PetscFree(touched);CHKERRQ(ierr); 1617674ae819SStefano Zampini /* permute indices in order to have a sorted set of vertices */ 1618984c4197SStefano Zampini ierr = PetscSortInt(total_primal_vertices,aux_primal_numbering); 1619984c4197SStefano Zampini 1620984c4197SStefano Zampini /* nonzero structure of constraint matrix */ 1621984c4197SStefano Zampini ierr = PetscMalloc(pcbddc->local_primal_size*sizeof(PetscInt),&nnz);CHKERRQ(ierr); 1622984c4197SStefano Zampini for (i=0;i<total_primal_vertices;i++) nnz[i]=1; 1623984c4197SStefano Zampini j=total_primal_vertices; 1624984c4197SStefano Zampini for (i=pcbddc->n_vertices;i<pcbddc->local_primal_size;i++) { 1625674ae819SStefano Zampini if (!change_basis[i]) { 1626674ae819SStefano Zampini nnz[j]=temp_indices[i+1]-temp_indices[i]; 1627674ae819SStefano Zampini j++; 1628674ae819SStefano Zampini } 1629674ae819SStefano Zampini } 1630674ae819SStefano Zampini ierr = MatSeqAIJSetPreallocation(pcbddc->ConstraintMatrix,0,nnz);CHKERRQ(ierr); 1631674ae819SStefano Zampini ierr = PetscFree(nnz);CHKERRQ(ierr); 1632674ae819SStefano Zampini /* set values in constraint matrix */ 1633984c4197SStefano Zampini for (i=0;i<total_primal_vertices;i++) { 1634984c4197SStefano Zampini ierr = MatSetValue(pcbddc->ConstraintMatrix,i,aux_primal_numbering[i],1.0,INSERT_VALUES);CHKERRQ(ierr); 1635674ae819SStefano Zampini } 1636984c4197SStefano Zampini total_counts = total_primal_vertices; 1637984c4197SStefano Zampini for (i=pcbddc->n_vertices;i<pcbddc->local_primal_size;i++) { 1638674ae819SStefano Zampini if (!change_basis[i]) { 1639674ae819SStefano Zampini size_of_constraint=temp_indices[i+1]-temp_indices[i]; 1640674ae819SStefano Zampini ierr = MatSetValues(pcbddc->ConstraintMatrix,1,&total_counts,size_of_constraint,&temp_indices_to_constraint[temp_indices[i]],&temp_quadrature_constraint[temp_indices[i]],INSERT_VALUES);CHKERRQ(ierr); 1641674ae819SStefano Zampini total_counts++; 1642674ae819SStefano Zampini } 1643674ae819SStefano Zampini } 1644674ae819SStefano Zampini /* assembling */ 1645674ae819SStefano Zampini ierr = MatAssemblyBegin(pcbddc->ConstraintMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1646674ae819SStefano Zampini ierr = MatAssemblyEnd(pcbddc->ConstraintMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1647984c4197SStefano Zampini /* 1648984c4197SStefano Zampini ierr = MatView(pcbddc->ConstraintMatrix,(PetscViewer)0);CHKERRQ(ierr); 1649984c4197SStefano Zampini */ 1650674ae819SStefano Zampini /* Create matrix for change of basis. We don't need it in case pcbddc->use_change_of_basis is FALSE */ 1651674ae819SStefano Zampini if (pcbddc->use_change_of_basis) { 1652984c4197SStefano Zampini PetscBool qr_needed = PETSC_FALSE; 1653984c4197SStefano Zampini /* change of basis acts on local interfaces -> dimension is n_B x n_B */ 1654674ae819SStefano Zampini ierr = MatCreate(PETSC_COMM_SELF,&pcbddc->ChangeOfBasisMatrix);CHKERRQ(ierr); 1655674ae819SStefano Zampini ierr = MatSetType(pcbddc->ChangeOfBasisMatrix,impMatType);CHKERRQ(ierr); 1656674ae819SStefano Zampini ierr = MatSetSizes(pcbddc->ChangeOfBasisMatrix,pcis->n_B,pcis->n_B,pcis->n_B,pcis->n_B);CHKERRQ(ierr); 1657674ae819SStefano Zampini /* work arrays */ 1658674ae819SStefano Zampini ierr = PetscMalloc(pcis->n_B*sizeof(PetscInt),&nnz);CHKERRQ(ierr); 1659984c4197SStefano Zampini for (i=0;i<pcis->n_B;i++) nnz[i]=1; 1660984c4197SStefano Zampini /* nonzeros per row */ 1661984c4197SStefano Zampini for (i=pcbddc->n_vertices;i<pcbddc->local_primal_size;i++) { 1662674ae819SStefano Zampini if (change_basis[i]) { 1663984c4197SStefano Zampini qr_needed = PETSC_TRUE; 1664674ae819SStefano Zampini size_of_constraint = temp_indices[i+1]-temp_indices[i]; 1665984c4197SStefano Zampini for (j=0;j<size_of_constraint;j++) nnz[temp_indices_to_constraint_B[temp_indices[i]+j]] = size_of_constraint; 1666674ae819SStefano Zampini } 1667674ae819SStefano Zampini } 1668674ae819SStefano Zampini ierr = MatSeqAIJSetPreallocation(pcbddc->ChangeOfBasisMatrix,0,nnz);CHKERRQ(ierr); 1669674ae819SStefano Zampini ierr = PetscFree(nnz);CHKERRQ(ierr); 1670674ae819SStefano Zampini /* Set initial identity in the matrix */ 1671674ae819SStefano Zampini for (i=0;i<pcis->n_B;i++) { 1672674ae819SStefano Zampini ierr = MatSetValue(pcbddc->ChangeOfBasisMatrix,i,i,1.0,INSERT_VALUES);CHKERRQ(ierr); 1673674ae819SStefano Zampini } 1674984c4197SStefano Zampini 1675674ae819SStefano Zampini /* Now we loop on the constraints which need a change of basis */ 1676674ae819SStefano Zampini /* Change of basis matrix is evaluated as the FIRST APPROACH in */ 1677984c4197SStefano Zampini /* Klawonn and Widlund, Dual-primal FETI-DP methods for linear elasticity, (see Sect 6.2.1) */ 1678984c4197SStefano Zampini /* Change of basis matrix T computed via QR decomposition of constraints */ 1679984c4197SStefano Zampini if (qr_needed) { 1680026de310SStefano Zampini /* dual and primal dofs on a single cc */ 1681984c4197SStefano Zampini PetscInt dual_dofs,primal_dofs; 1682026de310SStefano Zampini /* iterator on aux_primal_minloc (ordered as read from nearnullspace: vertices, edges and then constraints) */ 1683026de310SStefano Zampini PetscInt primal_counter; 1684984c4197SStefano Zampini /* working stuff for GEQRF */ 1685984c4197SStefano Zampini PetscScalar *qr_basis,*qr_tau,*qr_work,lqr_work_t; 1686984c4197SStefano Zampini PetscBLASInt lqr_work; 1687984c4197SStefano Zampini /* working stuff for UNGQR */ 1688984c4197SStefano Zampini PetscScalar *gqr_work,lgqr_work_t; 1689984c4197SStefano Zampini PetscBLASInt lgqr_work; 1690984c4197SStefano Zampini /* working stuff for TRTRS */ 1691984c4197SStefano Zampini PetscScalar *trs_rhs; 16923f08241aSStefano Zampini PetscBLASInt Blas_NRHS; 1693984c4197SStefano Zampini /* pointers for values insertion into change of basis matrix */ 1694984c4197SStefano Zampini PetscInt *start_rows,*start_cols; 1695984c4197SStefano Zampini PetscScalar *start_vals; 1696984c4197SStefano Zampini /* working stuff for values insertion */ 1697984c4197SStefano Zampini PetscBool *is_primal; 1698984c4197SStefano Zampini 1699984c4197SStefano Zampini /* space to store Q */ 1700984c4197SStefano Zampini ierr = PetscMalloc((max_size_of_constraint)*(max_size_of_constraint)*sizeof(PetscScalar),&qr_basis);CHKERRQ(ierr); 1701984c4197SStefano Zampini /* first we issue queries for optimal work */ 17023f08241aSStefano Zampini ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_M);CHKERRQ(ierr); 17033f08241aSStefano Zampini ierr = PetscBLASIntCast(max_constraints,&Blas_N);CHKERRQ(ierr); 17043f08241aSStefano Zampini ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 1705984c4197SStefano Zampini lqr_work = -1; 17063f08241aSStefano Zampini PetscStackCallBLAS("LAPACKgeqrf",LAPACKgeqrf_(&Blas_M,&Blas_N,qr_basis,&Blas_LDA,qr_tau,&lqr_work_t,&lqr_work,&lierr)); 1707984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to GEQRF Lapack routine %d",(int)lierr); 1708984c4197SStefano Zampini ierr = PetscBLASIntCast((PetscInt)PetscRealPart(lqr_work_t),&lqr_work);CHKERRQ(ierr); 1709984c4197SStefano Zampini ierr = PetscMalloc((PetscInt)PetscRealPart(lqr_work_t)*sizeof(*qr_work),&qr_work);CHKERRQ(ierr); 1710984c4197SStefano Zampini lgqr_work = -1; 17113f08241aSStefano Zampini ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_M);CHKERRQ(ierr); 17123f08241aSStefano Zampini ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_N);CHKERRQ(ierr); 17133f08241aSStefano Zampini ierr = PetscBLASIntCast(max_constraints,&Blas_K);CHKERRQ(ierr); 17143f08241aSStefano Zampini ierr = PetscBLASIntCast(max_size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 17153f08241aSStefano Zampini if (Blas_K>Blas_M) Blas_K=Blas_M; /* adjust just for computing optimal work */ 17163f08241aSStefano Zampini PetscStackCallBLAS("LAPACKungqr",LAPACKungqr_(&Blas_M,&Blas_N,&Blas_K,qr_basis,&Blas_LDA,qr_tau,&lgqr_work_t,&lgqr_work,&lierr)); 1717984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in query to UNGQR Lapack routine %d",(int)lierr); 1718984c4197SStefano Zampini ierr = PetscBLASIntCast((PetscInt)PetscRealPart(lgqr_work_t),&lgqr_work);CHKERRQ(ierr); 1719984c4197SStefano Zampini ierr = PetscMalloc((PetscInt)PetscRealPart(lgqr_work_t)*sizeof(*gqr_work),&gqr_work);CHKERRQ(ierr); 1720984c4197SStefano Zampini /* array to store scaling factors for reflectors */ 1721984c4197SStefano Zampini ierr = PetscMalloc(max_constraints*sizeof(*qr_tau),&qr_tau);CHKERRQ(ierr); 1722984c4197SStefano Zampini /* array to store rhs and solution of triangular solver */ 1723984c4197SStefano Zampini ierr = PetscMalloc(max_constraints*max_constraints*sizeof(*trs_rhs),&trs_rhs);CHKERRQ(ierr); 1724984c4197SStefano Zampini /* array to store whether a node is primal or not */ 1725984c4197SStefano Zampini ierr = PetscMalloc(pcis->n_B*sizeof(*is_primal),&is_primal);CHKERRQ(ierr); 1726984c4197SStefano Zampini ierr = PetscMemzero(is_primal,pcis->n_B*sizeof(*is_primal));CHKERRQ(ierr); 1727984c4197SStefano Zampini for (i=0;i<total_primal_vertices;i++) is_primal[local_to_B[aux_primal_numbering[i]]] = PETSC_TRUE; 1728984c4197SStefano Zampini 1729984c4197SStefano Zampini /* allocating workspace for check */ 1730984c4197SStefano Zampini if (pcbddc->dbg_flag) { 1731984c4197SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"--------------------------------------------------------------\n");CHKERRQ(ierr); 1732984c4197SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Checking change of basis computation for subdomain %04d\n",PetscGlobalRank);CHKERRQ(ierr); 1733984c4197SStefano Zampini ierr = PetscMalloc(max_size_of_constraint*(max_constraints+max_size_of_constraint)*sizeof(*work),&work);CHKERRQ(ierr); 1734674ae819SStefano Zampini } 1735984c4197SStefano Zampini 1736026de310SStefano Zampini /* loop on constraints and see whether or not they need a change of basis */ 1737026de310SStefano Zampini /* -> using implicit ordering contained in temp_indices data */ 1738026de310SStefano Zampini total_counts = pcbddc->n_vertices; 1739026de310SStefano Zampini primal_counter = total_counts; 1740026de310SStefano Zampini while (total_counts<pcbddc->local_primal_size) { 1741026de310SStefano Zampini primal_dofs = 1; 1742026de310SStefano Zampini if (change_basis[total_counts]) { 1743026de310SStefano Zampini /* get all constraints with same support: if more then one constraint is present on the cc then surely indices are stored contiguosly */ 1744026de310SStefano Zampini while (total_counts+primal_dofs < pcbddc->local_primal_size && temp_indices_to_constraint_B[temp_indices[total_counts]] == temp_indices_to_constraint_B[temp_indices[total_counts+primal_dofs]]) { 1745026de310SStefano Zampini primal_dofs++; 1746674ae819SStefano Zampini } 1747984c4197SStefano Zampini /* get constraint info */ 1748026de310SStefano Zampini size_of_constraint = temp_indices[total_counts+1]-temp_indices[total_counts]; 1749984c4197SStefano Zampini dual_dofs = size_of_constraint-primal_dofs; 1750984c4197SStefano Zampini 1751984c4197SStefano Zampini /* copy quadrature constraints for change of basis check */ 1752984c4197SStefano Zampini if (pcbddc->dbg_flag) { 1753026de310SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Constraint %d to %d need a change of basis (size %d)\n",total_counts,total_counts+primal_dofs,size_of_constraint);CHKERRQ(ierr); 1754026de310SStefano Zampini ierr = PetscMemcpy(work,&temp_quadrature_constraint[temp_indices[total_counts]],size_of_constraint*primal_dofs*sizeof(PetscScalar));CHKERRQ(ierr); 1755674ae819SStefano Zampini } 1756984c4197SStefano Zampini 1757984c4197SStefano Zampini /* copy temporary constraints into larger work vector (in order to store all columns of Q) */ 1758026de310SStefano Zampini ierr = PetscMemcpy(qr_basis,&temp_quadrature_constraint[temp_indices[total_counts]],size_of_constraint*primal_dofs*sizeof(PetscScalar));CHKERRQ(ierr); 1759984c4197SStefano Zampini 1760984c4197SStefano Zampini /* compute QR decomposition of constraints */ 17613f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr); 17623f08241aSStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_N);CHKERRQ(ierr); 17633f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 1764674ae819SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 17653f08241aSStefano Zampini PetscStackCallBLAS("LAPACKgeqrf",LAPACKgeqrf_(&Blas_M,&Blas_N,qr_basis,&Blas_LDA,qr_tau,qr_work,&lqr_work,&lierr)); 1766984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in GEQRF Lapack routine %d",(int)lierr); 1767674ae819SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1768984c4197SStefano Zampini 1769984c4197SStefano Zampini /* explictly compute R^-T */ 1770984c4197SStefano Zampini ierr = PetscMemzero(trs_rhs,primal_dofs*primal_dofs*sizeof(*trs_rhs));CHKERRQ(ierr); 1771984c4197SStefano Zampini for (j=0;j<primal_dofs;j++) trs_rhs[j*(primal_dofs+1)] = 1.0; 17723f08241aSStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_N);CHKERRQ(ierr); 17733f08241aSStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_NRHS);CHKERRQ(ierr); 17743f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 17753f08241aSStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_LDB);CHKERRQ(ierr); 1776984c4197SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 17773f08241aSStefano Zampini PetscStackCallBLAS("LAPACKtrtrs",LAPACKtrtrs_("U","T","N",&Blas_N,&Blas_NRHS,qr_basis,&Blas_LDA,trs_rhs,&Blas_LDB,&lierr)); 1778984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in TRTRS Lapack routine %d",(int)lierr); 1779984c4197SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1780984c4197SStefano Zampini 1781984c4197SStefano Zampini /* explcitly compute all columns of Q (Q = [Q1 | Q2] ) overwriting QR factorization in qr_basis */ 17823f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr); 17833f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr); 17843f08241aSStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_K);CHKERRQ(ierr); 17853f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 1786984c4197SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 17873f08241aSStefano Zampini PetscStackCallBLAS("LAPACKungqr",LAPACKungqr_(&Blas_M,&Blas_N,&Blas_K,qr_basis,&Blas_LDA,qr_tau,gqr_work,&lgqr_work,&lierr)); 1788984c4197SStefano Zampini if (lierr) SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_LIB,"Error in UNGQR Lapack routine %d",(int)lierr); 1789984c4197SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1790984c4197SStefano Zampini 1791984c4197SStefano Zampini /* first primal_dofs columns of Q need to be re-scaled in order to be unitary w.r.t constraints 1792984c4197SStefano Zampini i.e. C_{pxn}*Q_{nxn} should be equal to [I_pxp | 0_pxd] (see check below) 1793984c4197SStefano Zampini where n=size_of_constraint, p=primal_dofs, d=dual_dofs (n=p+d), I and 0 identity and null matrix resp. */ 17943f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_M);CHKERRQ(ierr); 17953f08241aSStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_N);CHKERRQ(ierr); 17963f08241aSStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_K);CHKERRQ(ierr); 17973f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 17983f08241aSStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_LDB);CHKERRQ(ierr); 17993f08241aSStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDC);CHKERRQ(ierr); 1800984c4197SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1801c4303822SStefano Zampini PetscStackCallBLAS("BLASgemm",BLASgemm_("N","N",&Blas_M,&Blas_N,&Blas_K,&one,qr_basis,&Blas_LDA,trs_rhs,&Blas_LDB,&zero,&temp_quadrature_constraint[temp_indices[total_counts]],&Blas_LDC)); 1802984c4197SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1803026de310SStefano Zampini ierr = PetscMemcpy(qr_basis,&temp_quadrature_constraint[temp_indices[total_counts]],size_of_constraint*primal_dofs*sizeof(PetscScalar));CHKERRQ(ierr); 1804984c4197SStefano Zampini 1805984c4197SStefano Zampini /* insert values in change of basis matrix respecting global ordering of new primal dofs */ 1806026de310SStefano Zampini start_rows = &temp_indices_to_constraint_B[temp_indices[total_counts]]; 1807984c4197SStefano Zampini /* insert cols for primal dofs */ 1808984c4197SStefano Zampini for (j=0;j<primal_dofs;j++) { 1809984c4197SStefano Zampini start_vals = &qr_basis[j*size_of_constraint]; 1810026de310SStefano Zampini start_cols = &temp_indices_to_constraint_B[temp_indices[total_counts]+aux_primal_minloc[primal_counter+j]]; 1811984c4197SStefano Zampini ierr = MatSetValues(pcbddc->ChangeOfBasisMatrix,size_of_constraint,start_rows,1,start_cols,start_vals,INSERT_VALUES);CHKERRQ(ierr); 1812984c4197SStefano Zampini } 1813984c4197SStefano Zampini /* insert cols for dual dofs */ 1814984c4197SStefano Zampini for (j=0,k=0;j<dual_dofs;k++) { 1815026de310SStefano Zampini if (!is_primal[temp_indices_to_constraint_B[temp_indices[total_counts]+k]]) { 1816984c4197SStefano Zampini start_vals = &qr_basis[(primal_dofs+j)*size_of_constraint]; 1817026de310SStefano Zampini start_cols = &temp_indices_to_constraint_B[temp_indices[total_counts]+k]; 1818984c4197SStefano Zampini ierr = MatSetValues(pcbddc->ChangeOfBasisMatrix,size_of_constraint,start_rows,1,start_cols,start_vals,INSERT_VALUES);CHKERRQ(ierr); 1819984c4197SStefano Zampini j++; 1820674ae819SStefano Zampini } 1821674ae819SStefano Zampini } 1822984c4197SStefano Zampini 1823984c4197SStefano Zampini /* check change of basis */ 1824984c4197SStefano Zampini if (pcbddc->dbg_flag) { 1825984c4197SStefano Zampini PetscInt ii,jj; 1826984c4197SStefano Zampini PetscBool valid_qr=PETSC_TRUE; 1827c4303822SStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_M);CHKERRQ(ierr); 1828c4303822SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_N);CHKERRQ(ierr); 1829c4303822SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_K);CHKERRQ(ierr); 1830c4303822SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDA);CHKERRQ(ierr); 1831c4303822SStefano Zampini ierr = PetscBLASIntCast(size_of_constraint,&Blas_LDB);CHKERRQ(ierr); 1832c4303822SStefano Zampini ierr = PetscBLASIntCast(primal_dofs,&Blas_LDC);CHKERRQ(ierr); 1833984c4197SStefano Zampini ierr = PetscFPTrapPush(PETSC_FP_TRAP_OFF);CHKERRQ(ierr); 1834c4303822SStefano Zampini PetscStackCallBLAS("BLASgemm",BLASgemm_("T","N",&Blas_M,&Blas_N,&Blas_K,&one,work,&Blas_LDA,qr_basis,&Blas_LDB,&zero,&work[size_of_constraint*primal_dofs],&Blas_LDC)); 1835984c4197SStefano Zampini ierr = PetscFPTrapPop();CHKERRQ(ierr); 1836984c4197SStefano Zampini for (jj=0;jj<size_of_constraint;jj++) { 1837984c4197SStefano Zampini for (ii=0;ii<primal_dofs;ii++) { 1838984c4197SStefano Zampini if (ii != jj && PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]) > 1.e-12) valid_qr = PETSC_FALSE; 1839984c4197SStefano Zampini if (ii == jj && PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]-1.0) > 1.e-12) valid_qr = PETSC_FALSE; 1840674ae819SStefano Zampini } 1841674ae819SStefano Zampini } 1842984c4197SStefano Zampini if (!valid_qr) { 1843984c4197SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\t-> wrong change of basis!\n",PetscGlobalRank);CHKERRQ(ierr); 1844984c4197SStefano Zampini for (jj=0;jj<size_of_constraint;jj++) { 1845984c4197SStefano Zampini for (ii=0;ii<primal_dofs;ii++) { 1846984c4197SStefano Zampini if (ii != jj && PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]) > 1.e-12) { 1847984c4197SStefano Zampini PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\tQr basis function %d is not orthogonal to constraint %d (%1.14e)!\n",jj,ii,PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii])); 1848674ae819SStefano Zampini } 1849984c4197SStefano Zampini if (ii == jj && PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii]-1.0) > 1.e-12) { 1850984c4197SStefano Zampini PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\tQr basis function %d is not unitary w.r.t constraint %d (%1.14e)!\n",jj,ii,PetscAbsScalar(work[size_of_constraint*primal_dofs+jj*primal_dofs+ii])); 1851984c4197SStefano Zampini } 1852984c4197SStefano Zampini } 1853984c4197SStefano Zampini } 1854674ae819SStefano Zampini } else { 1855984c4197SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"\t-> right change of basis!\n",PetscGlobalRank);CHKERRQ(ierr); 1856674ae819SStefano Zampini } 1857674ae819SStefano Zampini } 1858026de310SStefano Zampini /* increment primal counter */ 1859026de310SStefano Zampini primal_counter += primal_dofs; 1860984c4197SStefano Zampini } else { 1861984c4197SStefano Zampini if (pcbddc->dbg_flag) { 1862026de310SStefano Zampini ierr = PetscViewerASCIISynchronizedPrintf(pcbddc->dbg_viewer,"Constraint %d does not need a change of basis (size %d)\n",total_counts,temp_indices[total_counts+1]-temp_indices[total_counts]);CHKERRQ(ierr); 1863674ae819SStefano Zampini } 1864674ae819SStefano Zampini } 1865026de310SStefano Zampini /* increment constraint counter total_counts */ 1866026de310SStefano Zampini total_counts += primal_dofs; 1867674ae819SStefano Zampini } 1868984c4197SStefano Zampini if (pcbddc->dbg_flag) { 1869984c4197SStefano Zampini ierr = PetscViewerFlush(pcbddc->dbg_viewer);CHKERRQ(ierr); 1870984c4197SStefano Zampini ierr = PetscFree(work);CHKERRQ(ierr); 1871984c4197SStefano Zampini } 1872984c4197SStefano Zampini /* free workspace */ 1873984c4197SStefano Zampini ierr = PetscFree(trs_rhs);CHKERRQ(ierr); 1874984c4197SStefano Zampini ierr = PetscFree(qr_tau);CHKERRQ(ierr); 1875984c4197SStefano Zampini ierr = PetscFree(qr_work);CHKERRQ(ierr); 1876984c4197SStefano Zampini ierr = PetscFree(gqr_work);CHKERRQ(ierr); 1877984c4197SStefano Zampini ierr = PetscFree(is_primal);CHKERRQ(ierr); 1878984c4197SStefano Zampini ierr = PetscFree(qr_basis);CHKERRQ(ierr); 1879674ae819SStefano Zampini } 1880674ae819SStefano Zampini /* assembling */ 1881674ae819SStefano Zampini ierr = MatAssemblyBegin(pcbddc->ChangeOfBasisMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1882674ae819SStefano Zampini ierr = MatAssemblyEnd(pcbddc->ChangeOfBasisMatrix,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 1883984c4197SStefano Zampini /* 1884984c4197SStefano Zampini ierr = MatView(pcbddc->ChangeOfBasisMatrix,(PetscViewer)0);CHKERRQ(ierr); 1885984c4197SStefano Zampini */ 1886674ae819SStefano Zampini } 1887e310c8b4SStefano Zampini /* free workspace */ 1888984c4197SStefano Zampini ierr = PetscFree(aux_primal_numbering);CHKERRQ(ierr); 1889984c4197SStefano Zampini ierr = PetscFree(aux_primal_minloc);CHKERRQ(ierr); 1890674ae819SStefano Zampini ierr = PetscFree(temp_indices);CHKERRQ(ierr); 1891674ae819SStefano Zampini ierr = PetscFree(change_basis);CHKERRQ(ierr); 1892674ae819SStefano Zampini ierr = PetscFree(temp_indices_to_constraint);CHKERRQ(ierr); 1893674ae819SStefano Zampini ierr = PetscFree(temp_indices_to_constraint_B);CHKERRQ(ierr); 1894674ae819SStefano Zampini ierr = PetscFree(local_to_B);CHKERRQ(ierr); 1895674ae819SStefano Zampini ierr = PetscFree(temp_quadrature_constraint);CHKERRQ(ierr); 1896674ae819SStefano Zampini PetscFunctionReturn(0); 1897674ae819SStefano Zampini } 1898674ae819SStefano Zampini 1899674ae819SStefano Zampini #undef __FUNCT__ 1900674ae819SStefano Zampini #define __FUNCT__ "PCBDDCAnalyzeInterface" 1901674ae819SStefano Zampini PetscErrorCode PCBDDCAnalyzeInterface(PC pc) 1902674ae819SStefano Zampini { 1903674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)pc->data; 1904674ae819SStefano Zampini PC_IS *pcis = (PC_IS*)pc->data; 1905674ae819SStefano Zampini Mat_IS *matis = (Mat_IS*)pc->pmat->data; 1906674ae819SStefano Zampini PetscInt bs,ierr,i,vertex_size; 1907674ae819SStefano Zampini PetscViewer viewer=pcbddc->dbg_viewer; 1908674ae819SStefano Zampini 1909674ae819SStefano Zampini PetscFunctionBegin; 1910674ae819SStefano Zampini /* Init local Graph struct */ 1911674ae819SStefano Zampini ierr = PCBDDCGraphInit(pcbddc->mat_graph,matis->mapping);CHKERRQ(ierr); 1912674ae819SStefano Zampini 1913575ad6abSStefano Zampini /* Check validity of the csr graph passed in by the user */ 1914575ad6abSStefano Zampini if (pcbddc->mat_graph->nvtxs_csr != pcbddc->mat_graph->nvtxs) { 1915575ad6abSStefano Zampini ierr = PCBDDCGraphResetCSR(pcbddc->mat_graph);CHKERRQ(ierr); 1916575ad6abSStefano Zampini } 1917674ae819SStefano Zampini /* Set default CSR adjacency of local dofs if not provided by the user with PCBDDCSetLocalAdjacencyGraph */ 1918674ae819SStefano Zampini if (!pcbddc->mat_graph->xadj || !pcbddc->mat_graph->adjncy) { 1919674ae819SStefano Zampini Mat mat_adj; 1920674ae819SStefano Zampini const PetscInt *xadj,*adjncy; 1921674ae819SStefano Zampini PetscBool flg_row=PETSC_TRUE; 1922674ae819SStefano Zampini 1923674ae819SStefano Zampini ierr = MatConvert(matis->A,MATMPIADJ,MAT_INITIAL_MATRIX,&mat_adj);CHKERRQ(ierr); 1924674ae819SStefano Zampini ierr = MatGetRowIJ(mat_adj,0,PETSC_TRUE,PETSC_FALSE,&i,&xadj,&adjncy,&flg_row);CHKERRQ(ierr); 1925674ae819SStefano Zampini if (!flg_row) { 1926674ae819SStefano Zampini SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error in MatGetRowIJ called in %s\n",__FUNCT__); 1927674ae819SStefano Zampini } 1928674ae819SStefano Zampini ierr = PCBDDCSetLocalAdjacencyGraph(pc,i,xadj,adjncy,PETSC_COPY_VALUES);CHKERRQ(ierr); 1929674ae819SStefano Zampini ierr = MatRestoreRowIJ(mat_adj,0,PETSC_TRUE,PETSC_FALSE,&i,&xadj,&adjncy,&flg_row);CHKERRQ(ierr); 1930674ae819SStefano Zampini if (!flg_row) { 1931674ae819SStefano Zampini SETERRQ1(PETSC_COMM_SELF,PETSC_ERR_PLIB,"Error in MatRestoreRowIJ called in %s\n",__FUNCT__); 1932674ae819SStefano Zampini } 1933674ae819SStefano Zampini ierr = MatDestroy(&mat_adj);CHKERRQ(ierr); 1934674ae819SStefano Zampini } 1935674ae819SStefano Zampini 1936674ae819SStefano Zampini /* Set default dofs' splitting if no information has been provided by the user with PCBDDCSetDofsSplitting */ 1937674ae819SStefano Zampini vertex_size = 1; 1938674ae819SStefano Zampini if (!pcbddc->n_ISForDofs) { 1939674ae819SStefano Zampini IS *custom_ISForDofs; 1940674ae819SStefano Zampini 1941674ae819SStefano Zampini ierr = MatGetBlockSize(matis->A,&bs);CHKERRQ(ierr); 1942674ae819SStefano Zampini ierr = PetscMalloc(bs*sizeof(IS),&custom_ISForDofs);CHKERRQ(ierr); 1943674ae819SStefano Zampini for (i=0;i<bs;i++) { 1944674ae819SStefano Zampini ierr = ISCreateStride(PETSC_COMM_SELF,pcis->n/bs,i,bs,&custom_ISForDofs[i]);CHKERRQ(ierr); 1945674ae819SStefano Zampini } 1946674ae819SStefano Zampini ierr = PCBDDCSetDofsSplitting(pc,bs,custom_ISForDofs);CHKERRQ(ierr); 1947674ae819SStefano Zampini /* remove my references to IS objects */ 1948674ae819SStefano Zampini for (i=0;i<bs;i++) { 1949674ae819SStefano Zampini ierr = ISDestroy(&custom_ISForDofs[i]);CHKERRQ(ierr); 1950674ae819SStefano Zampini } 1951674ae819SStefano Zampini ierr = PetscFree(custom_ISForDofs);CHKERRQ(ierr); 1952674ae819SStefano Zampini } else { /* mat block size as vertex size (used for elasticity) */ 1953674ae819SStefano Zampini ierr = MatGetBlockSize(matis->A,&vertex_size);CHKERRQ(ierr); 1954674ae819SStefano Zampini } 1955674ae819SStefano Zampini 1956674ae819SStefano Zampini /* Setup of Graph */ 1957674ae819SStefano Zampini ierr = PCBDDCGraphSetUp(pcbddc->mat_graph,vertex_size,pcbddc->NeumannBoundaries,pcbddc->DirichletBoundaries,pcbddc->n_ISForDofs,pcbddc->ISForDofs,pcbddc->user_primal_vertices); 1958674ae819SStefano Zampini 1959674ae819SStefano Zampini /* Graph's connected components analysis */ 1960674ae819SStefano Zampini ierr = PCBDDCGraphComputeConnectedComponents(pcbddc->mat_graph);CHKERRQ(ierr); 1961674ae819SStefano Zampini 1962674ae819SStefano Zampini /* print some info to stdout */ 1963674ae819SStefano Zampini if (pcbddc->dbg_flag) { 1964e49050b4SStefano Zampini ierr = PCBDDCGraphASCIIView(pcbddc->mat_graph,pcbddc->dbg_flag,viewer); 1965674ae819SStefano Zampini } 1966674ae819SStefano Zampini PetscFunctionReturn(0); 1967674ae819SStefano Zampini } 1968674ae819SStefano Zampini 1969674ae819SStefano Zampini #undef __FUNCT__ 1970674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGetPrimalVerticesLocalIdx" 1971674ae819SStefano Zampini PetscErrorCode PCBDDCGetPrimalVerticesLocalIdx(PC pc, PetscInt *n_vertices, PetscInt *vertices_idx[]) 1972674ae819SStefano Zampini { 1973674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 1974674ae819SStefano Zampini PetscInt *vertices,*row_cmat_indices,n,i,size_of_constraint,local_primal_size; 1975674ae819SStefano Zampini PetscErrorCode ierr; 1976674ae819SStefano Zampini 1977674ae819SStefano Zampini PetscFunctionBegin; 1978674ae819SStefano Zampini n = 0; 1979674ae819SStefano Zampini vertices = 0; 1980674ae819SStefano Zampini if (pcbddc->ConstraintMatrix) { 1981674ae819SStefano Zampini ierr = MatGetSize(pcbddc->ConstraintMatrix,&local_primal_size,&i);CHKERRQ(ierr); 1982b120a5c6SStefano Zampini for (i=0;i<local_primal_size;i++) { 1983b120a5c6SStefano Zampini ierr = MatGetRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,NULL,NULL);CHKERRQ(ierr); 1984b120a5c6SStefano Zampini if (size_of_constraint == 1) n++; 1985b120a5c6SStefano Zampini ierr = MatRestoreRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,NULL,NULL);CHKERRQ(ierr); 1986b120a5c6SStefano Zampini } 1987811e8ca2SStefano Zampini if (vertices_idx) { 1988b120a5c6SStefano Zampini ierr = PetscMalloc(n*sizeof(PetscInt),&vertices);CHKERRQ(ierr); 1989b120a5c6SStefano Zampini n = 0; 1990674ae819SStefano Zampini for (i=0;i<local_primal_size;i++) { 1991674ae819SStefano Zampini ierr = MatGetRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,NULL);CHKERRQ(ierr); 1992674ae819SStefano Zampini if (size_of_constraint == 1) { 1993674ae819SStefano Zampini vertices[n++]=row_cmat_indices[0]; 1994674ae819SStefano Zampini } 1995674ae819SStefano Zampini ierr = MatRestoreRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,NULL);CHKERRQ(ierr); 1996674ae819SStefano Zampini } 1997674ae819SStefano Zampini } 1998811e8ca2SStefano Zampini } 1999674ae819SStefano Zampini *n_vertices = n; 2000811e8ca2SStefano Zampini if (vertices_idx) *vertices_idx = vertices; 2001674ae819SStefano Zampini PetscFunctionReturn(0); 2002674ae819SStefano Zampini } 2003674ae819SStefano Zampini 2004674ae819SStefano Zampini #undef __FUNCT__ 2005674ae819SStefano Zampini #define __FUNCT__ "PCBDDCGetPrimalConstraintsLocalIdx" 2006674ae819SStefano Zampini PetscErrorCode PCBDDCGetPrimalConstraintsLocalIdx(PC pc, PetscInt *n_constraints, PetscInt *constraints_idx[]) 2007674ae819SStefano Zampini { 2008674ae819SStefano Zampini PC_BDDC *pcbddc = (PC_BDDC*)(pc->data); 2009674ae819SStefano Zampini PetscInt *constraints_index,*row_cmat_indices,*row_cmat_global_indices; 2010674ae819SStefano Zampini PetscInt n,i,j,size_of_constraint,local_primal_size,local_size,max_size_of_constraint,min_index,min_loc; 2011674ae819SStefano Zampini PetscBool *touched; 2012674ae819SStefano Zampini PetscErrorCode ierr; 2013674ae819SStefano Zampini 2014674ae819SStefano Zampini PetscFunctionBegin; 2015674ae819SStefano Zampini n = 0; 2016674ae819SStefano Zampini constraints_index = 0; 2017674ae819SStefano Zampini if (pcbddc->ConstraintMatrix) { 2018674ae819SStefano Zampini ierr = MatGetSize(pcbddc->ConstraintMatrix,&local_primal_size,&local_size);CHKERRQ(ierr); 2019674ae819SStefano Zampini max_size_of_constraint = 0; 2020674ae819SStefano Zampini for (i=0;i<local_primal_size;i++) { 2021674ae819SStefano Zampini ierr = MatGetRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,NULL,NULL);CHKERRQ(ierr); 2022674ae819SStefano Zampini if (size_of_constraint > 1) { 2023674ae819SStefano Zampini n++; 2024674ae819SStefano Zampini } 2025674ae819SStefano Zampini max_size_of_constraint = PetscMax(size_of_constraint,max_size_of_constraint); 2026674ae819SStefano Zampini ierr = MatRestoreRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,NULL,NULL);CHKERRQ(ierr); 2027674ae819SStefano Zampini } 2028811e8ca2SStefano Zampini if (constraints_idx) { 2029674ae819SStefano Zampini ierr = PetscMalloc(n*sizeof(PetscInt),&constraints_index);CHKERRQ(ierr); 2030674ae819SStefano Zampini ierr = PetscMalloc(max_size_of_constraint*sizeof(PetscInt),&row_cmat_global_indices);CHKERRQ(ierr); 2031674ae819SStefano Zampini ierr = PetscMalloc(local_size*sizeof(PetscBool),&touched);CHKERRQ(ierr); 2032674ae819SStefano Zampini ierr = PetscMemzero(touched,local_size*sizeof(PetscBool));CHKERRQ(ierr); 2033674ae819SStefano Zampini n = 0; 2034674ae819SStefano Zampini for (i=0;i<local_primal_size;i++) { 2035674ae819SStefano Zampini ierr = MatGetRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,NULL);CHKERRQ(ierr); 2036674ae819SStefano Zampini if (size_of_constraint > 1) { 2037674ae819SStefano Zampini ierr = ISLocalToGlobalMappingApply(pcbddc->mat_graph->l2gmap,size_of_constraint,row_cmat_indices,row_cmat_global_indices);CHKERRQ(ierr); 203882d3d8afSStefano Zampini /* find first untouched local node */ 203982d3d8afSStefano Zampini j = 0; 204082d3d8afSStefano Zampini while(touched[row_cmat_indices[j]]) j++; 204182d3d8afSStefano Zampini min_index = row_cmat_global_indices[j]; 204282d3d8afSStefano Zampini min_loc = j; 204382d3d8afSStefano Zampini /* search the minimum among nodes not yet touched on the connected component 204482d3d8afSStefano Zampini since there can be more than one constraint on a single cc */ 2045674ae819SStefano Zampini for (j=1;j<size_of_constraint;j++) { 2046674ae819SStefano Zampini if (min_index > row_cmat_global_indices[j] && !touched[row_cmat_indices[j]]) { 2047674ae819SStefano Zampini min_index = row_cmat_global_indices[j]; 2048674ae819SStefano Zampini min_loc = j; 2049674ae819SStefano Zampini } 2050674ae819SStefano Zampini } 2051674ae819SStefano Zampini touched[row_cmat_indices[min_loc]] = PETSC_TRUE; 2052674ae819SStefano Zampini constraints_index[n++] = row_cmat_indices[min_loc]; 2053674ae819SStefano Zampini } 2054674ae819SStefano Zampini ierr = MatRestoreRow(pcbddc->ConstraintMatrix,i,&size_of_constraint,(const PetscInt**)&row_cmat_indices,NULL);CHKERRQ(ierr); 2055674ae819SStefano Zampini } 2056674ae819SStefano Zampini ierr = PetscFree(touched);CHKERRQ(ierr); 2057674ae819SStefano Zampini ierr = PetscFree(row_cmat_global_indices);CHKERRQ(ierr); 2058811e8ca2SStefano Zampini } 2059811e8ca2SStefano Zampini } 2060674ae819SStefano Zampini *n_constraints = n; 2061811e8ca2SStefano Zampini if (constraints_idx) *constraints_idx = constraints_index; 2062674ae819SStefano Zampini PetscFunctionReturn(0); 2063674ae819SStefano Zampini } 2064674ae819SStefano Zampini 2065674ae819SStefano Zampini /* the next two functions has been adapted from pcis.c */ 2066674ae819SStefano Zampini #undef __FUNCT__ 2067674ae819SStefano Zampini #define __FUNCT__ "PCBDDCApplySchur" 2068674ae819SStefano Zampini PetscErrorCode PCBDDCApplySchur(PC pc, Vec v, Vec vec1_B, Vec vec2_B, Vec vec1_D, Vec vec2_D) 2069674ae819SStefano Zampini { 2070674ae819SStefano Zampini PetscErrorCode ierr; 2071674ae819SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 2072674ae819SStefano Zampini 2073674ae819SStefano Zampini PetscFunctionBegin; 2074674ae819SStefano Zampini if (!vec2_B) { vec2_B = v; } 2075674ae819SStefano Zampini ierr = MatMult(pcis->A_BB,v,vec1_B);CHKERRQ(ierr); 2076674ae819SStefano Zampini ierr = MatMult(pcis->A_IB,v,vec1_D);CHKERRQ(ierr); 2077674ae819SStefano Zampini ierr = KSPSolve(pcis->ksp_D,vec1_D,vec2_D);CHKERRQ(ierr); 2078674ae819SStefano Zampini ierr = MatMult(pcis->A_BI,vec2_D,vec2_B);CHKERRQ(ierr); 2079674ae819SStefano Zampini ierr = VecAXPY(vec1_B,-1.0,vec2_B);CHKERRQ(ierr); 2080674ae819SStefano Zampini PetscFunctionReturn(0); 2081674ae819SStefano Zampini } 2082674ae819SStefano Zampini 2083674ae819SStefano Zampini #undef __FUNCT__ 2084674ae819SStefano Zampini #define __FUNCT__ "PCBDDCApplySchurTranspose" 2085674ae819SStefano Zampini PetscErrorCode PCBDDCApplySchurTranspose(PC pc, Vec v, Vec vec1_B, Vec vec2_B, Vec vec1_D, Vec vec2_D) 2086674ae819SStefano Zampini { 2087674ae819SStefano Zampini PetscErrorCode ierr; 2088674ae819SStefano Zampini PC_IS *pcis = (PC_IS*)(pc->data); 2089674ae819SStefano Zampini 2090674ae819SStefano Zampini PetscFunctionBegin; 2091674ae819SStefano Zampini if (!vec2_B) { vec2_B = v; } 2092674ae819SStefano Zampini ierr = MatMultTranspose(pcis->A_BB,v,vec1_B);CHKERRQ(ierr); 2093674ae819SStefano Zampini ierr = MatMultTranspose(pcis->A_BI,v,vec1_D);CHKERRQ(ierr); 2094674ae819SStefano Zampini ierr = KSPSolveTranspose(pcis->ksp_D,vec1_D,vec2_D);CHKERRQ(ierr); 2095674ae819SStefano Zampini ierr = MatMultTranspose(pcis->A_IB,vec2_D,vec2_B);CHKERRQ(ierr); 2096674ae819SStefano Zampini ierr = VecAXPY(vec1_B,-1.0,vec2_B);CHKERRQ(ierr); 2097674ae819SStefano Zampini PetscFunctionReturn(0); 2098674ae819SStefano Zampini } 2099674ae819SStefano Zampini 2100674ae819SStefano Zampini #undef __FUNCT__ 2101674ae819SStefano Zampini #define __FUNCT__ "PCBDDCSubsetNumbering" 2102674ae819SStefano Zampini PetscErrorCode PCBDDCSubsetNumbering(MPI_Comm comm,ISLocalToGlobalMapping l2gmap, PetscInt n_local_dofs, PetscInt local_dofs[], PetscInt local_dofs_mult[], PetscInt* n_global_subset, PetscInt* global_numbering_subset[]) 2103674ae819SStefano Zampini { 2104674ae819SStefano Zampini Vec local_vec,global_vec; 2105674ae819SStefano Zampini IS seqis,paris; 2106674ae819SStefano Zampini VecScatter scatter_ctx; 2107674ae819SStefano Zampini PetscScalar *array; 2108674ae819SStefano Zampini PetscInt *temp_global_dofs; 2109674ae819SStefano Zampini PetscScalar globalsum; 2110674ae819SStefano Zampini PetscInt i,j,s; 2111674ae819SStefano Zampini PetscInt nlocals,first_index,old_index,max_local; 2112674ae819SStefano Zampini PetscMPIInt rank_prec_comm,size_prec_comm,max_global; 2113674ae819SStefano Zampini PetscMPIInt *dof_sizes,*dof_displs; 2114674ae819SStefano Zampini PetscBool first_found; 2115674ae819SStefano Zampini PetscErrorCode ierr; 2116674ae819SStefano Zampini 2117674ae819SStefano Zampini PetscFunctionBegin; 2118674ae819SStefano Zampini /* mpi buffers */ 2119674ae819SStefano Zampini MPI_Comm_size(comm,&size_prec_comm); 2120674ae819SStefano Zampini MPI_Comm_rank(comm,&rank_prec_comm); 2121674ae819SStefano Zampini j = ( !rank_prec_comm ? size_prec_comm : 0); 2122674ae819SStefano Zampini ierr = PetscMalloc(j*sizeof(*dof_sizes),&dof_sizes);CHKERRQ(ierr); 2123674ae819SStefano Zampini ierr = PetscMalloc(j*sizeof(*dof_displs),&dof_displs);CHKERRQ(ierr); 2124674ae819SStefano Zampini /* get maximum size of subset */ 2125674ae819SStefano Zampini ierr = PetscMalloc(n_local_dofs*sizeof(PetscInt),&temp_global_dofs);CHKERRQ(ierr); 2126674ae819SStefano Zampini ierr = ISLocalToGlobalMappingApply(l2gmap,n_local_dofs,local_dofs,temp_global_dofs);CHKERRQ(ierr); 2127674ae819SStefano Zampini max_local = 0; 2128674ae819SStefano Zampini if (n_local_dofs) { 2129674ae819SStefano Zampini max_local = temp_global_dofs[0]; 2130674ae819SStefano Zampini for (i=1;i<n_local_dofs;i++) { 2131674ae819SStefano Zampini if (max_local < temp_global_dofs[i] ) { 2132674ae819SStefano Zampini max_local = temp_global_dofs[i]; 2133674ae819SStefano Zampini } 2134674ae819SStefano Zampini } 2135674ae819SStefano Zampini } 2136674ae819SStefano Zampini ierr = MPI_Allreduce(&max_local,&max_global,1,MPIU_INT,MPI_MAX,comm); 2137674ae819SStefano Zampini max_global++; 2138674ae819SStefano Zampini max_local = 0; 2139674ae819SStefano Zampini if (n_local_dofs) { 2140674ae819SStefano Zampini max_local = local_dofs[0]; 2141674ae819SStefano Zampini for (i=1;i<n_local_dofs;i++) { 2142674ae819SStefano Zampini if (max_local < local_dofs[i] ) { 2143674ae819SStefano Zampini max_local = local_dofs[i]; 2144674ae819SStefano Zampini } 2145674ae819SStefano Zampini } 2146674ae819SStefano Zampini } 2147674ae819SStefano Zampini max_local++; 2148674ae819SStefano Zampini /* allocate workspace */ 2149674ae819SStefano Zampini ierr = VecCreate(PETSC_COMM_SELF,&local_vec);CHKERRQ(ierr); 2150674ae819SStefano Zampini ierr = VecSetSizes(local_vec,PETSC_DECIDE,max_local);CHKERRQ(ierr); 2151674ae819SStefano Zampini ierr = VecSetType(local_vec,VECSEQ);CHKERRQ(ierr); 2152674ae819SStefano Zampini ierr = VecCreate(comm,&global_vec);CHKERRQ(ierr); 2153674ae819SStefano Zampini ierr = VecSetSizes(global_vec,PETSC_DECIDE,max_global);CHKERRQ(ierr); 2154674ae819SStefano Zampini ierr = VecSetType(global_vec,VECMPI);CHKERRQ(ierr); 2155674ae819SStefano Zampini /* create scatter */ 2156674ae819SStefano Zampini ierr = ISCreateGeneral(PETSC_COMM_SELF,n_local_dofs,local_dofs,PETSC_COPY_VALUES,&seqis);CHKERRQ(ierr); 2157674ae819SStefano Zampini ierr = ISCreateGeneral(comm,n_local_dofs,temp_global_dofs,PETSC_COPY_VALUES,&paris);CHKERRQ(ierr); 2158674ae819SStefano Zampini ierr = VecScatterCreate(local_vec,seqis,global_vec,paris,&scatter_ctx);CHKERRQ(ierr); 2159674ae819SStefano Zampini ierr = ISDestroy(&seqis);CHKERRQ(ierr); 2160674ae819SStefano Zampini ierr = ISDestroy(&paris);CHKERRQ(ierr); 2161674ae819SStefano Zampini /* init array */ 2162674ae819SStefano Zampini ierr = VecSet(global_vec,0.0);CHKERRQ(ierr); 2163674ae819SStefano Zampini ierr = VecSet(local_vec,0.0);CHKERRQ(ierr); 2164674ae819SStefano Zampini ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr); 2165674ae819SStefano Zampini if (local_dofs_mult) { 2166674ae819SStefano Zampini for (i=0;i<n_local_dofs;i++) { 2167674ae819SStefano Zampini array[local_dofs[i]]=(PetscScalar)local_dofs_mult[i]; 2168674ae819SStefano Zampini } 2169674ae819SStefano Zampini } else { 2170674ae819SStefano Zampini for (i=0;i<n_local_dofs;i++) { 2171674ae819SStefano Zampini array[local_dofs[i]]=1.0; 2172674ae819SStefano Zampini } 2173674ae819SStefano Zampini } 2174674ae819SStefano Zampini ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr); 2175674ae819SStefano Zampini /* scatter into global vec and get total number of global dofs */ 2176674ae819SStefano Zampini ierr = VecScatterBegin(scatter_ctx,local_vec,global_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2177674ae819SStefano Zampini ierr = VecScatterEnd(scatter_ctx,local_vec,global_vec,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2178674ae819SStefano Zampini ierr = VecSum(global_vec,&globalsum);CHKERRQ(ierr); 21795b08dc53SStefano Zampini *n_global_subset = (PetscInt)PetscRealPart(globalsum); 2180674ae819SStefano Zampini /* Fill global_vec with cumulative function for global numbering */ 2181674ae819SStefano Zampini ierr = VecGetArray(global_vec,&array);CHKERRQ(ierr); 2182674ae819SStefano Zampini ierr = VecGetLocalSize(global_vec,&s);CHKERRQ(ierr); 2183674ae819SStefano Zampini nlocals = 0; 2184674ae819SStefano Zampini first_index = -1; 2185674ae819SStefano Zampini first_found = PETSC_FALSE; 2186674ae819SStefano Zampini for (i=0;i<s;i++) { 21875b08dc53SStefano Zampini if (!first_found && PetscRealPart(array[i]) > 0.0) { 2188674ae819SStefano Zampini first_found = PETSC_TRUE; 2189674ae819SStefano Zampini first_index = i; 2190674ae819SStefano Zampini } 21915b08dc53SStefano Zampini nlocals += (PetscInt)PetscRealPart(array[i]); 2192674ae819SStefano Zampini } 2193674ae819SStefano Zampini ierr = MPI_Gather(&nlocals,1,MPIU_INT,dof_sizes,1,MPIU_INT,0,comm);CHKERRQ(ierr); 2194674ae819SStefano Zampini if (!rank_prec_comm) { 2195674ae819SStefano Zampini dof_displs[0]=0; 2196674ae819SStefano Zampini for (i=1;i<size_prec_comm;i++) { 2197674ae819SStefano Zampini dof_displs[i] = dof_displs[i-1]+dof_sizes[i-1]; 2198674ae819SStefano Zampini } 2199674ae819SStefano Zampini } 2200674ae819SStefano Zampini ierr = MPI_Scatter(dof_displs,1,MPIU_INT,&nlocals,1,MPIU_INT,0,comm);CHKERRQ(ierr); 2201674ae819SStefano Zampini if (first_found) { 2202674ae819SStefano Zampini array[first_index] += (PetscScalar)nlocals; 2203674ae819SStefano Zampini old_index = first_index; 2204674ae819SStefano Zampini for (i=first_index+1;i<s;i++) { 22055b08dc53SStefano Zampini if (PetscRealPart(array[i]) > 0.0) { 2206674ae819SStefano Zampini array[i] += array[old_index]; 2207674ae819SStefano Zampini old_index = i; 2208674ae819SStefano Zampini } 2209674ae819SStefano Zampini } 2210674ae819SStefano Zampini } 2211674ae819SStefano Zampini ierr = VecRestoreArray(global_vec,&array);CHKERRQ(ierr); 2212674ae819SStefano Zampini ierr = VecSet(local_vec,0.0);CHKERRQ(ierr); 2213674ae819SStefano Zampini ierr = VecScatterBegin(scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2214674ae819SStefano Zampini ierr = VecScatterEnd (scatter_ctx,global_vec,local_vec,INSERT_VALUES,SCATTER_REVERSE);CHKERRQ(ierr); 2215674ae819SStefano Zampini /* get global ordering of local dofs */ 2216674ae819SStefano Zampini ierr = VecGetArray(local_vec,&array);CHKERRQ(ierr); 2217674ae819SStefano Zampini if (local_dofs_mult) { 2218674ae819SStefano Zampini for (i=0;i<n_local_dofs;i++) { 22195b08dc53SStefano Zampini temp_global_dofs[i] = (PetscInt)PetscRealPart(array[local_dofs[i]])-local_dofs_mult[i]; 2220674ae819SStefano Zampini } 2221674ae819SStefano Zampini } else { 2222674ae819SStefano Zampini for (i=0;i<n_local_dofs;i++) { 22235b08dc53SStefano Zampini temp_global_dofs[i] = (PetscInt)PetscRealPart(array[local_dofs[i]])-1; 2224674ae819SStefano Zampini } 2225674ae819SStefano Zampini } 2226674ae819SStefano Zampini ierr = VecRestoreArray(local_vec,&array);CHKERRQ(ierr); 2227674ae819SStefano Zampini /* free workspace */ 2228674ae819SStefano Zampini ierr = VecScatterDestroy(&scatter_ctx);CHKERRQ(ierr); 2229674ae819SStefano Zampini ierr = VecDestroy(&local_vec);CHKERRQ(ierr); 2230674ae819SStefano Zampini ierr = VecDestroy(&global_vec);CHKERRQ(ierr); 2231674ae819SStefano Zampini ierr = PetscFree(dof_sizes);CHKERRQ(ierr); 2232674ae819SStefano Zampini ierr = PetscFree(dof_displs);CHKERRQ(ierr); 2233674ae819SStefano Zampini /* return pointer to global ordering of local dofs */ 2234674ae819SStefano Zampini *global_numbering_subset = temp_global_dofs; 2235674ae819SStefano Zampini PetscFunctionReturn(0); 2236674ae819SStefano Zampini } 22379a7d3425SStefano Zampini 22389a7d3425SStefano Zampini #undef __FUNCT__ 22399a7d3425SStefano Zampini #define __FUNCT__ "PCBDDCOrthonormalizeVecs" 22409a7d3425SStefano Zampini PetscErrorCode PCBDDCOrthonormalizeVecs(PetscInt n, Vec vecs[]) 22419a7d3425SStefano Zampini { 22429a7d3425SStefano Zampini PetscInt i,j; 22439a7d3425SStefano Zampini PetscScalar *alphas; 22449a7d3425SStefano Zampini PetscErrorCode ierr; 22459a7d3425SStefano Zampini 22469a7d3425SStefano Zampini PetscFunctionBegin; 22479a7d3425SStefano Zampini /* this implements stabilized Gram-Schmidt */ 22489a7d3425SStefano Zampini ierr = PetscMalloc(n*sizeof(PetscScalar),&alphas);CHKERRQ(ierr); 22499a7d3425SStefano Zampini for (i=0;i<n;i++) { 22509a7d3425SStefano Zampini ierr = VecNormalize(vecs[i],NULL);CHKERRQ(ierr); 22519a7d3425SStefano Zampini if (i<n) { ierr = VecMDot(vecs[i],n-i-1,&vecs[i+1],&alphas[i+1]);CHKERRQ(ierr); } 22529a7d3425SStefano Zampini for (j=i+1;j<n;j++) { ierr = VecAXPY(vecs[j],PetscConj(-alphas[j]),vecs[i]);CHKERRQ(ierr); } 22539a7d3425SStefano Zampini } 22549a7d3425SStefano Zampini ierr = PetscFree(alphas);CHKERRQ(ierr); 22559a7d3425SStefano Zampini PetscFunctionReturn(0); 22569a7d3425SStefano Zampini } 22579a7d3425SStefano Zampini 22589a7d3425SStefano Zampini 2259