Lines Matching refs:nadj
883 PetscInt i, j, nadj; in TSAdjointStep_RK() local
911 for (nadj = 0; nadj < ts->numcost; nadj++) { in TSAdjointStep_RK()
915 …PetscCall(VecCopy(ts->vecs_sensi[nadj], VecsSensiTemp[nadj])); /* VecDeltaLam is an vec array of s… in TSAdjointStep_RK()
916 PetscCall(VecMAXPY(VecsSensiTemp[nadj], s - i - 1, w, &VecsDeltaLam[nadj * s + i + 1])); in TSAdjointStep_RK()
917 …PetscCall(MatMultTranspose(J, VecsSensiTemp[nadj], VecsDeltaLam[nadj * s + i])); /* VecsSensiTemp … in TSAdjointStep_RK()
918 PetscCall(VecScale(VecsDeltaLam[nadj * s + i], -h * b[i])); in TSAdjointStep_RK()
920 PetscCall(MatDenseGetColumn(Jquad, nadj, &xarr)); in TSAdjointStep_RK()
922 PetscCall(VecAXPY(VecsDeltaLam[nadj * s + i], -h * b[i], VecDRDUTransCol)); in TSAdjointStep_RK()
928 PetscCall(VecSet(VecsSensiTemp[nadj], 0)); in TSAdjointStep_RK()
929 PetscCall(VecMAXPY(VecsSensiTemp[nadj], s - i - 1, w, &VecsDeltaLam[nadj * s + i + 1])); in TSAdjointStep_RK()
930 PetscCall(MatMultTranspose(J, VecsSensiTemp[nadj], VecsDeltaLam[nadj * s + i])); in TSAdjointStep_RK()
931 PetscCall(VecScale(VecsDeltaLam[nadj * s + i], -h)); in TSAdjointStep_RK()
936 PetscCall(MatMultTranspose(ts->Jacprhs, VecsSensiTemp[nadj], VecDeltaMu)); in TSAdjointStep_RK()
940 PetscCall(MatDenseGetColumn(quadts->Jacprhs, nadj, &xarr)); in TSAdjointStep_RK()
949 … PetscCall(VecAXPY(ts->vecs_sensip[nadj], 1., VecDeltaMu)); /* update sensip for each stage */ in TSAdjointStep_RK()
986 for (nadj = 0; nadj < ts->numcost; nadj++) { in TSAdjointStep_RK()
990 PetscCall(VecCopy(ts->vecs_sensi2[nadj], VecsSensi2Temp[nadj])); in TSAdjointStep_RK()
991 PetscCall(VecMAXPY(VecsSensi2Temp[nadj], s - i - 1, w, &VecsDeltaLam2[nadj * s + i + 1])); in TSAdjointStep_RK()
992 PetscCall(MatMultTranspose(J, VecsSensi2Temp[nadj], VecsDeltaLam2[nadj * s + i])); in TSAdjointStep_RK()
993 PetscCall(VecScale(VecsDeltaLam2[nadj * s + i], -h * b[i])); in TSAdjointStep_RK()
994 PetscCall(VecAXPY(VecsDeltaLam2[nadj * s + i], -h * b[i], ts->vecs_guu[nadj])); in TSAdjointStep_RK()
995 …if (ts->vecs_sensip) PetscCall(VecAXPY(VecsDeltaLam2[nadj * s + i], -h * b[i], ts->vecs_gup[nadj])… in TSAdjointStep_RK()
998 PetscCall(VecSet(VecsDeltaLam2[nadj * s + i], 0)); in TSAdjointStep_RK()
999 PetscCall(VecMAXPY(VecsSensi2Temp[nadj], s - i - 1, w, &VecsDeltaLam2[nadj * s + i + 1])); in TSAdjointStep_RK()
1000 PetscCall(MatMultTranspose(J, VecsSensi2Temp[nadj], VecsDeltaLam2[nadj * s + i])); in TSAdjointStep_RK()
1001 PetscCall(VecScale(VecsDeltaLam2[nadj * s + i], -h)); in TSAdjointStep_RK()
1002 PetscCall(VecAXPY(VecsDeltaLam2[nadj * s + i], -h, ts->vecs_guu[nadj])); in TSAdjointStep_RK()
1003 … if (ts->vecs_sensip) PetscCall(VecAXPY(VecsDeltaLam2[nadj * s + i], -h, ts->vecs_gup[nadj])); in TSAdjointStep_RK()
1006 PetscCall(MatMultTranspose(ts->Jacprhs, VecsSensi2Temp[nadj], VecDeltaMu2)); in TSAdjointStep_RK()
1009 PetscCall(VecAXPY(VecDeltaMu2, -h * b[i], ts->vecs_gpu[nadj])); in TSAdjointStep_RK()
1010 PetscCall(VecAXPY(VecDeltaMu2, -h * b[i], ts->vecs_gpp[nadj])); in TSAdjointStep_RK()
1013 PetscCall(VecAXPY(VecDeltaMu2, -h, ts->vecs_gpu[nadj])); in TSAdjointStep_RK()
1014 PetscCall(VecAXPY(VecDeltaMu2, -h, ts->vecs_gpp[nadj])); in TSAdjointStep_RK()
1016 … PetscCall(VecAXPY(ts->vecs_sensi2p[nadj], 1, VecDeltaMu2)); /* update sensi2p for each stage */ in TSAdjointStep_RK()
1023 for (nadj = 0; nadj < ts->numcost; nadj++) { /* no need to do this for mu's */ in TSAdjointStep_RK()
1024 PetscCall(VecMAXPY(ts->vecs_sensi[nadj], s, w, &VecsDeltaLam[nadj * s])); in TSAdjointStep_RK()
1025 if (ts->vecs_sensi2) PetscCall(VecMAXPY(ts->vecs_sensi2[nadj], s, w, &VecsDeltaLam2[nadj * s])); in TSAdjointStep_RK()