Lines Matching refs:nadj
1569 PetscInt i, j, nadj; in TSAdjointStep_ARKIMEX() local
1595 for (nadj = 0; nadj < ts->numcost; nadj++) { in TSAdjointStep_ARKIMEX()
1597 PetscCall(VecSet(VecsSensiTemp[nadj], 0)); in TSAdjointStep_ARKIMEX()
1601 PetscCall(VecMAXPY(VecsSensiTemp[nadj], s - i - 1, w, &VecsDeltaLam[nadj * s + i + 1])); in TSAdjointStep_ARKIMEX()
1604 PetscCall(VecAXPY(VecsSensiTemp[nadj], -bt[i], ts->vecs_sensi[nadj])); in TSAdjointStep_ARKIMEX()
1607 PetscCall(MatMultTranspose(Jim, VecsSensiTemp[nadj], VecsDeltaLam[nadj * s + i])); in TSAdjointStep_ARKIMEX()
1609 …PetscCall(VecAXPY(VecsDeltaLam[nadj * s + i], ark->scoeff / adjoint_time_step, VecsSensiTemp[nadj]… in TSAdjointStep_ARKIMEX()
1612 PetscCall(MatMultTranspose(ts->Jacp, VecsSensiTemp[nadj], VecsSensiPTemp[nadj])); in TSAdjointStep_ARKIMEX()
1614 PetscCall(VecAXPY(ts->vecs_sensip[nadj], adjoint_time_step, VecsSensiPTemp[nadj])); in TSAdjointStep_ARKIMEX()
1617 PetscCall(VecSet(VecsDeltaLam[nadj * s + i], 0)); // make sure it is initialized in TSAdjointStep_ARKIMEX()
1620 PetscCall(VecSet(VecsSensiTemp[nadj], 0)); in TSAdjointStep_ARKIMEX()
1624 PetscCall(VecMAXPY(VecsSensiTemp[nadj], s - i - 1, w, &VecsDeltaLam[nadj * s + i + 1])); in TSAdjointStep_ARKIMEX()
1627 PetscCall(VecAXPY(VecsSensiTemp[nadj], b[i], ts->vecs_sensi[nadj])); in TSAdjointStep_ARKIMEX()
1630 …PetscCall(MatMultTransposeAdd(Jex, VecsSensiTemp[nadj], VecsDeltaLam[nadj * s + i], VecsDeltaLam[n… in TSAdjointStep_ARKIMEX()
1633 PetscCall(MatMultTranspose(ts->Jacprhs, VecsSensiTemp[nadj], VecsSensiPTemp[nadj])); in TSAdjointStep_ARKIMEX()
1635 PetscCall(VecAXPY(ts->vecs_sensip[nadj], adjoint_time_step, VecsSensiPTemp[nadj])); in TSAdjointStep_ARKIMEX()
1640 PetscCall(VecScale(VecsDeltaLam[nadj * s + i], adjoint_time_step)); in TSAdjointStep_ARKIMEX()
1646 PetscCall(VecScale(VecsDeltaLam[nadj * s + i], 1. / At[i * s + i])); in TSAdjointStep_ARKIMEX()
1647 PetscCall(KSPSolveTranspose(ksp, VecsDeltaLam[nadj * s + i], VecsDeltaLam[nadj * s + i])); in TSAdjointStep_ARKIMEX()
1651 …st function, transposed linear solve fails, stopping 1st-order adjoint solve\n", ts->steps, nadj)); in TSAdjointStep_ARKIMEX()
1655 PetscCall(MatMultTranspose(ts->Jacp, VecsDeltaLam[nadj * s + i], VecsSensiPTemp[nadj])); in TSAdjointStep_ARKIMEX()
1657 …PetscCall(VecAXPY(ts->vecs_sensip[nadj], -At[i * s + i] * adjoint_time_step, VecsSensiPTemp[nadj])… in TSAdjointStep_ARKIMEX()
1663 for (nadj = 0; nadj < ts->numcost; nadj++) // no need to do this for mu's in TSAdjointStep_ARKIMEX()
1664 PetscCall(VecMAXPY(ts->vecs_sensi[nadj], s, w, &VecsDeltaLam[nadj * s])); in TSAdjointStep_ARKIMEX()