Lines Matching refs:aa
88 const MatScalar *aa; in MatFindZeroDiagonals_SeqAIJ_Private() local
94 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatFindZeroDiagonals_SeqAIJ_Private()
97 if ((diag[i] >= ii[i + 1]) || (jj[diag[i]] != i) || (aa[diag[i]] == 0.0)) cnt++; in MatFindZeroDiagonals_SeqAIJ_Private()
102 if ((diag[i] >= ii[i + 1]) || (jj[diag[i]] != i) || (aa[diag[i]] == 0.0)) rows[cnt++] = i; in MatFindZeroDiagonals_SeqAIJ_Private()
106 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatFindZeroDiagonals_SeqAIJ_Private()
124 const MatScalar *aa; in MatFindNonzeroRows_SeqAIJ() local
130 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatFindNonzeroRows_SeqAIJ()
140 if (aa[j] != 0.0) goto ok1; in MatFindNonzeroRows_SeqAIJ()
146 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatFindNonzeroRows_SeqAIJ()
155 if (aa[j] != 0.0) { in MatFindNonzeroRows_SeqAIJ()
161 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatFindNonzeroRows_SeqAIJ()
170 MatScalar *aa; in MatDiagonalSet_SeqAIJ() local
179 PetscCall(MatSeqAIJGetArray(Y, &aa)); in MatDiagonalSet_SeqAIJ()
181 for (i = 0; i < m; i++) aa[diag[i]] = v[i]; in MatDiagonalSet_SeqAIJ()
183 for (i = 0; i < m; i++) aa[diag[i]] += v[i]; in MatDiagonalSet_SeqAIJ()
185 PetscCall(MatSeqAIJRestoreArray(Y, &aa)); in MatDiagonalSet_SeqAIJ()
339 PetscScalar *aa; in MatSetValuesRow_SeqAIJ() local
342 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatSetValuesRow_SeqAIJ()
343 PetscCall(PetscArraycpy(aa + ai[row], v, ai[row + 1] - ai[row])); in MatSetValuesRow_SeqAIJ()
344 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatSetValuesRow_SeqAIJ()
367 MatScalar *ap, value, *aa; in MatSeqAIJSetValuesLocalFast() local
371 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatSeqAIJSetValuesLocalFast()
374 ap = aa + ai[row]; in MatSeqAIJSetValuesLocalFast()
398 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatSeqAIJSetValuesLocalFast()
408 MatScalar *ap = NULL, value = 0.0, *aa; in MatSetValues_SeqAIJ() local
413 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatSetValues_SeqAIJ()
419 if (!A->structure_only) ap = PetscSafePointerPlusOffset(aa, ai[row]); in MatSetValues_SeqAIJ()
458 …MatSeqXAIJReallocateAIJ(A, A->rmap->n, 1, nrow, row, col, rmax, aa, ai, aj, rp, ap, imax, nonew, M… in MatSetValues_SeqAIJ()
475 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatSetValues_SeqAIJ()
485 MatScalar *aa, *ap; in MatSetValues_SeqAIJ_SortedFullNoPreallocation() local
491 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatSetValues_SeqAIJ_SortedFullNoPreallocation()
495 ap = PetscSafePointerPlusOffset(aa, ai[row]); in MatSetValues_SeqAIJ_SortedFullNoPreallocation()
511 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatSetValues_SeqAIJ_SortedFullNoPreallocation()
568 MatScalar *aa, *ap; in MatSetValues_SeqAIJ_SortedFull() local
571 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatSetValues_SeqAIJ_SortedFull()
576 ap = aa + ai[row]; in MatSetValues_SeqAIJ_SortedFull()
589 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatSetValues_SeqAIJ_SortedFull()
598 const MatScalar *ap, *aa; in MatGetValues_SeqAIJ() local
601 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatGetValues_SeqAIJ()
610 ap = PetscSafePointerPlusOffset(aa, ai[row]); in MatGetValues_SeqAIJ()
637 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatGetValues_SeqAIJ()
976 const PetscScalar *aa; in MatView_SeqAIJ_Draw_Zoom() local
984 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatView_SeqAIJ_Draw_Zoom()
995 if (PetscRealPart(aa[j]) >= 0.) continue; in MatView_SeqAIJ_Draw_Zoom()
1006 if (aa[j] != 0.) continue; in MatView_SeqAIJ_Draw_Zoom()
1017 if (PetscRealPart(aa[j]) <= 0.) continue; in MatView_SeqAIJ_Draw_Zoom()
1030 if (PetscAbsScalar(aa[i]) > maxv) maxv = PetscAbsScalar(aa[i]); in MatView_SeqAIJ_Draw_Zoom()
1043 color = PetscDrawRealToColor(PetscAbsScalar(aa[count]), minv, maxv); in MatView_SeqAIJ_Draw_Zoom()
1050 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatView_SeqAIJ_Draw_Zoom()
1102 MatScalar *aa = a->a, *ap; in MatAssemblyEnd_SeqAIJ() local
1120 ap = aa + ai[i]; in MatAssemblyEnd_SeqAIJ()
1162 MatScalar *aa; in MatRealPart_SeqAIJ() local
1165 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatRealPart_SeqAIJ()
1166 for (i = 0; i < nz; i++) aa[i] = PetscRealPart(aa[i]); in MatRealPart_SeqAIJ()
1167 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatRealPart_SeqAIJ()
1175 MatScalar *aa; in MatImaginaryPart_SeqAIJ() local
1178 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatImaginaryPart_SeqAIJ()
1179 for (i = 0; i < nz; i++) aa[i] = PetscImaginaryPart(aa[i]); in MatImaginaryPart_SeqAIJ()
1180 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatImaginaryPart_SeqAIJ()
1187 MatScalar *aa; in MatZeroEntries_SeqAIJ() local
1190 PetscCall(MatSeqAIJGetArrayWrite(A, &aa)); in MatZeroEntries_SeqAIJ()
1191 PetscCall(PetscArrayzero(aa, a->i[A->rmap->n])); in MatZeroEntries_SeqAIJ()
1192 PetscCall(MatSeqAIJRestoreArrayWrite(A, &aa)); in MatZeroEntries_SeqAIJ()
1364 const PetscScalar *aa; in MatGetDiagonal_SeqAIJ() local
1371 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatGetDiagonal_SeqAIJ()
1375 for (PetscInt i = 0; i < n; i++) x[i] = 1.0 / aa[diag[i]]; in MatGetDiagonal_SeqAIJ()
1377 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatGetDiagonal_SeqAIJ()
1385 for (PetscInt i = 0; i < n; i++) x[i] = aa[diag[i]]; in MatGetDiagonal_SeqAIJ()
1387 for (PetscInt i = 0; i < n; i++) x[i] = (diag[i] == ai[i + 1]) ? 0.0 : aa[diag[i]]; in MatGetDiagonal_SeqAIJ()
1390 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatGetDiagonal_SeqAIJ()
1398 const MatScalar *aa; in MatMultTransposeAdd_SeqAIJ() local
1415 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatMultTransposeAdd_SeqAIJ()
1418 fortranmulttransposeaddaij_(&m, x, a->i, a->j, aa, y); in MatMultTransposeAdd_SeqAIJ()
1429 v = aa + ii[i]; in MatMultTransposeAdd_SeqAIJ()
1442 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatMultTransposeAdd_SeqAIJ()
1467 #pragma disjoint(*x, *y, *aa) in MatMult_SeqAIJ()
1488 const PetscScalar *aa = a_a + ii[i]; in MatMult_SeqAIJ() local
1490 PetscSparseDensePlusDot(sum, x, aa, aj, n); in MatMult_SeqAIJ()
1502 const PetscScalar *aa = a_a + ii[i]; in MatMult_SeqAIJ()
1504 PetscSparseDensePlusDot(sum, x, aa, aj, n); in MatMult_SeqAIJ()
1522 const MatScalar *aa, *a_a; in MatMultMax_SeqAIJ() local
1530 #pragma disjoint(*x, *y, *aa) in MatMultMax_SeqAIJ()
1544 aa = a_a + ii[i]; in MatMultMax_SeqAIJ()
1547 PetscSparseDenseMaxDot(sum, x, aa, aj, n); in MatMultMax_SeqAIJ()
1556 aa = a_a + ii[i]; in MatMultMax_SeqAIJ()
1559 PetscSparseDenseMaxDot(sum, x, aa, aj, n); in MatMultMax_SeqAIJ()
1576 const MatScalar *aa, *a_a; in MatMultAddMax_SeqAIJ() local
1594 aa = a_a + ii[i]; in MatMultAddMax_SeqAIJ()
1596 PetscSparseDenseMaxDot(sum, x, aa, aj, n); in MatMultAddMax_SeqAIJ()
1604 aa = a_a + ii[i]; in MatMultAddMax_SeqAIJ()
1606 PetscSparseDenseMaxDot(sum, x, aa, aj, n); in MatMultAddMax_SeqAIJ()
1644 const PetscScalar *aa = a_a + ii[i]; in MatMultAdd_SeqAIJ() local
1646 PetscSparseDensePlusDot(sum, x, aa, aj, n); in MatMultAdd_SeqAIJ()
1658 const PetscScalar *aa = a_a + ii[i]; in MatMultAdd_SeqAIJ() local
1660 PetscSparseDensePlusDot(sum, x, aa, aj, n); in MatMultAdd_SeqAIJ()
1858 const MatScalar *v, *idiag = NULL, *mdiag, *aa; in MatSOR_SeqAIJ() local
1875 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatSOR_SeqAIJ()
1886 v = aa + diag[i] + 1; in MatSOR_SeqAIJ()
1893 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatSOR_SeqAIJ()
1913 v = aa + diag[i] + 1; in MatSOR_SeqAIJ()
1920 v = aa; in MatSOR_SeqAIJ()
1929 v = aa + a->i[i]; in MatSOR_SeqAIJ()
1947 v = aa + a->i[i]; in MatSOR_SeqAIJ()
1960 v = aa + diag[i] + 1; in MatSOR_SeqAIJ()
1979 v = aa + a->i[i]; in MatSOR_SeqAIJ()
1986 v = aa + diag[i] + 1; in MatSOR_SeqAIJ()
2000 v = aa + a->i[i]; in MatSOR_SeqAIJ()
2006 v = aa + diag[i] + 1; in MatSOR_SeqAIJ()
2018 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatSOR_SeqAIJ()
2053 PetscScalar *bb, *aa; in MatZeroRows_SeqAIJ() local
2071 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatZeroRows_SeqAIJ()
2075 PetscCall(PetscArrayzero(&aa[a->i[rows[i]]], a->ilen[rows[i]])); in MatZeroRows_SeqAIJ()
2082 aa[diag[d]] = diagv; in MatZeroRows_SeqAIJ()
2094 aa[a->i[rows[i]]] = diagv; in MatZeroRows_SeqAIJ()
2109 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatZeroRows_SeqAIJ()
2120 PetscScalar *bb, *aa; in MatZeroRowsColumns_SeqAIJ() local
2127 PetscCall(MatSeqAIJGetArray(A, &aa)); in MatZeroRowsColumns_SeqAIJ()
2136 PetscCall(PetscArrayzero(PetscSafePointerPlusOffset(aa, a->i[rows[i]]), a->ilen[rows[i]])); in MatZeroRowsColumns_SeqAIJ()
2144 if (vecs) bb[i] -= aa[j] * xx[a->j[j]]; in MatZeroRowsColumns_SeqAIJ()
2145 aa[j] = 0.0; in MatZeroRowsColumns_SeqAIJ()
2163 for (i = 0; i < N; i++) aa[diag[rows[i]]] = diagv; in MatZeroRowsColumns_SeqAIJ()
2166 PetscCall(MatSeqAIJRestoreArray(A, &aa)); in MatZeroRowsColumns_SeqAIJ()
2174 const PetscScalar *aa; in MatGetRow_SeqAIJ() local
2177 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatGetRow_SeqAIJ()
2179 if (v) *v = PetscSafePointerPlusOffset((PetscScalar *)aa, a->i[row]); in MatGetRow_SeqAIJ()
2184 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatGetRow_SeqAIJ()
2394 const PetscScalar *aa; in MatCreateSubMatrix_SeqAIJ() local
2457 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatCreateSubMatrix_SeqAIJ()
2463 PetscCall(PetscArraycpy(a_new, aa + starts[i], lensi)); in MatCreateSubMatrix_SeqAIJ()
2470 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatCreateSubMatrix_SeqAIJ()
2510 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatCreateSubMatrix_SeqAIJ()
2525 *mat_a++ = aa[k]; in MatCreateSubMatrix_SeqAIJ()
2530 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatCreateSubMatrix_SeqAIJ()
2871 const PetscScalar *aa; in MatCopy_SeqAIJ() local
2874 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatCopy_SeqAIJ()
2878 PetscCall(PetscArraycpy(bb, aa, a->i[A->rmap->n])); in MatCopy_SeqAIJ()
2880 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatCopy_SeqAIJ()
3009 const MatScalar *aa, *av; in MatGetRowMaxAbs_SeqAIJ() local
3014 aa = av; in MatGetRowMaxAbs_SeqAIJ()
3026 atmp = PetscAbsScalar(*aa); in MatGetRowMaxAbs_SeqAIJ()
3031 aa++; in MatGetRowMaxAbs_SeqAIJ()
3045 const MatScalar *aa, *av; in MatGetRowSumAbs_SeqAIJ() local
3050 aa = av; in MatGetRowSumAbs_SeqAIJ()
3061 x[i] += PetscAbsScalar(*aa); in MatGetRowSumAbs_SeqAIJ()
3062 aa++; in MatGetRowSumAbs_SeqAIJ()
3075 const MatScalar *aa, *av; in MatGetRowMax_SeqAIJ() local
3080 aa = av; in MatGetRowMax_SeqAIJ()
3091 x[i] = *aa; in MatGetRowMax_SeqAIJ()
3107 if (PetscRealPart(x[i]) < PetscRealPart(*aa)) { in MatGetRowMax_SeqAIJ()
3108 x[i] = *aa; in MatGetRowMax_SeqAIJ()
3111 aa++; in MatGetRowMax_SeqAIJ()
3125 const MatScalar *aa, *av; in MatGetRowMinAbs_SeqAIJ() local
3129 aa = av; in MatGetRowMinAbs_SeqAIJ()
3140 x[i] = *aa; in MatGetRowMinAbs_SeqAIJ()
3156 if (PetscAbsScalar(x[i]) > PetscAbsScalar(*aa)) { in MatGetRowMinAbs_SeqAIJ()
3157 x[i] = *aa; in MatGetRowMinAbs_SeqAIJ()
3160 aa++; in MatGetRowMinAbs_SeqAIJ()
3175 const MatScalar *aa, *av; in MatGetRowMin_SeqAIJ() local
3180 aa = av; in MatGetRowMin_SeqAIJ()
3191 x[i] = *aa; in MatGetRowMin_SeqAIJ()
3207 if (PetscRealPart(x[i]) > PetscRealPart(*aa)) { in MatGetRowMin_SeqAIJ()
3208 x[i] = *aa; in MatGetRowMin_SeqAIJ()
3211 aa++; in MatGetRowMin_SeqAIJ()
3352 PetscScalar a, *aa; in MatSetRandom_SeqAIJ() local
3366 PetscCall(MatSeqAIJGetArrayWrite(x, &aa)); in MatSetRandom_SeqAIJ()
3367 for (i = 0; i < aij->nz; i++) PetscCall(PetscRandomGetValue(rctx, aa + i)); in MatSetRandom_SeqAIJ()
3368 PetscCall(MatSeqAIJRestoreArrayWrite(x, &aa)); in MatSetRandom_SeqAIJ()
4115 const PetscScalar *aa, *ba; in MatSeqAIJKron_SeqAIJ() local
4147 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatSeqAIJKron_SeqAIJ()
4152 for (q = b->i[p]; q < b->i[p + 1]; ++q) v[nnz++] = aa[n] * ba[q]; in MatSeqAIJKron_SeqAIJ()
4157 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatSeqAIJKron_SeqAIJ()
4908 const PetscScalar *aa; in MatDuplicateNoCreate_SeqAIJ() local
4910 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatDuplicateNoCreate_SeqAIJ()
4911 PetscCall(PetscArraycpy(c->a, aa, a->i[m])); in MatDuplicateNoCreate_SeqAIJ()
4912 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatDuplicateNoCreate_SeqAIJ()
5057 const PetscScalar *aa, *ba; in MatEqual_SeqAIJ() local
5074 PetscCall(MatSeqAIJGetArrayRead(A, &aa)); in MatEqual_SeqAIJ()
5077 PetscCall(PetscArraycmp(aa, ba, a->nz, flg)); in MatEqual_SeqAIJ()
5078 PetscCall(MatSeqAIJRestoreArrayRead(A, &aa)); in MatEqual_SeqAIJ()
5314 MatScalar *aa = a->a; in MatEliminateZeros_SeqAIJ() local
5324 if (aa[k] == 0 && (aj[k] != i - 1 || !keep)) fshift++; in MatEliminateZeros_SeqAIJ()
5326 …if (aa[k] == 0 && aj[k] == i - 1) PetscCall(PetscInfo(A, "Keep the diagonal zero at row %" PetscIn… in MatEliminateZeros_SeqAIJ()
5327 aa[k - fshift] = aa[k]; in MatEliminateZeros_SeqAIJ()
5497 MatScalar *ap, value, *aa; in matsetvaluesseqaij_() local
5507 aa = a->a; in matsetvaluesseqaij_()
5514 ap = aa + ai[row]; in matsetvaluesseqaij_()
5547 …MatSeqXAIJReallocateAIJ(A, A->rmap->n, 1, nrow, row, col, rmax, aa, ai, aj, rp, ap, imax, nonew, M… in matsetvaluesseqaij_()