Lines Matching refs:aj
279 const PetscInt *aj, *ranges, *Jranges, *rranges, *cranges; in TaoSNESJacobian_PDIPM() local
319 PetscCall(MatGetRow(tao->jacobian_equality, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
322 while (aj[j] >= cranges[proc + 1]) proc++; in TaoSNESJacobian_PDIPM()
323 cols[0] = aj[j] - cranges[proc] + Jranges[proc]; in TaoSNESJacobian_PDIPM()
326 PetscCall(MatRestoreRow(tao->jacobian_equality, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
339 PetscCall(MatGetRow(tao->jacobian_inequality, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
342 while (aj[j] >= cranges[proc + 1]) proc++; in TaoSNESJacobian_PDIPM()
343 cols[0] = aj[j] - cranges[proc] + Jranges[proc]; in TaoSNESJacobian_PDIPM()
346 PetscCall(MatRestoreRow(tao->jacobian_inequality, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
371 PetscCall(MatGetRow(tao->hessian, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
374 while (aj[j] >= cranges[proc + 1]) proc++; in TaoSNESJacobian_PDIPM()
375 cols[0] = aj[j] - cranges[proc] + Jranges[proc]; in TaoSNESJacobian_PDIPM()
383 PetscCall(MatRestoreRow(tao->hessian, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
387 PetscCall(MatGetRow(pdipm->jac_equality_trans, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
392 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSNESJacobian_PDIPM()
394 cols[0] = aj[j] - ranges[proc] + Jranges[proc] + nx_all; in TaoSNESJacobian_PDIPM()
397 PetscCall(MatRestoreRow(pdipm->jac_equality_trans, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
402 PetscCall(MatGetRow(pdipm->jac_inequality_trans, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
407 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSNESJacobian_PDIPM()
409 cols[0] = aj[j] - ranges[proc] + Jranges[proc] + nx_all + nce_all[proc]; in TaoSNESJacobian_PDIPM()
412 PetscCall(MatRestoreRow(pdipm->jac_inequality_trans, i + rjstart, &nc, &aj, &aa)); in TaoSNESJacobian_PDIPM()
845 const PetscInt *cols, *rranges, *cranges, *aj, *ranges; in TaoSetup_PDIPM() local
1072 PetscCall(MatGetRow(tao->hessian, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1075 while (aj[j] >= cranges[proc + 1]) proc++; in TaoSetup_PDIPM()
1076 col = aj[j] - cranges[proc] + Jranges[proc]; in TaoSetup_PDIPM()
1079 PetscCall(MatRestoreRow(tao->hessian, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1083 PetscCall(MatGetRow(pdipm->jac_equality_trans, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1088 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSetup_PDIPM()
1090 col = aj[j] - ranges[proc] + Jranges[proc] + nx_all; in TaoSetup_PDIPM()
1093 PetscCall(MatRestoreRow(pdipm->jac_equality_trans, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1098 PetscCall(MatGetRow(Jce_xfixed_trans, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1103 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSetup_PDIPM()
1105 col = aj[j] - ranges[proc] + Jranges[proc] + nx_all + ng_all[proc]; in TaoSetup_PDIPM()
1108 PetscCall(MatRestoreRow(Jce_xfixed_trans, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1113 PetscCall(MatGetRow(pdipm->jac_inequality_trans, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1118 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSetup_PDIPM()
1120 col = aj[j] - ranges[proc] + Jranges[proc] + nx_all + nce_all[proc]; in TaoSetup_PDIPM()
1123 PetscCall(MatRestoreRow(pdipm->jac_inequality_trans, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1127 PetscCall(MatGetRow(Jci_xb_trans, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1132 while (aj[j] >= ranges[proc + 1]) proc++; in TaoSetup_PDIPM()
1134 col = aj[j] - ranges[proc] + Jranges[proc] + nx_all + nce_all[proc] + nh_all[proc]; in TaoSetup_PDIPM()
1137 PetscCall(MatRestoreRow(Jci_xb_trans, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1146 PetscCall(MatGetRow(tao->jacobian_equality, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1149 while (aj[j] >= cranges[proc + 1]) proc++; in TaoSetup_PDIPM()
1150 col = aj[j] - cranges[proc] + Jranges[proc]; in TaoSetup_PDIPM()
1153 PetscCall(MatRestoreRow(tao->jacobian_equality, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1180 PetscCall(MatGetRow(tao->jacobian_inequality, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()
1183 while (aj[j] >= cranges[proc + 1]) proc++; in TaoSetup_PDIPM()
1184 col = aj[j] - cranges[proc] + Jranges[proc]; in TaoSetup_PDIPM()
1187 PetscCall(MatRestoreRow(tao->jacobian_inequality, i + rjstart, &nc, &aj, NULL)); in TaoSetup_PDIPM()