Lines Matching refs:k
15 PetscInt nz, *vj, k, idx, k1; in MatSolve_SeqSBAIJ_N_inplace() local
31 for (k = 0; k < mbs; k++) { /* t <- perm(b) */ in MatSolve_SeqSBAIJ_N_inplace()
32 idx = bs * r[k]; in MatSolve_SeqSBAIJ_N_inplace()
35 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_N_inplace()
36 v = aa + bs2 * ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
37 xk = t + k * bs; /* Dk*xk = k-th block of x */ in MatSolve_SeqSBAIJ_N_inplace()
39 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
40 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
50 diag = aa + k * bs2; /* ptr to inv(Dk) */ in MatSolve_SeqSBAIJ_N_inplace()
55 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_N_inplace()
56 v = aa + bs2 * ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
57 xk = t + k * bs; /* xk */ in MatSolve_SeqSBAIJ_N_inplace()
58 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
59 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_N_inplace()
68 idx = bs * r[k]; in MatSolve_SeqSBAIJ_N_inplace()
94 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering() local
102 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
103 v = aa + bs2 * ai[k]; in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
104 xk = x + k * bs; /* Dk*xk = k-th block of x */ in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
106 nz = ai[k + 1] - ai[k]; in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
107 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
117 diag = aa + k * bs2; /* ptr to inv(Dk) */ in MatForwardSolve_SeqSBAIJ_N_NaturalOrdering()
126 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering() local
132 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering()
133 v = aa + bs2 * ai[k]; in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering()
134 xk = x + k * bs; /* xk */ in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering()
135 nz = ai[k + 1] - ai[k]; in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering()
136 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_N_NaturalOrdering()
220 PetscInt nz, k, idx; in MatSolve_SeqSBAIJ_7_inplace() local
233 for (k = 0; k < mbs; k++) { /* t <- perm(b) */ in MatSolve_SeqSBAIJ_7_inplace()
234 idx = 7 * r[k]; in MatSolve_SeqSBAIJ_7_inplace()
245 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_7_inplace()
246 v = aa + 49 * ai[k]; in MatSolve_SeqSBAIJ_7_inplace()
247 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_7_inplace()
248 tp = t + k * 7; in MatSolve_SeqSBAIJ_7_inplace()
256 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_7_inplace()
272 d = aa + k * 49; /* ptr to inv(Dk) */ in MatSolve_SeqSBAIJ_7_inplace()
273 tp = t + k * 7; in MatSolve_SeqSBAIJ_7_inplace()
284 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_7_inplace()
285 v = aa + 49 * ai[k]; in MatSolve_SeqSBAIJ_7_inplace()
286 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_7_inplace()
287 tp = t + k * 7; in MatSolve_SeqSBAIJ_7_inplace()
295 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_7_inplace()
311 tp = t + k * 7; in MatSolve_SeqSBAIJ_7_inplace()
319 idx = 7 * r[k]; in MatSolve_SeqSBAIJ_7_inplace()
340 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_7_NaturalOrdering() local
344 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_7_NaturalOrdering()
345 v = aa + 49 * ai[k]; in MatForwardSolve_SeqSBAIJ_7_NaturalOrdering()
346 xp = x + k * 7; in MatForwardSolve_SeqSBAIJ_7_NaturalOrdering()
354 nz = ai[k + 1] - ai[k]; in MatForwardSolve_SeqSBAIJ_7_NaturalOrdering()
355 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_7_NaturalOrdering()
372 d = aa + k * 49; /* ptr to inv(Dk) */ in MatForwardSolve_SeqSBAIJ_7_NaturalOrdering()
373 xp = x + k * 7; in MatForwardSolve_SeqSBAIJ_7_NaturalOrdering()
389 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering() local
393 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering()
394 v = aa + 49 * ai[k]; in MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering()
395 xp = x + k * 7; in MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering()
403 nz = ai[k + 1] - ai[k]; in MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering()
404 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering()
420 xp = x + k * 7; in MatBackwardSolve_SeqSBAIJ_7_NaturalOrdering()
500 PetscInt nz, k, idx; in MatSolve_SeqSBAIJ_6_inplace() local
513 for (k = 0; k < mbs; k++) { /* t <- perm(b) */ in MatSolve_SeqSBAIJ_6_inplace()
514 idx = 6 * r[k]; in MatSolve_SeqSBAIJ_6_inplace()
524 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_6_inplace()
525 v = aa + 36 * ai[k]; in MatSolve_SeqSBAIJ_6_inplace()
526 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_6_inplace()
527 tp = t + k * 6; in MatSolve_SeqSBAIJ_6_inplace()
534 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_6_inplace()
549 d = aa + k * 36; /* ptr to inv(Dk) */ in MatSolve_SeqSBAIJ_6_inplace()
550 tp = t + k * 6; in MatSolve_SeqSBAIJ_6_inplace()
560 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_6_inplace()
561 v = aa + 36 * ai[k]; in MatSolve_SeqSBAIJ_6_inplace()
562 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_6_inplace()
563 tp = t + k * 6; in MatSolve_SeqSBAIJ_6_inplace()
570 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_6_inplace()
585 tp = t + k * 6; in MatSolve_SeqSBAIJ_6_inplace()
592 idx = 6 * r[k]; in MatSolve_SeqSBAIJ_6_inplace()
612 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_6_NaturalOrdering() local
616 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_6_NaturalOrdering()
617 v = aa + 36 * ai[k]; in MatForwardSolve_SeqSBAIJ_6_NaturalOrdering()
618 xp = x + k * 6; in MatForwardSolve_SeqSBAIJ_6_NaturalOrdering()
625 nz = ai[k + 1] - ai[k]; in MatForwardSolve_SeqSBAIJ_6_NaturalOrdering()
626 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_6_NaturalOrdering()
642 d = aa + k * 36; /* ptr to inv(Dk) */ in MatForwardSolve_SeqSBAIJ_6_NaturalOrdering()
643 xp = x + k * 6; in MatForwardSolve_SeqSBAIJ_6_NaturalOrdering()
657 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering() local
661 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering()
662 v = aa + 36 * ai[k]; in MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering()
663 xp = x + k * 6; in MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering()
670 nz = ai[k + 1] - ai[k]; in MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering()
671 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering()
686 xp = x + k * 6; in MatBackwardSolve_SeqSBAIJ_6_NaturalOrdering()
766 PetscInt nz, k, idx; in MatSolve_SeqSBAIJ_5_inplace() local
779 for (k = 0; k < mbs; k++) { /* t <- perm(b) */ in MatSolve_SeqSBAIJ_5_inplace()
780 idx = 5 * r[k]; in MatSolve_SeqSBAIJ_5_inplace()
789 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_5_inplace()
790 v = aa + 25 * ai[k]; in MatSolve_SeqSBAIJ_5_inplace()
791 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_5_inplace()
792 tp = t + k * 5; in MatSolve_SeqSBAIJ_5_inplace()
798 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_5_inplace()
813 diag = aa + k * 25; /* ptr to inv(Dk) */ in MatSolve_SeqSBAIJ_5_inplace()
814 tp = t + k * 5; in MatSolve_SeqSBAIJ_5_inplace()
823 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_5_inplace()
824 v = aa + 25 * ai[k]; in MatSolve_SeqSBAIJ_5_inplace()
825 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_5_inplace()
826 tp = t + k * 5; in MatSolve_SeqSBAIJ_5_inplace()
832 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_5_inplace()
846 tp = t + k * 5; in MatSolve_SeqSBAIJ_5_inplace()
852 idx = 5 * r[k]; in MatSolve_SeqSBAIJ_5_inplace()
871 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_5_NaturalOrdering() local
875 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_5_NaturalOrdering()
876 v = aa + 25 * ai[k]; in MatForwardSolve_SeqSBAIJ_5_NaturalOrdering()
877 xp = x + k * 5; in MatForwardSolve_SeqSBAIJ_5_NaturalOrdering()
883 nz = ai[k + 1] - ai[k]; in MatForwardSolve_SeqSBAIJ_5_NaturalOrdering()
884 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_5_NaturalOrdering()
899 diag = aa + k * 25; /* ptr to inv(Dk) */ in MatForwardSolve_SeqSBAIJ_5_NaturalOrdering()
900 xp = x + k * 5; in MatForwardSolve_SeqSBAIJ_5_NaturalOrdering()
914 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering() local
918 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering()
919 v = aa + 25 * ai[k]; in MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering()
920 xp = x + k * 5; in MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering()
926 nz = ai[k + 1] - ai[k]; in MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering()
927 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering()
941 xp = x + k * 5; in MatBackwardSolve_SeqSBAIJ_5_NaturalOrdering()
1020 PetscInt nz, k, idx; in MatSolve_SeqSBAIJ_4_inplace() local
1033 for (k = 0; k < mbs; k++) { /* t <- perm(b) */ in MatSolve_SeqSBAIJ_4_inplace()
1034 idx = 4 * r[k]; in MatSolve_SeqSBAIJ_4_inplace()
1042 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_4_inplace()
1043 v = aa + 16 * ai[k]; in MatSolve_SeqSBAIJ_4_inplace()
1044 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_4_inplace()
1045 tp = t + k * 4; in MatSolve_SeqSBAIJ_4_inplace()
1050 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_4_inplace()
1064 diag = aa + k * 16; /* ptr to inv(Dk) */ in MatSolve_SeqSBAIJ_4_inplace()
1065 tp = t + k * 4; in MatSolve_SeqSBAIJ_4_inplace()
1073 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_4_inplace()
1074 v = aa + 16 * ai[k]; in MatSolve_SeqSBAIJ_4_inplace()
1075 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_4_inplace()
1076 tp = t + k * 4; in MatSolve_SeqSBAIJ_4_inplace()
1081 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_4_inplace()
1094 tp = t + k * 4; in MatSolve_SeqSBAIJ_4_inplace()
1099 idx = 4 * r[k]; in MatSolve_SeqSBAIJ_4_inplace()
1117 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_4_NaturalOrdering() local
1121 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_4_NaturalOrdering()
1122 v = aa + 16 * ai[k]; in MatForwardSolve_SeqSBAIJ_4_NaturalOrdering()
1123 xp = x + k * 4; in MatForwardSolve_SeqSBAIJ_4_NaturalOrdering()
1128 nz = ai[k + 1] - ai[k]; in MatForwardSolve_SeqSBAIJ_4_NaturalOrdering()
1129 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_4_NaturalOrdering()
1143 diag = aa + k * 16; /* ptr to inv(Dk) */ in MatForwardSolve_SeqSBAIJ_4_NaturalOrdering()
1144 xp = x + k * 4; in MatForwardSolve_SeqSBAIJ_4_NaturalOrdering()
1157 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering() local
1161 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering()
1162 v = aa + 16 * ai[k]; in MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering()
1163 xp = x + k * 4; in MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering()
1168 nz = ai[k + 1] - ai[k]; in MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering()
1169 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering()
1182 xp = x + k * 4; in MatBackwardSolve_SeqSBAIJ_4_NaturalOrdering()
1259 PetscInt nz, k, idx; in MatSolve_SeqSBAIJ_3_inplace() local
1273 for (k = 0; k < mbs; k++) { /* t <- perm(b) */ in MatSolve_SeqSBAIJ_3_inplace()
1274 idx = 3 * r[k]; in MatSolve_SeqSBAIJ_3_inplace()
1281 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_3_inplace()
1282 v = aa + 9 * ai[k]; in MatSolve_SeqSBAIJ_3_inplace()
1283 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_3_inplace()
1284 tp = t + k * 3; in MatSolve_SeqSBAIJ_3_inplace()
1288 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_3_inplace()
1301 diag = aa + k * 9; /* ptr to inv(Dk) */ in MatSolve_SeqSBAIJ_3_inplace()
1302 tp = t + k * 3; in MatSolve_SeqSBAIJ_3_inplace()
1309 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_3_inplace()
1310 v = aa + 9 * ai[k]; in MatSolve_SeqSBAIJ_3_inplace()
1311 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_3_inplace()
1312 tp = t + k * 3; in MatSolve_SeqSBAIJ_3_inplace()
1316 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_3_inplace()
1328 tp = t + k * 3; in MatSolve_SeqSBAIJ_3_inplace()
1332 idx = 3 * r[k]; in MatSolve_SeqSBAIJ_3_inplace()
1349 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_3_NaturalOrdering() local
1353 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_3_NaturalOrdering()
1354 v = aa + 9 * ai[k]; in MatForwardSolve_SeqSBAIJ_3_NaturalOrdering()
1355 xp = x + k * 3; in MatForwardSolve_SeqSBAIJ_3_NaturalOrdering()
1359 nz = ai[k + 1] - ai[k]; in MatForwardSolve_SeqSBAIJ_3_NaturalOrdering()
1360 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_3_NaturalOrdering()
1373 diag = aa + k * 9; /* ptr to inv(Dk) */ in MatForwardSolve_SeqSBAIJ_3_NaturalOrdering()
1374 xp = x + k * 3; in MatForwardSolve_SeqSBAIJ_3_NaturalOrdering()
1386 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering() local
1390 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering()
1391 v = aa + 9 * ai[k]; in MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering()
1392 xp = x + k * 3; in MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering()
1396 nz = ai[k + 1] - ai[k]; in MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering()
1397 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering()
1409 xp = x + k * 3; in MatBackwardSolve_SeqSBAIJ_3_NaturalOrdering()
1486 PetscInt nz, k, k2, idx; in MatSolve_SeqSBAIJ_2_inplace() local
1498 for (k = 0; k < mbs; k++) { /* t <- perm(b) */ in MatSolve_SeqSBAIJ_2_inplace()
1499 idx = 2 * r[k]; in MatSolve_SeqSBAIJ_2_inplace()
1500 t[k * 2] = b[idx]; in MatSolve_SeqSBAIJ_2_inplace()
1501 t[k * 2 + 1] = b[idx + 1]; in MatSolve_SeqSBAIJ_2_inplace()
1503 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_2_inplace()
1504 v = aa + 4 * ai[k]; in MatSolve_SeqSBAIJ_2_inplace()
1505 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_2_inplace()
1506 k2 = k * 2; in MatSolve_SeqSBAIJ_2_inplace()
1509 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_2_inplace()
1516 diag = aa + k * 4; /* ptr to inv(Dk) */ in MatSolve_SeqSBAIJ_2_inplace()
1522 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_2_inplace()
1523 v = aa + 4 * ai[k]; in MatSolve_SeqSBAIJ_2_inplace()
1524 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_2_inplace()
1525 k2 = k * 2; in MatSolve_SeqSBAIJ_2_inplace()
1528 nz = ai[k + 1] - ai[k]; in MatSolve_SeqSBAIJ_2_inplace()
1537 idx = 2 * r[k]; in MatSolve_SeqSBAIJ_2_inplace()
1553 PetscInt nz, k, k2; in MatForwardSolve_SeqSBAIJ_2_NaturalOrdering() local
1557 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_2_NaturalOrdering()
1558 v = aa + 4 * ai[k]; in MatForwardSolve_SeqSBAIJ_2_NaturalOrdering()
1559 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_2_NaturalOrdering()
1560 k2 = k * 2; in MatForwardSolve_SeqSBAIJ_2_NaturalOrdering()
1563 nz = ai[k + 1] - ai[k]; in MatForwardSolve_SeqSBAIJ_2_NaturalOrdering()
1574 diag = aa + k * 4; /* ptr to inv(Dk) */ in MatForwardSolve_SeqSBAIJ_2_NaturalOrdering()
1585 PetscInt nz, k, k2; in MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering() local
1589 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering()
1590 v = aa + 4 * ai[k]; in MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering()
1591 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering()
1592 k2 = k * 2; in MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering()
1595 nz = ai[k + 1] - ai[k]; in MatBackwardSolve_SeqSBAIJ_2_NaturalOrdering()
1682 PetscInt nz, k, j; in MatSolve_SeqSBAIJ_1() local
1691 for (k = 0; k < mbs; k++) t[k] = b[rp[k]]; in MatSolve_SeqSBAIJ_1()
1692 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_1()
1693 v = aa + ai[k]; in MatSolve_SeqSBAIJ_1()
1694 vj = aj + ai[k]; in MatSolve_SeqSBAIJ_1()
1695 xk = t[k]; in MatSolve_SeqSBAIJ_1()
1696 nz = ai[k + 1] - ai[k] - 1; in MatSolve_SeqSBAIJ_1()
1698 t[k] = xk * v[nz]; /* v[nz] = 1/D(k) */ in MatSolve_SeqSBAIJ_1()
1702 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_1()
1703 v = aa + adiag[k] - 1; in MatSolve_SeqSBAIJ_1()
1704 vj = aj + adiag[k] - 1; in MatSolve_SeqSBAIJ_1()
1705 nz = ai[k + 1] - ai[k] - 1; in MatSolve_SeqSBAIJ_1()
1706 for (j = 0; j < nz; j++) t[k] += v[-j] * t[vj[-j]]; in MatSolve_SeqSBAIJ_1()
1707 x[rp[k]] = t[k]; in MatSolve_SeqSBAIJ_1()
1725 PetscInt nz, k; in MatSolve_SeqSBAIJ_1_inplace() local
1734 for (k = 0; k < mbs; k++) t[k] = b[rp[k]]; in MatSolve_SeqSBAIJ_1_inplace()
1735 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_1_inplace()
1736 v = aa + ai[k] + 1; in MatSolve_SeqSBAIJ_1_inplace()
1737 vj = aj + ai[k] + 1; in MatSolve_SeqSBAIJ_1_inplace()
1738 xk = t[k]; in MatSolve_SeqSBAIJ_1_inplace()
1739 nz = ai[k + 1] - ai[k] - 1; in MatSolve_SeqSBAIJ_1_inplace()
1741 t[k] = xk * aa[ai[k]]; /* aa[k] = 1/D(k) */ in MatSolve_SeqSBAIJ_1_inplace()
1745 for (k = mbs - 1; k >= 0; k--) { in MatSolve_SeqSBAIJ_1_inplace()
1746 v = aa + ai[k] + 1; in MatSolve_SeqSBAIJ_1_inplace()
1747 vj = aj + ai[k] + 1; in MatSolve_SeqSBAIJ_1_inplace()
1748 nz = ai[k + 1] - ai[k] - 1; in MatSolve_SeqSBAIJ_1_inplace()
1749 while (nz--) t[k] += (*v++) * t[*vj++]; in MatSolve_SeqSBAIJ_1_inplace()
1750 x[rp[k]] = t[k]; in MatSolve_SeqSBAIJ_1_inplace()
1769 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_1() local
1777 for (k = 0; k < mbs; k++) x[k] = b[rp[k]]; in MatForwardSolve_SeqSBAIJ_1()
1778 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_1()
1779 v = aa + ai[k]; in MatForwardSolve_SeqSBAIJ_1()
1780 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_1()
1781 xk = x[k]; in MatForwardSolve_SeqSBAIJ_1()
1782 nz = ai[k + 1] - ai[k] - 1; in MatForwardSolve_SeqSBAIJ_1()
1785 diagk = PetscRealPart(aa[adiag[k]]); /* note: aa[diag[k]] = 1/D(k) */ in MatForwardSolve_SeqSBAIJ_1()
1786 …PetscCheck(!PetscImaginaryPart(aa[adiag[k]]) && diagk >= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Diago… in MatForwardSolve_SeqSBAIJ_1()
1787 x[k] = xk * PetscSqrtReal(diagk); in MatForwardSolve_SeqSBAIJ_1()
1805 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_1_inplace() local
1813 for (k = 0; k < mbs; k++) x[k] = b[rp[k]]; in MatForwardSolve_SeqSBAIJ_1_inplace()
1814 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_1_inplace()
1815 v = aa + ai[k] + 1; in MatForwardSolve_SeqSBAIJ_1_inplace()
1816 vj = aj + ai[k] + 1; in MatForwardSolve_SeqSBAIJ_1_inplace()
1817 xk = x[k]; in MatForwardSolve_SeqSBAIJ_1_inplace()
1818 nz = ai[k + 1] - ai[k] - 1; in MatForwardSolve_SeqSBAIJ_1_inplace()
1821 diagk = PetscRealPart(aa[ai[k]]); /* note: aa[diag[k]] = 1/D(k) */ in MatForwardSolve_SeqSBAIJ_1_inplace()
1822 …PetscCheck(!PetscImaginaryPart(aa[ai[k]]) && diagk >= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Diagonal… in MatForwardSolve_SeqSBAIJ_1_inplace()
1823 x[k] = xk * PetscSqrtReal(diagk); in MatForwardSolve_SeqSBAIJ_1_inplace()
1841 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_1() local
1850 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_1()
1851 v = aa + ai[k]; in MatBackwardSolve_SeqSBAIJ_1()
1852 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_1()
1853 diagk = PetscRealPart(aa[adiag[k]]); in MatBackwardSolve_SeqSBAIJ_1()
1854 …PetscCheck(!PetscImaginaryPart(aa[adiag[k]]) && diagk >= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Diago… in MatBackwardSolve_SeqSBAIJ_1()
1855 t[k] = b[k] * PetscSqrtReal(diagk); in MatBackwardSolve_SeqSBAIJ_1()
1856 nz = ai[k + 1] - ai[k] - 1; in MatBackwardSolve_SeqSBAIJ_1()
1857 while (nz--) t[k] += (*v++) * t[*vj++]; in MatBackwardSolve_SeqSBAIJ_1()
1858 x[rp[k]] = t[k]; in MatBackwardSolve_SeqSBAIJ_1()
1876 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_1_inplace() local
1885 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_1_inplace()
1886 v = aa + ai[k] + 1; in MatBackwardSolve_SeqSBAIJ_1_inplace()
1887 vj = aj + ai[k] + 1; in MatBackwardSolve_SeqSBAIJ_1_inplace()
1888 diagk = PetscRealPart(aa[ai[k]]); in MatBackwardSolve_SeqSBAIJ_1_inplace()
1889 …PetscCheck(!PetscImaginaryPart(aa[ai[k]]) && diagk >= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Diagonal… in MatBackwardSolve_SeqSBAIJ_1_inplace()
1890 t[k] = b[k] * PetscSqrtReal(diagk); in MatBackwardSolve_SeqSBAIJ_1_inplace()
1891 nz = ai[k + 1] - ai[k] - 1; in MatBackwardSolve_SeqSBAIJ_1_inplace()
1892 while (nz--) t[k] += (*v++) * t[*vj++]; in MatBackwardSolve_SeqSBAIJ_1_inplace()
1893 x[rp[k]] = t[k]; in MatBackwardSolve_SeqSBAIJ_1_inplace()
1915 PetscInt nz, k, n, i, j; in MatSolves_SeqSBAIJ_1() local
1930 for (k = 0; k < mbs; k++) { in MatSolves_SeqSBAIJ_1()
1931 … for (i = 0; i < n; i++) t[n * k + i] = b[rp[k] + i * mbs]; /* values are stored interlaced in t */ in MatSolves_SeqSBAIJ_1()
1933 for (k = 0; k < mbs; k++) { in MatSolves_SeqSBAIJ_1()
1934 v = aa + ai[k]; in MatSolves_SeqSBAIJ_1()
1935 vj = aj + ai[k]; in MatSolves_SeqSBAIJ_1()
1936 nz = ai[k + 1] - ai[k] - 1; in MatSolves_SeqSBAIJ_1()
1938 for (i = 0; i < n; i++) t[n * (*vj) + i] += (*v) * t[n * k + i]; in MatSolves_SeqSBAIJ_1()
1942 for (i = 0; i < n; i++) t[n * k + i] *= aa[nz]; /* note: aa[nz] = 1/D(k) */ in MatSolves_SeqSBAIJ_1()
1946 for (k = mbs - 1; k >= 0; k--) { in MatSolves_SeqSBAIJ_1()
1947 v = aa + ai[k] - 1; in MatSolves_SeqSBAIJ_1()
1948 vj = aj + ai[k] - 1; in MatSolves_SeqSBAIJ_1()
1949 nz = ai[k + 1] - ai[k] - 1; in MatSolves_SeqSBAIJ_1()
1951 for (i = 0; i < n; i++) t[n * k + i] += (*v) * t[n * (*vj) + i]; in MatSolves_SeqSBAIJ_1()
1955 for (i = 0; i < n; i++) x[rp[k] + i * mbs] = t[n * k + i]; in MatSolves_SeqSBAIJ_1()
1979 PetscInt nz, k, n, i; in MatSolves_SeqSBAIJ_1_inplace() local
1994 for (k = 0; k < mbs; k++) { in MatSolves_SeqSBAIJ_1_inplace()
1995 … for (i = 0; i < n; i++) t[n * k + i] = b[rp[k] + i * mbs]; /* values are stored interlaced in t */ in MatSolves_SeqSBAIJ_1_inplace()
1997 for (k = 0; k < mbs; k++) { in MatSolves_SeqSBAIJ_1_inplace()
1998 v = aa + ai[k]; in MatSolves_SeqSBAIJ_1_inplace()
1999 vj = aj + ai[k]; in MatSolves_SeqSBAIJ_1_inplace()
2000 nz = ai[k + 1] - ai[k]; in MatSolves_SeqSBAIJ_1_inplace()
2002 for (i = 0; i < n; i++) t[n * (*vj) + i] += (*v) * t[n * k + i]; in MatSolves_SeqSBAIJ_1_inplace()
2006 for (i = 0; i < n; i++) t[n * k + i] *= aa[k]; /* note: aa[k] = 1/D(k) */ in MatSolves_SeqSBAIJ_1_inplace()
2010 for (k = mbs - 1; k >= 0; k--) { in MatSolves_SeqSBAIJ_1_inplace()
2011 v = aa + ai[k]; in MatSolves_SeqSBAIJ_1_inplace()
2012 vj = aj + ai[k]; in MatSolves_SeqSBAIJ_1_inplace()
2013 nz = ai[k + 1] - ai[k]; in MatSolves_SeqSBAIJ_1_inplace()
2015 for (i = 0; i < n; i++) t[n * k + i] += (*v) * t[n * (*vj) + i]; in MatSolves_SeqSBAIJ_1_inplace()
2019 for (i = 0; i < n; i++) x[rp[k] + i * mbs] = t[n * k + i]; in MatSolves_SeqSBAIJ_1_inplace()
2125 PetscInt nz, k; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace() local
2133 for (k = 0; k < mbs; k++) { in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2134 v = aa + ai[k] + 1; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2135 vj = aj + ai[k] + 1; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2136 xk = x[k]; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2137 nz = ai[k + 1] - ai[k] - 1; /* exclude diag[k] */ in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2139 x[k] = xk * aa[ai[k]]; /* note: aa[diag[k]] = 1/D(k) */ in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2143 for (k = mbs - 2; k >= 0; k--) { in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2144 v = aa + ai[k] + 1; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2145 vj = aj + ai[k] + 1; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2146 xk = x[k]; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2147 nz = ai[k + 1] - ai[k] - 1; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2149 x[k] = xk; in MatSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2166 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering() local
2173 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering()
2174 v = aa + ai[k]; in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering()
2175 vj = aj + ai[k]; in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering()
2176 nz = ai[k + 1] - ai[k] - 1; /* exclude diag[k] */ in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering()
2177 while (nz--) x[*vj++] += (*v++) * x[k]; in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering()
2178 diagk = PetscRealPart(aa[adiag[k]]); /* note: aa[adiag[k]] = 1/D(k) */ in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering()
2179 …k]]) && diagk >= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Diagonal (%g,%g) must be real and nonnegative… in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering()
2180 x[k] *= PetscSqrtReal(diagk); in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering()
2196 PetscInt nz, k; in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace() local
2203 for (k = 0; k < mbs; k++) { in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2204 v = aa + ai[k] + 1; in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2205 vj = aj + ai[k] + 1; in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2206 nz = ai[k + 1] - ai[k] - 1; /* exclude diag[k] */ in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2207 while (nz--) x[*vj++] += (*v++) * x[k]; in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2208 diagk = PetscRealPart(aa[ai[k]]); /* note: aa[diag[k]] = 1/D(k) */ in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2209 …k]]) && diagk >= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Diagonal (%g,%g) must be real and nonnegative… in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2210 x[k] *= PetscSqrtReal(diagk); in MatForwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2226 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering() local
2233 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering()
2234 v = aa + ai[k]; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering()
2235 vj = aj + ai[k]; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering()
2236 diagk = PetscRealPart(aa[adiag[k]]); /* note: aa[diag[k]] = 1/D(k) */ in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering()
2237 …PetscCheck(!PetscImaginaryPart(aa[adiag[k]]) && diagk >= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Diago… in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering()
2238 x[k] = PetscSqrtReal(diagk) * b[k]; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering()
2239 nz = ai[k + 1] - ai[k] - 1; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering()
2240 while (nz--) x[k] += (*v++) * x[*vj++]; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering()
2256 PetscInt nz, k; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace() local
2263 for (k = mbs - 1; k >= 0; k--) { in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2264 v = aa + ai[k] + 1; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2265 vj = aj + ai[k] + 1; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2266 diagk = PetscRealPart(aa[ai[k]]); /* note: aa[diag[k]] = 1/D(k) */ in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2267 …PetscCheck(!PetscImaginaryPart(aa[ai[k]]) && diagk >= 0, PETSC_COMM_SELF, PETSC_ERR_SUP, "Diagonal… in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2268 x[k] = PetscSqrtReal(diagk) * b[k]; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2269 nz = ai[k + 1] - ai[k] - 1; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2270 while (nz--) x[k] += (*v++) * x[*vj++]; in MatBackwardSolve_SeqSBAIJ_1_NaturalOrdering_inplace()
2285 PetscInt *prowl, *q, jmin, jmax, juidx, nzk, qm, *iu, *ju, k, j, vj, umax, maxadd; in MatICCFactorSymbolic_SeqSBAIJ_MSR() local
2319 for (k = 0; k < mbs; k++) { in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2321 q[k] = mbs; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2323 nz = ai[rip[k] + 1] - ai[rip[k]]; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2324 j = ai[rip[k]]; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2327 if (vj > k) { in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2328 qm = k; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2343 prow = k; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2346 while (prow < k) { in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2350 qm = k; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2372 i = q[k]; /* col value of first nonzero element in k_th row of U */ in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2373 prowl[k] = prowl[i]; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2374 prowl[i] = k; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2376 iu[k + 1] = iu[k] + nzk; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2379 if (iu[k + 1] > umax) { in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2384 if (maxadd < nzk) maxadd = (mbs - k) * (nzk + 1) / 2; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2389 PetscCall(PetscArraycpy(jutmp, ju, iu[k])); in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2394 PetscCall(PetscArraycpy(jutmp, lev, iu[k] - shift)); in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2401 i = k; in MatICCFactorSymbolic_SeqSBAIJ_MSR()
2477 PetscInt jmin, jmax, nzk, k, j, *jl, prow, *il, nextprow; in MatICCFactorSymbolic_SeqSBAIJ() local
2543 for (k = 0; k < am; k++) { /* for each active row k */ in MatICCFactorSymbolic_SeqSBAIJ()
2546 ncols = ai[k + 1] - ai[k]; in MatICCFactorSymbolic_SeqSBAIJ()
2547 …Check(ncols, PETSC_COMM_SELF, PETSC_ERR_MAT_CH_ZRPVT, "Empty row %" PetscInt_FMT " in matrix ", k); in MatICCFactorSymbolic_SeqSBAIJ()
2548 cols = aj + ai[k]; in MatICCFactorSymbolic_SeqSBAIJ()
2553 prow = jl[k]; /* 1st pivot row */ in MatICCFactorSymbolic_SeqSBAIJ()
2555 while (prow < k) { in MatICCFactorSymbolic_SeqSBAIJ()
2583 i = am - k + 1; /* num of unfactored rows */ in MatICCFactorSymbolic_SeqSBAIJ()
2591 …!= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "Empty row %" PetscInt_FMT " in ICC matrix factor", k); in MatICCFactorSymbolic_SeqSBAIJ()
2597 jl[k] = jl[i]; in MatICCFactorSymbolic_SeqSBAIJ()
2598 jl[i] = k; in MatICCFactorSymbolic_SeqSBAIJ()
2599 il[k] = ui[k] + 1; in MatICCFactorSymbolic_SeqSBAIJ()
2601 uj_ptr[k] = current_space->array; in MatICCFactorSymbolic_SeqSBAIJ()
2602 uj_lvl_ptr[k] = current_space_lvl->array; in MatICCFactorSymbolic_SeqSBAIJ()
2611 ui[k + 1] = ui[k] + nzk; in MatICCFactorSymbolic_SeqSBAIJ()
2679 PetscInt jmin, jmax, nzk, k, j, *jl, prow, *il, nextprow; in MatICCFactorSymbolic_SeqSBAIJ_inplace() local
2741 for (k = 0; k < am; k++) { /* for each active row k */ in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2744 ncols = ai[rip[k] + 1] - ai[rip[k]]; in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2745 cols = aj + ai[rip[k]]; in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2750 prow = jl[k]; /* 1st pivot row */ in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2752 while (prow < k) { in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2778 …i = am - k + 1; /* num of unfactored r… in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2791 jl[k] = jl[i]; in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2792 jl[i] = k; in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2793 il[k] = ui[k] + 1; in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2795 uj_ptr[k] = current_space->array; in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2796 uj_lvl_ptr[k] = current_space_lvl->array; in MatICCFactorSymbolic_SeqSBAIJ_inplace()
2805 ui[k + 1] = ui[k] + nzk; in MatICCFactorSymbolic_SeqSBAIJ_inplace()