1aad13602SShrirang Abhyankar /* Program usage: mpiexec -n 2 ./ex1 [-help] [all TAO options] */ 2aad13602SShrirang Abhyankar 3aad13602SShrirang Abhyankar /* ---------------------------------------------------------------------- 4628da978Sresundermann min f(x) = (x0 - 2)^2 + (x1 - 2)^2 - 2*(x0 + x1) 5aad13602SShrirang Abhyankar s.t. x0^2 + x1 - 2 = 0 6aad13602SShrirang Abhyankar 0 <= x0^2 - x1 <= 1 7aad13602SShrirang Abhyankar -1 <= x0, x1 <= 2 8628da978Sresundermann --> 9628da978Sresundermann g(x) = 0 10628da978Sresundermann h(x) >= 0 11628da978Sresundermann -1 <= x0, x1 <= 2 12628da978Sresundermann where 13628da978Sresundermann g(x) = x0^2 + x1 - 2 14628da978Sresundermann h(x) = [x0^2 - x1 15628da978Sresundermann 1 -(x0^2 - x1)] 16aad13602SShrirang Abhyankar ---------------------------------------------------------------------- */ 17aad13602SShrirang Abhyankar 18aad13602SShrirang Abhyankar #include <petsctao.h> 19aad13602SShrirang Abhyankar 20aad13602SShrirang Abhyankar static char help[]= "Solves constrained optimiztion problem using pdipm.\n\ 21aad13602SShrirang Abhyankar Input parameters include:\n\ 22aad13602SShrirang Abhyankar -tao_type pdipm : sets Tao solver\n\ 238e58fa1dSresundermann -no_eq : removes the equaility constraints from the problem\n\ 24aad13602SShrirang Abhyankar -snes_fd : snes with finite difference Jacobian (needed for pdipm)\n\ 25628da978Sresundermann -snes_compare_explicit : compare user Jacobian with finite difference Jacobian \n\ 26aad13602SShrirang Abhyankar -tao_cmonitor : convergence monitor with constraint norm \n\ 27*a5b23f4aSJose E. Roman -tao_view_solution : view exact solution at each iteration\n\ 28628da978Sresundermann Note: external package MUMPS is required to run pdipm. This is designed for a maximum of 2 processors, the code will error if size > 2.\n"; 29aad13602SShrirang Abhyankar 30aad13602SShrirang Abhyankar /* 31628da978Sresundermann User-defined application context - contains data needed by the application 32aad13602SShrirang Abhyankar */ 33aad13602SShrirang Abhyankar typedef struct { 34aad13602SShrirang Abhyankar PetscInt n; /* Global length of x */ 35aad13602SShrirang Abhyankar PetscInt ne; /* Global number of equality constraints */ 36aad13602SShrirang Abhyankar PetscInt ni; /* Global number of inequality constraints */ 378e58fa1dSresundermann PetscBool noeqflag; 38aad13602SShrirang Abhyankar Vec x,xl,xu; 39aad13602SShrirang Abhyankar Vec ce,ci,bl,bu,Xseq; 40aad13602SShrirang Abhyankar Mat Ae,Ai,H; 41aad13602SShrirang Abhyankar VecScatter scat; 42aad13602SShrirang Abhyankar } AppCtx; 43aad13602SShrirang Abhyankar 44aad13602SShrirang Abhyankar /* -------- User-defined Routines --------- */ 45aad13602SShrirang Abhyankar PetscErrorCode InitializeProblem(AppCtx *); 46aad13602SShrirang Abhyankar PetscErrorCode DestroyProblem(AppCtx *); 47aad13602SShrirang Abhyankar PetscErrorCode FormFunctionGradient(Tao,Vec,PetscReal *,Vec,void *); 48aad13602SShrirang Abhyankar PetscErrorCode FormHessian(Tao,Vec,Mat,Mat, void*); 49aad13602SShrirang Abhyankar PetscErrorCode FormInequalityConstraints(Tao,Vec,Vec,void*); 50aad13602SShrirang Abhyankar PetscErrorCode FormEqualityConstraints(Tao,Vec,Vec,void*); 51aad13602SShrirang Abhyankar PetscErrorCode FormInequalityJacobian(Tao,Vec,Mat,Mat, void*); 52aad13602SShrirang Abhyankar PetscErrorCode FormEqualityJacobian(Tao,Vec,Mat,Mat, void*); 53aad13602SShrirang Abhyankar 54aad13602SShrirang Abhyankar PetscErrorCode main(int argc,char **argv) 55aad13602SShrirang Abhyankar { 56aad13602SShrirang Abhyankar PetscErrorCode ierr; /* used to check for functions returning nonzeros */ 57aad13602SShrirang Abhyankar Tao tao; 58aad13602SShrirang Abhyankar KSP ksp; 59aad13602SShrirang Abhyankar PC pc; 60aad13602SShrirang Abhyankar AppCtx user; /* application context */ 61aad13602SShrirang Abhyankar Vec x; 62aad13602SShrirang Abhyankar PetscMPIInt rank; 63661095bbSAlp Dener TaoType type; 64661095bbSAlp Dener PetscBool pdipm; 65aad13602SShrirang Abhyankar 66aad13602SShrirang Abhyankar ierr = PetscInitialize(&argc,&argv,(char*)0,help);if (ierr) return ierr; 67ffc4695bSBarry Smith ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr); 68d8185827SBarry Smith if (rank>1) SETERRQ(PETSC_COMM_SELF,PETSC_ERR_WRONG_MPI_SIZE,"More than 2 processors detected. Example written to use max of 2 processors.\n"); 69aad13602SShrirang Abhyankar 70aad13602SShrirang Abhyankar ierr = PetscPrintf(PETSC_COMM_WORLD,"---- Constrained Problem -----\n");CHKERRQ(ierr); 71aad13602SShrirang Abhyankar ierr = InitializeProblem(&user);CHKERRQ(ierr); /* sets up problem, function below */ 72aad13602SShrirang Abhyankar ierr = TaoCreate(PETSC_COMM_WORLD,&tao);CHKERRQ(ierr); 73aad13602SShrirang Abhyankar ierr = TaoSetType(tao,TAOPDIPM);CHKERRQ(ierr); 74aad13602SShrirang Abhyankar ierr = TaoSetInitialVector(tao,user.x);CHKERRQ(ierr); /* gets solution vector from problem */ 75aad13602SShrirang Abhyankar ierr = TaoSetVariableBounds(tao,user.xl,user.xu);CHKERRQ(ierr); /* sets lower upper bounds from given solution */ 76aad13602SShrirang Abhyankar ierr = TaoSetObjectiveAndGradientRoutine(tao,FormFunctionGradient,(void*)&user);CHKERRQ(ierr); 77aad13602SShrirang Abhyankar 788e58fa1dSresundermann if (!user.noeqflag) { 79aad13602SShrirang Abhyankar ierr = TaoSetEqualityConstraintsRoutine(tao,user.ce,FormEqualityConstraints,(void*)&user);CHKERRQ(ierr); 808e58fa1dSresundermann } 81aad13602SShrirang Abhyankar ierr = TaoSetInequalityConstraintsRoutine(tao,user.ci,FormInequalityConstraints,(void*)&user);CHKERRQ(ierr); 828e58fa1dSresundermann if (!user.noeqflag) { 83aad13602SShrirang Abhyankar ierr = TaoSetJacobianEqualityRoutine(tao,user.Ae,user.Ae,FormEqualityJacobian,(void*)&user);CHKERRQ(ierr); /* equality jacobian */ 848e58fa1dSresundermann } 85aad13602SShrirang Abhyankar ierr = TaoSetJacobianInequalityRoutine(tao,user.Ai,user.Ai,FormInequalityJacobian,(void*)&user);CHKERRQ(ierr); /* inequality jacobian */ 86aad13602SShrirang Abhyankar ierr = TaoSetTolerances(tao,1.e-6,1.e-6,1.e-6);CHKERRQ(ierr); 87aad13602SShrirang Abhyankar ierr = TaoSetConstraintTolerances(tao,1.e-6,1.e-6);CHKERRQ(ierr); 88aad13602SShrirang Abhyankar 89aad13602SShrirang Abhyankar ierr = TaoGetKSP(tao,&ksp);CHKERRQ(ierr); 90aad13602SShrirang Abhyankar ierr = KSPGetPC(ksp,&pc);CHKERRQ(ierr); 9112d688e0SRylee Sundermann ierr = PCSetType(pc,PCCHOLESKY);CHKERRQ(ierr); 92aad13602SShrirang Abhyankar /* 93aad13602SShrirang Abhyankar This algorithm produces matrices with zeros along the diagonal therefore we use 9412d688e0SRylee Sundermann MUMPS which provides solver for indefinite matrices 95aad13602SShrirang Abhyankar */ 9612d688e0SRylee Sundermann ierr = PCFactorSetMatSolverType(pc,MATSOLVERMUMPS);CHKERRQ(ierr); /* requires mumps to solve pdipm */ 97aad13602SShrirang Abhyankar ierr = KSPSetType(ksp,KSPPREONLY);CHKERRQ(ierr); 98aad13602SShrirang Abhyankar ierr = KSPSetFromOptions(ksp);CHKERRQ(ierr); 99aad13602SShrirang Abhyankar 100aad13602SShrirang Abhyankar ierr = TaoSetFromOptions(tao);CHKERRQ(ierr); 101661095bbSAlp Dener ierr = TaoGetType(tao,&type);CHKERRQ(ierr); 102661095bbSAlp Dener ierr = PetscObjectTypeCompare((PetscObject)tao,TAOPDIPM,&pdipm);CHKERRQ(ierr); 103661095bbSAlp Dener if (pdipm) { 104661095bbSAlp Dener ierr = TaoSetHessianRoutine(tao,user.H,user.H,FormHessian,(void*)&user);CHKERRQ(ierr); 105661095bbSAlp Dener } 106aad13602SShrirang Abhyankar 107aad13602SShrirang Abhyankar ierr = TaoSolve(tao);CHKERRQ(ierr); 108aad13602SShrirang Abhyankar ierr = TaoGetSolutionVector(tao,&x);CHKERRQ(ierr); 109aad13602SShrirang Abhyankar ierr = VecView(x,PETSC_VIEWER_STDOUT_WORLD);CHKERRQ(ierr); 110aad13602SShrirang Abhyankar 111aad13602SShrirang Abhyankar /* Free objects */ 112aad13602SShrirang Abhyankar ierr = DestroyProblem(&user);CHKERRQ(ierr); 113aad13602SShrirang Abhyankar ierr = TaoDestroy(&tao);CHKERRQ(ierr); 114aad13602SShrirang Abhyankar ierr = PetscFinalize(); 115aad13602SShrirang Abhyankar return ierr; 116aad13602SShrirang Abhyankar } 117aad13602SShrirang Abhyankar 118aad13602SShrirang Abhyankar PetscErrorCode InitializeProblem(AppCtx *user) 119aad13602SShrirang Abhyankar { 120aad13602SShrirang Abhyankar PetscErrorCode ierr; 121aad13602SShrirang Abhyankar PetscMPIInt size; 122aad13602SShrirang Abhyankar PetscMPIInt rank; 123aad13602SShrirang Abhyankar PetscInt nloc,neloc,niloc; 124aad13602SShrirang Abhyankar 125aad13602SShrirang Abhyankar PetscFunctionBegin; 126ffc4695bSBarry Smith ierr = MPI_Comm_size(PETSC_COMM_WORLD,&size);CHKERRMPI(ierr); 127ffc4695bSBarry Smith ierr = MPI_Comm_rank(PETSC_COMM_WORLD,&rank);CHKERRMPI(ierr); 1288e58fa1dSresundermann user->noeqflag = PETSC_FALSE; 1298e58fa1dSresundermann ierr = PetscOptionsGetBool(NULL,NULL,"-no_eq",&user->noeqflag,NULL);CHKERRQ(ierr); 130628da978Sresundermann 1318e58fa1dSresundermann if (!user->noeqflag) { 1328e58fa1dSresundermann ierr = PetscPrintf(PETSC_COMM_WORLD,"Solution should be f(1,1)=-2\n");CHKERRQ(ierr); 1338e58fa1dSresundermann } 134aad13602SShrirang Abhyankar 1352d4ee042Sprj- /* create vector x and set initial values */ 136aad13602SShrirang Abhyankar user->n = 2; /* global length */ 137aad13602SShrirang Abhyankar nloc = (rank==0)?user->n:0; 138661095bbSAlp Dener ierr = VecCreate(PETSC_COMM_WORLD,&user->x);CHKERRQ(ierr); 139661095bbSAlp Dener ierr = VecSetSizes(user->x,nloc,user->n);CHKERRQ(ierr); 140661095bbSAlp Dener ierr = VecSetFromOptions(user->x);CHKERRQ(ierr); 141aad13602SShrirang Abhyankar ierr = VecSet(user->x,0);CHKERRQ(ierr); 142aad13602SShrirang Abhyankar 143aad13602SShrirang Abhyankar /* create and set lower and upper bound vectors */ 144aad13602SShrirang Abhyankar ierr = VecDuplicate(user->x,&user->xl);CHKERRQ(ierr); 145aad13602SShrirang Abhyankar ierr = VecDuplicate(user->x,&user->xu);CHKERRQ(ierr); 146aad13602SShrirang Abhyankar ierr = VecSet(user->xl,-1.0);CHKERRQ(ierr); 147aad13602SShrirang Abhyankar ierr = VecSet(user->xu,2.0);CHKERRQ(ierr); 148aad13602SShrirang Abhyankar 149aad13602SShrirang Abhyankar /* create scater to zero */ 150aad13602SShrirang Abhyankar ierr = VecScatterCreateToZero(user->x,&user->scat,&user->Xseq);CHKERRQ(ierr); 151aad13602SShrirang Abhyankar 152aad13602SShrirang Abhyankar user->ne = 1; 153628da978Sresundermann user->ni = 2; 154aad13602SShrirang Abhyankar neloc = (rank==0)?user->ne:0; 155628da978Sresundermann niloc = (rank==0)?user->ni:0; 156aad13602SShrirang Abhyankar 1578e58fa1dSresundermann if (!user->noeqflag) { 158661095bbSAlp Dener ierr = VecCreate(PETSC_COMM_WORLD,&user->ce);CHKERRQ(ierr); /* a 1x1 vec for equality constraints */ 159661095bbSAlp Dener ierr = VecSetSizes(user->ce,neloc,user->ne);CHKERRQ(ierr); 160661095bbSAlp Dener ierr = VecSetFromOptions(user->ce);CHKERRQ(ierr); 161661095bbSAlp Dener ierr = VecSetUp(user->ce);CHKERRQ(ierr); 1628e58fa1dSresundermann } 163628da978Sresundermann 164661095bbSAlp Dener ierr = VecCreate(PETSC_COMM_WORLD,&user->ci);CHKERRQ(ierr); /* a 2x1 vec for inequality constraints */ 165661095bbSAlp Dener ierr = VecSetSizes(user->ci,niloc,user->ni);CHKERRQ(ierr); 166661095bbSAlp Dener ierr = VecSetFromOptions(user->ci);CHKERRQ(ierr); 167661095bbSAlp Dener ierr = VecSetUp(user->ci);CHKERRQ(ierr); 168aad13602SShrirang Abhyankar 169*a5b23f4aSJose E. Roman /* nexn & nixn matricies for equally and inequalty constraints */ 1708e58fa1dSresundermann if (!user->noeqflag) { 171aad13602SShrirang Abhyankar ierr = MatCreate(PETSC_COMM_WORLD,&user->Ae);CHKERRQ(ierr); 1728e58fa1dSresundermann ierr = MatSetSizes(user->Ae,neloc,nloc,user->ne,user->n);CHKERRQ(ierr); 1738e58fa1dSresundermann ierr = MatSetFromOptions(user->Ae);CHKERRQ(ierr); 174661095bbSAlp Dener ierr = MatSetUp(user->Ae);CHKERRQ(ierr); 1758e58fa1dSresundermann } 1768e58fa1dSresundermann 177aad13602SShrirang Abhyankar ierr = MatCreate(PETSC_COMM_WORLD,&user->Ai);CHKERRQ(ierr); 178aad13602SShrirang Abhyankar ierr = MatSetSizes(user->Ai,niloc,nloc,user->ni,user->n);CHKERRQ(ierr); 179aad13602SShrirang Abhyankar ierr = MatSetFromOptions(user->Ai);CHKERRQ(ierr); 180661095bbSAlp Dener ierr = MatSetUp(user->Ai);CHKERRQ(ierr); 181628da978Sresundermann 182628da978Sresundermann ierr = MatCreate(PETSC_COMM_WORLD,&user->H);CHKERRQ(ierr); 183628da978Sresundermann ierr = MatSetSizes(user->H,nloc,nloc,user->n,user->n);CHKERRQ(ierr); 184628da978Sresundermann ierr = MatSetFromOptions(user->H);CHKERRQ(ierr); 185661095bbSAlp Dener ierr = MatSetUp(user->H);CHKERRQ(ierr); 186aad13602SShrirang Abhyankar PetscFunctionReturn(0); 187aad13602SShrirang Abhyankar } 188aad13602SShrirang Abhyankar 189aad13602SShrirang Abhyankar PetscErrorCode DestroyProblem(AppCtx *user) 190aad13602SShrirang Abhyankar { 191aad13602SShrirang Abhyankar PetscErrorCode ierr; 192aad13602SShrirang Abhyankar 193aad13602SShrirang Abhyankar PetscFunctionBegin; 1948e58fa1dSresundermann if (!user->noeqflag) { 195aad13602SShrirang Abhyankar ierr = MatDestroy(&user->Ae);CHKERRQ(ierr); 1968e58fa1dSresundermann } 197aad13602SShrirang Abhyankar ierr = MatDestroy(&user->Ai);CHKERRQ(ierr); 198aad13602SShrirang Abhyankar ierr = MatDestroy(&user->H);CHKERRQ(ierr); 199aad13602SShrirang Abhyankar 200aad13602SShrirang Abhyankar ierr = VecDestroy(&user->x);CHKERRQ(ierr); 2018e58fa1dSresundermann if (!user->noeqflag) { 202aad13602SShrirang Abhyankar ierr = VecDestroy(&user->ce);CHKERRQ(ierr); 2038e58fa1dSresundermann } 204aad13602SShrirang Abhyankar ierr = VecDestroy(&user->ci);CHKERRQ(ierr); 205aad13602SShrirang Abhyankar ierr = VecDestroy(&user->xl);CHKERRQ(ierr); 206aad13602SShrirang Abhyankar ierr = VecDestroy(&user->xu);CHKERRQ(ierr); 207aad13602SShrirang Abhyankar ierr = VecDestroy(&user->Xseq);CHKERRQ(ierr); 208aad13602SShrirang Abhyankar ierr = VecScatterDestroy(&user->scat);CHKERRQ(ierr); 209aad13602SShrirang Abhyankar PetscFunctionReturn(0); 210aad13602SShrirang Abhyankar } 211aad13602SShrirang Abhyankar 212628da978Sresundermann /* Evaluate 213628da978Sresundermann f(x) = (x0 - 2)^2 + (x1 - 2)^2 - 2*(x0 + x1) 214628da978Sresundermann G = grad f = [2*(x0 - 2) - 2; 215628da978Sresundermann 2*(x1 - 2) - 2] 216aad13602SShrirang Abhyankar */ 217aad13602SShrirang Abhyankar PetscErrorCode FormFunctionGradient(Tao tao, Vec X, PetscReal *f, Vec G, void *ctx) 218aad13602SShrirang Abhyankar { 219aad13602SShrirang Abhyankar PetscScalar g; 220aad13602SShrirang Abhyankar const PetscScalar *x; 221aad13602SShrirang Abhyankar MPI_Comm comm; 222aad13602SShrirang Abhyankar PetscMPIInt rank; 223aad13602SShrirang Abhyankar PetscErrorCode ierr; 224aad13602SShrirang Abhyankar PetscReal fin; 225aad13602SShrirang Abhyankar AppCtx *user=(AppCtx*)ctx; 226aad13602SShrirang Abhyankar Vec Xseq=user->Xseq; 227aad13602SShrirang Abhyankar VecScatter scat=user->scat; 228aad13602SShrirang Abhyankar 229aad13602SShrirang Abhyankar PetscFunctionBegin; 230aad13602SShrirang Abhyankar ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr); 231ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 232aad13602SShrirang Abhyankar 233aad13602SShrirang Abhyankar ierr = VecScatterBegin(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 234aad13602SShrirang Abhyankar ierr = VecScatterEnd(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 235aad13602SShrirang Abhyankar 236aad13602SShrirang Abhyankar fin = 0.0; 237aad13602SShrirang Abhyankar if (!rank) { 238aad13602SShrirang Abhyankar ierr = VecGetArrayRead(Xseq,&x);CHKERRQ(ierr); 239aad13602SShrirang Abhyankar fin = (x[0]-2.0)*(x[0]-2.0) + (x[1]-2.0)*(x[1]-2.0) - 2.0*(x[0]+x[1]); 240aad13602SShrirang Abhyankar g = 2.0*(x[0]-2.0) - 2.0; 241aad13602SShrirang Abhyankar ierr = VecSetValue(G,0,g,INSERT_VALUES);CHKERRQ(ierr); 242aad13602SShrirang Abhyankar g = 2.0*(x[1]-2.0) - 2.0; 243aad13602SShrirang Abhyankar ierr = VecSetValue(G,1,g,INSERT_VALUES);CHKERRQ(ierr); 244aad13602SShrirang Abhyankar ierr = VecRestoreArrayRead(Xseq,&x);CHKERRQ(ierr); 245aad13602SShrirang Abhyankar } 246ffc4695bSBarry Smith ierr = MPI_Allreduce(&fin,f,1,MPIU_REAL,MPIU_SUM,comm);CHKERRMPI(ierr); 247aad13602SShrirang Abhyankar ierr = VecAssemblyBegin(G);CHKERRQ(ierr); 248aad13602SShrirang Abhyankar ierr = VecAssemblyEnd(G);CHKERRQ(ierr); 249aad13602SShrirang Abhyankar PetscFunctionReturn(0); 250aad13602SShrirang Abhyankar } 251aad13602SShrirang Abhyankar 252628da978Sresundermann /* Evaluate 253628da978Sresundermann H = fxx + grad (grad g^T*DI) - grad (grad h^T*DE)] 254628da978Sresundermann = [ 2*(1+de[0]-di[0]+di[1]), 0; 255628da978Sresundermann 0, 2] 256628da978Sresundermann */ 257aad13602SShrirang Abhyankar PetscErrorCode FormHessian(Tao tao, Vec x,Mat H, Mat Hpre, void *ctx) 258aad13602SShrirang Abhyankar { 259aad13602SShrirang Abhyankar AppCtx *user=(AppCtx*)ctx; 260aad13602SShrirang Abhyankar Vec DE,DI; 261aad13602SShrirang Abhyankar const PetscScalar *de,*di; 262aad13602SShrirang Abhyankar PetscInt zero=0,one=1; 263aad13602SShrirang Abhyankar PetscScalar two=2.0; 264aad13602SShrirang Abhyankar PetscScalar val=0.0; 265aad13602SShrirang Abhyankar Vec Deseq,Diseq=user->Xseq; 266aad13602SShrirang Abhyankar VecScatter Descat,Discat=user->scat; 267aad13602SShrirang Abhyankar PetscMPIInt rank; 268aad13602SShrirang Abhyankar MPI_Comm comm; 269aad13602SShrirang Abhyankar PetscErrorCode ierr; 270aad13602SShrirang Abhyankar 271aad13602SShrirang Abhyankar PetscFunctionBegin; 272aad13602SShrirang Abhyankar ierr = TaoGetDualVariables(tao,&DE,&DI);CHKERRQ(ierr); 273aad13602SShrirang Abhyankar 274aad13602SShrirang Abhyankar ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr); 275ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 276aad13602SShrirang Abhyankar 2778e58fa1dSresundermann if (!user->noeqflag) { 278aad13602SShrirang Abhyankar ierr = VecScatterCreateToZero(DE,&Descat,&Deseq);CHKERRQ(ierr); 279aad13602SShrirang Abhyankar ierr = VecScatterBegin(Descat,DE,Deseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 280aad13602SShrirang Abhyankar ierr = VecScatterEnd(Descat,DE,Deseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 2818e58fa1dSresundermann } 2828e58fa1dSresundermann ierr = VecScatterBegin(Discat,DI,Diseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 283aad13602SShrirang Abhyankar ierr = VecScatterEnd(Discat,DI,Diseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 284aad13602SShrirang Abhyankar 285aad13602SShrirang Abhyankar if (!rank) { 2868e58fa1dSresundermann if (!user->noeqflag) { 287aad13602SShrirang Abhyankar ierr = VecGetArrayRead(Deseq,&de);CHKERRQ(ierr); /* places equality constraint dual into array */ 2888e58fa1dSresundermann } 289aad13602SShrirang Abhyankar ierr = VecGetArrayRead(Diseq,&di);CHKERRQ(ierr); /* places inequality constraint dual into array */ 290628da978Sresundermann 2918e58fa1dSresundermann if (!user->noeqflag) { 292628da978Sresundermann val = 2.0 * (1 + de[0] - di[0] + di[1]); 293aad13602SShrirang Abhyankar ierr = VecRestoreArrayRead(Deseq,&de);CHKERRQ(ierr); 294628da978Sresundermann ierr = VecRestoreArrayRead(Diseq,&di);CHKERRQ(ierr); 2958e58fa1dSresundermann } else { 296628da978Sresundermann val = 2.0 * (1 - di[0] + di[1]); 2978e58fa1dSresundermann } 298aad13602SShrirang Abhyankar ierr = VecRestoreArrayRead(Diseq,&di);CHKERRQ(ierr); 299aad13602SShrirang Abhyankar ierr = MatSetValues(H,1,&zero,1,&zero,&val,INSERT_VALUES);CHKERRQ(ierr); 300aad13602SShrirang Abhyankar ierr = MatSetValues(H,1,&one,1,&one,&two,INSERT_VALUES);CHKERRQ(ierr); 301aad13602SShrirang Abhyankar } 302aad13602SShrirang Abhyankar ierr = MatAssemblyBegin(H,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 303aad13602SShrirang Abhyankar ierr = MatAssemblyEnd(H,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 3048e58fa1dSresundermann if (!user->noeqflag) { 305aad13602SShrirang Abhyankar ierr = VecScatterDestroy(&Descat);CHKERRQ(ierr); 306aad13602SShrirang Abhyankar ierr = VecDestroy(&Deseq);CHKERRQ(ierr); 3078e58fa1dSresundermann } 308aad13602SShrirang Abhyankar PetscFunctionReturn(0); 309aad13602SShrirang Abhyankar } 310aad13602SShrirang Abhyankar 311628da978Sresundermann /* Evaluate 312628da978Sresundermann h = [ x0^2 - x1; 313628da978Sresundermann 1 -(x0^2 - x1)] 314628da978Sresundermann */ 315aad13602SShrirang Abhyankar PetscErrorCode FormInequalityConstraints(Tao tao,Vec X,Vec CI,void *ctx) 316aad13602SShrirang Abhyankar { 317aad13602SShrirang Abhyankar const PetscScalar *x; 318aad13602SShrirang Abhyankar PetscScalar ci; 319aad13602SShrirang Abhyankar PetscErrorCode ierr; 320aad13602SShrirang Abhyankar MPI_Comm comm; 321aad13602SShrirang Abhyankar PetscMPIInt rank; 322aad13602SShrirang Abhyankar AppCtx *user=(AppCtx*)ctx; 323aad13602SShrirang Abhyankar Vec Xseq=user->Xseq; 324aad13602SShrirang Abhyankar VecScatter scat=user->scat; 325aad13602SShrirang Abhyankar 326aad13602SShrirang Abhyankar PetscFunctionBegin; 327aad13602SShrirang Abhyankar ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr); 328ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 329aad13602SShrirang Abhyankar 330aad13602SShrirang Abhyankar ierr = VecScatterBegin(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 331aad13602SShrirang Abhyankar ierr = VecScatterEnd(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 332aad13602SShrirang Abhyankar 333aad13602SShrirang Abhyankar if (!rank) { 334aad13602SShrirang Abhyankar ierr = VecGetArrayRead(Xseq,&x);CHKERRQ(ierr); 335aad13602SShrirang Abhyankar ci = x[0]*x[0] - x[1]; 336aad13602SShrirang Abhyankar ierr = VecSetValue(CI,0,ci,INSERT_VALUES);CHKERRQ(ierr); 337aad13602SShrirang Abhyankar ci = -x[0]*x[0] + x[1] + 1.0; 338aad13602SShrirang Abhyankar ierr = VecSetValue(CI,1,ci,INSERT_VALUES);CHKERRQ(ierr); 339aad13602SShrirang Abhyankar ierr = VecRestoreArrayRead(Xseq,&x);CHKERRQ(ierr); 340aad13602SShrirang Abhyankar } 341aad13602SShrirang Abhyankar ierr = VecAssemblyBegin(CI);CHKERRQ(ierr); 342aad13602SShrirang Abhyankar ierr = VecAssemblyEnd(CI);CHKERRQ(ierr); 343aad13602SShrirang Abhyankar PetscFunctionReturn(0); 344aad13602SShrirang Abhyankar } 345aad13602SShrirang Abhyankar 346628da978Sresundermann /* Evaluate 347628da978Sresundermann g = [ x0^2 + x1 - 2] 348628da978Sresundermann */ 349aad13602SShrirang Abhyankar PetscErrorCode FormEqualityConstraints(Tao tao,Vec X,Vec CE,void *ctx) 350aad13602SShrirang Abhyankar { 351aad13602SShrirang Abhyankar const PetscScalar *x; 352aad13602SShrirang Abhyankar PetscScalar ce; 353aad13602SShrirang Abhyankar PetscErrorCode ierr; 354aad13602SShrirang Abhyankar MPI_Comm comm; 355aad13602SShrirang Abhyankar PetscMPIInt rank; 356aad13602SShrirang Abhyankar AppCtx *user=(AppCtx*)ctx; 357aad13602SShrirang Abhyankar Vec Xseq=user->Xseq; 358aad13602SShrirang Abhyankar VecScatter scat=user->scat; 359aad13602SShrirang Abhyankar 360aad13602SShrirang Abhyankar PetscFunctionBegin; 361aad13602SShrirang Abhyankar ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr); 362ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 363aad13602SShrirang Abhyankar 364aad13602SShrirang Abhyankar ierr = VecScatterBegin(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 365aad13602SShrirang Abhyankar ierr = VecScatterEnd(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 366aad13602SShrirang Abhyankar 367aad13602SShrirang Abhyankar if (!rank) { 368aad13602SShrirang Abhyankar ierr = VecGetArrayRead(Xseq,&x);CHKERRQ(ierr); 369aad13602SShrirang Abhyankar ce = x[0]*x[0] + x[1] - 2.0; 370aad13602SShrirang Abhyankar ierr = VecSetValue(CE,0,ce,INSERT_VALUES);CHKERRQ(ierr); 371aad13602SShrirang Abhyankar ierr = VecRestoreArrayRead(Xseq,&x);CHKERRQ(ierr); 372aad13602SShrirang Abhyankar } 373aad13602SShrirang Abhyankar ierr = VecAssemblyBegin(CE);CHKERRQ(ierr); 374aad13602SShrirang Abhyankar ierr = VecAssemblyEnd(CE);CHKERRQ(ierr); 375aad13602SShrirang Abhyankar PetscFunctionReturn(0); 376aad13602SShrirang Abhyankar } 377aad13602SShrirang Abhyankar 378628da978Sresundermann /* 379628da978Sresundermann grad h = [ 2*x0, -1; 380628da978Sresundermann -2*x0, 1] 381628da978Sresundermann */ 382aad13602SShrirang Abhyankar PetscErrorCode FormInequalityJacobian(Tao tao, Vec X, Mat JI, Mat JIpre, void *ctx) 383aad13602SShrirang Abhyankar { 384aad13602SShrirang Abhyankar AppCtx *user=(AppCtx*)ctx; 385aad13602SShrirang Abhyankar PetscInt cols[2],min,max,i; 386aad13602SShrirang Abhyankar PetscScalar vals[2]; 387aad13602SShrirang Abhyankar const PetscScalar *x; 388aad13602SShrirang Abhyankar PetscErrorCode ierr; 389aad13602SShrirang Abhyankar Vec Xseq=user->Xseq; 390aad13602SShrirang Abhyankar VecScatter scat=user->scat; 391aad13602SShrirang Abhyankar MPI_Comm comm; 392aad13602SShrirang Abhyankar PetscMPIInt rank; 393aad13602SShrirang Abhyankar 394aad13602SShrirang Abhyankar PetscFunctionBegin; 395aad13602SShrirang Abhyankar ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr); 396ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 397aad13602SShrirang Abhyankar ierr = VecScatterBegin(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 398aad13602SShrirang Abhyankar ierr = VecScatterEnd(scat,X,Xseq,INSERT_VALUES,SCATTER_FORWARD);CHKERRQ(ierr); 399aad13602SShrirang Abhyankar 400aad13602SShrirang Abhyankar ierr = VecGetArrayRead(Xseq,&x);CHKERRQ(ierr); 401aad13602SShrirang Abhyankar ierr = MatGetOwnershipRange(JI,&min,&max);CHKERRQ(ierr); 402aad13602SShrirang Abhyankar 403aad13602SShrirang Abhyankar cols[0] = 0; cols[1] = 1; 404aad13602SShrirang Abhyankar for (i=min;i<max;i++) { 405aad13602SShrirang Abhyankar if (i==0) { 406628da978Sresundermann vals[0] = 2*x[0]; vals[1] = -1.0; 407aad13602SShrirang Abhyankar ierr = MatSetValues(JI,1,&i,2,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 408aad13602SShrirang Abhyankar } 409aad13602SShrirang Abhyankar if (i==1) { 410628da978Sresundermann vals[0] = -2*x[0]; vals[1] = 1.0; 411aad13602SShrirang Abhyankar ierr = MatSetValues(JI,1,&i,2,cols,vals,INSERT_VALUES);CHKERRQ(ierr); 412aad13602SShrirang Abhyankar } 413aad13602SShrirang Abhyankar } 414aad13602SShrirang Abhyankar ierr = VecRestoreArrayRead(Xseq,&x);CHKERRQ(ierr); 415aad13602SShrirang Abhyankar 416aad13602SShrirang Abhyankar ierr = MatAssemblyBegin(JI,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 417aad13602SShrirang Abhyankar ierr = MatAssemblyEnd(JI,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 418aad13602SShrirang Abhyankar PetscFunctionReturn(0); 419aad13602SShrirang Abhyankar } 420aad13602SShrirang Abhyankar 421628da978Sresundermann /* 422628da978Sresundermann grad g = [2*x0 423628da978Sresundermann 1.0 ] 424628da978Sresundermann */ 425aad13602SShrirang Abhyankar PetscErrorCode FormEqualityJacobian(Tao tao,Vec X,Mat JE,Mat JEpre,void *ctx) 426aad13602SShrirang Abhyankar { 427aad13602SShrirang Abhyankar PetscInt rows[2]; 428aad13602SShrirang Abhyankar PetscScalar vals[2]; 429aad13602SShrirang Abhyankar const PetscScalar *x; 430aad13602SShrirang Abhyankar PetscMPIInt rank; 431aad13602SShrirang Abhyankar MPI_Comm comm; 432aad13602SShrirang Abhyankar PetscErrorCode ierr; 433aad13602SShrirang Abhyankar 434aad13602SShrirang Abhyankar PetscFunctionBegin; 435aad13602SShrirang Abhyankar ierr = PetscObjectGetComm((PetscObject)tao,&comm);CHKERRQ(ierr); 436ffc4695bSBarry Smith ierr = MPI_Comm_rank(comm,&rank);CHKERRMPI(ierr); 437aad13602SShrirang Abhyankar 438aad13602SShrirang Abhyankar if (!rank) { 439aad13602SShrirang Abhyankar ierr = VecGetArrayRead(X,&x);CHKERRQ(ierr); 440aad13602SShrirang Abhyankar rows[0] = 0; rows[1] = 1; 441aad13602SShrirang Abhyankar vals[0] = 2*x[0]; vals[1] = 1.0; 442aad13602SShrirang Abhyankar ierr = MatSetValues(JE,1,rows,2,rows,vals,INSERT_VALUES);CHKERRQ(ierr); 443aad13602SShrirang Abhyankar ierr = VecRestoreArrayRead(X,&x);CHKERRQ(ierr); 444aad13602SShrirang Abhyankar } 445aad13602SShrirang Abhyankar ierr = MatAssemblyBegin(JE,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 446aad13602SShrirang Abhyankar ierr = MatAssemblyEnd(JE,MAT_FINAL_ASSEMBLY);CHKERRQ(ierr); 447aad13602SShrirang Abhyankar PetscFunctionReturn(0); 448aad13602SShrirang Abhyankar } 449aad13602SShrirang Abhyankar 450aad13602SShrirang Abhyankar /*TEST 451aad13602SShrirang Abhyankar 452aad13602SShrirang Abhyankar build: 453dfd57a17SPierre Jolivet requires: !complex !defined(PETSC_USE_CXX) mumps 454aad13602SShrirang Abhyankar 455aad13602SShrirang Abhyankar test: 45609ee8bb0SRylee Sundermann args: -tao_converged_reason -tao_pdipm_kkt_shift_pd 457aad13602SShrirang Abhyankar 458aad13602SShrirang Abhyankar test: 459aad13602SShrirang Abhyankar suffix: 2 460aad13602SShrirang Abhyankar nsize: 2 46109ee8bb0SRylee Sundermann args: -tao_converged_reason -tao_pdipm_kkt_shift_pd 462aad13602SShrirang Abhyankar 4638e58fa1dSresundermann test: 4648e58fa1dSresundermann suffix: 3 4658e58fa1dSresundermann args: -tao_converged_reason -no_eq 4668e58fa1dSresundermann 4678e58fa1dSresundermann test: 4688e58fa1dSresundermann suffix: 4 4698e58fa1dSresundermann nsize: 2 4708e58fa1dSresundermann args: -tao_converged_reason -no_eq 4718e58fa1dSresundermann 472661095bbSAlp Dener test: 473661095bbSAlp Dener suffix: 5 474661095bbSAlp Dener args: -tao_cmonitor -tao_type almm 475661095bbSAlp Dener 476661095bbSAlp Dener test: 477661095bbSAlp Dener suffix: 6 478661095bbSAlp Dener args: -tao_cmonitor -tao_type almm -tao_almm_type phr 479661095bbSAlp Dener 480661095bbSAlp Dener test: 481661095bbSAlp Dener suffix: 7 482661095bbSAlp Dener nsize: 2 483661095bbSAlp Dener args: -tao_cmonitor -tao_type almm 484661095bbSAlp Dener 485661095bbSAlp Dener test: 486661095bbSAlp Dener suffix: 8 487661095bbSAlp Dener nsize: 2 488661095bbSAlp Dener requires: cuda 489661095bbSAlp Dener args: -tao_cmonitor -tao_type almm -vec_type cuda -mat_type aijcusparse 490661095bbSAlp Dener 491661095bbSAlp Dener test: 492661095bbSAlp Dener suffix: 9 493661095bbSAlp Dener nsize: 2 494661095bbSAlp Dener args: -tao_cmonitor -tao_type almm -no_eq 495661095bbSAlp Dener 496661095bbSAlp Dener test: 497661095bbSAlp Dener suffix: 10 498661095bbSAlp Dener nsize: 2 499661095bbSAlp Dener args: -tao_cmonitor -tao_type almm -tao_almm_type phr -no_eq 500661095bbSAlp Dener 501aad13602SShrirang Abhyankar TEST*/ 502