Lines Matching refs:S

86 PetscErrorCode MatKAIJGetS(Mat A, PetscInt *m, PetscInt *n, PetscScalar *S[])  in MatKAIJGetS()  argument
93 if (S) *S = b->S; in MatKAIJGetS()
117 PetscErrorCode MatKAIJGetSRead(Mat A, PetscInt *m, PetscInt *n, const PetscScalar *S[]) in MatKAIJGetSRead() argument
124 if (S) *S = b->S; in MatKAIJGetSRead()
145 PetscErrorCode MatKAIJRestoreS(Mat A, PetscScalar *S[]) in MatKAIJRestoreS() argument
148 if (S) *S = NULL; in MatKAIJRestoreS()
170 PetscErrorCode MatKAIJRestoreSRead(Mat A, const PetscScalar *S[]) in MatKAIJRestoreSRead() argument
173 if (S) *S = NULL; in MatKAIJRestoreSRead()
346 PetscErrorCode MatKAIJSetS(Mat A, PetscInt p, PetscInt q, const PetscScalar S[]) in MatKAIJSetS() argument
351 PetscCall(PetscFree(a->S)); in MatKAIJSetS()
352 if (S) { in MatKAIJSetS()
353 PetscCall(PetscMalloc1(p * q, &a->S)); in MatKAIJSetS()
354 PetscCall(PetscArraycpy(a->S, S, p * q)); in MatKAIJSetS()
355 } else a->S = NULL; in MatKAIJSetS()
387 if (!a->isTI || a->S) { in MatKAIJGetScaledIdentity()
391 if (a->S && PetscAbsScalar(a->S[i + j * a->p]) > PETSC_SMALL) *identity = PETSC_FALSE; in MatKAIJGetScaledIdentity()
394 … if (a->S && PetscAbsScalar(a->S[i * (a->p + 1)] - a->S[0]) > PETSC_SMALL) *identity = PETSC_FALSE; in MatKAIJGetScaledIdentity()
464 PetscCall(PetscFree(b->S)); in MatDestroy_SeqKAIJ()
505 PetscCall(MatCreateKAIJ(mpiaij->A, a->p, a->q, a->S, T, &a->AIJ)); in MatKAIJ_build_AIJ_OAIJ()
583 if (!a->S) { in MatView_KAIJ()
589 …ntf(viewer, "%18.16e %18.16e ", (double)PetscRealPart(a->S[i]), (double)PetscImaginaryPart(a->S[i]… in MatView_KAIJ()
591 PetscCall(PetscViewerASCIIPrintf(viewer, "%18.16e ", (double)PetscRealPart(a->S[i]))); in MatView_KAIJ()
642 PetscCall(PetscFree(b->S)); in MatDestroy_MPIKAIJ()
656 const PetscScalar *s = b->S, *t = b->T; in MatMultAdd_SeqKAIJ()
734 const PetscScalar *S = b->S; in MatInvertBlockDiagonal_SeqKAIJ() local
743 …PetscCheck(S || T || b->isTI, PetscObjectComm((PetscObject)A), PETSC_ERR_SUP, "MATKAIJ: Cannot inv… in MatInvertBlockDiagonal_SeqKAIJ()
758 if (S) { in MatInvertBlockDiagonal_SeqKAIJ()
759 PetscCall(PetscArraycpy(diag, S, dof2)); in MatInvertBlockDiagonal_SeqKAIJ()
821 if (diagDense || !a->S) d = m; in MatConvert_KAIJ_AIJ()
1184 PetscScalar *vaij, *v, *S = b->S, *T = b->T; in MatGetRow_SeqKAIJ() local
1191 if ((!S) && (!T) && (!b->isTI)) { in MatGetRow_SeqKAIJ()
1212 if (S) nz += q; in MatGetRow_SeqKAIJ()
1213 if (T || b->isTI) nz += (diag && S ? (nzaij - 1) * q : nzaij * q); in MatGetRow_SeqKAIJ()
1236 if (S) { in MatGetRow_SeqKAIJ()
1239 v[c * q + j] += S[s + j * p]; in MatGetRow_SeqKAIJ()
1266 PetscScalar *v, *vals, *ovals, *S = b->S, *T = b->T; in MatGetRow_MPIKAIJ() local
1277 if ((!S) && (!T) && (!b->isTI)) { in MatGetRow_MPIKAIJ()
1304 if (S) nz += q; in MatGetRow_MPIKAIJ()
1305 if (T || b->isTI) nz += (diag && S ? (ncolsaij + ncolsoaij - 1) * q : (ncolsaij + ncolsoaij) * q); in MatGetRow_MPIKAIJ()
1340 if (S) { in MatGetRow_MPIKAIJ()
1343 v[c * q + j] += S[s + j * p]; in MatGetRow_MPIKAIJ()
1419 PetscErrorCode MatCreateKAIJ(Mat A, PetscInt p, PetscInt q, const PetscScalar S[], const PetscScala… in MatCreateKAIJ() argument
1425 PetscCall(MatKAIJSetS(*kaij, p, q, S)); in MatCreateKAIJ()