Lines Matching refs:kaij

785   Mat_MPIKAIJ *kaij = (Mat_MPIKAIJ *)A->data;  in MatGetDiagonalBlock_MPIKAIJ()  local
788 *B = kaij->AIJ; in MatGetDiagonalBlock_MPIKAIJ()
857 Mat_SeqKAIJ *kaij = (Mat_SeqKAIJ *)A->data; in MatSOR_SeqKAIJ() local
858 Mat_SeqAIJ *a = (Mat_SeqAIJ *)kaij->AIJ->data; in MatSOR_SeqKAIJ()
859 const PetscScalar *aa = a->a, *T = kaij->T, *v; in MatSOR_SeqKAIJ()
860 …const PetscInt m = kaij->AIJ->rmap->n, *ai = a->i, *aj = a->j, p = kaij->p, q = kaij->q, *diag… in MatSOR_SeqKAIJ()
877 if (!kaij->ibdiagvalid) PetscCall(MatInvertBlockDiagonal_SeqKAIJ(A, NULL)); in MatSOR_SeqKAIJ()
878 idiag = kaij->ibdiag; in MatSOR_SeqKAIJ()
879 PetscCall(MatGetDiagonalMarkers_SeqAIJ(kaij->AIJ, &diag, NULL)); in MatSOR_SeqKAIJ()
881 if (!kaij->sor.setup) { in MatSOR_SeqKAIJ()
882 …PetscCall(PetscMalloc5(bs, &kaij->sor.w, bs, &kaij->sor.y, m * bs, &kaij->sor.work, m * bs, &kaij-… in MatSOR_SeqKAIJ()
883 kaij->sor.setup = PETSC_TRUE; in MatSOR_SeqKAIJ()
885 y = kaij->sor.y; in MatSOR_SeqKAIJ()
886 w = kaij->sor.w; in MatSOR_SeqKAIJ()
887 work = kaij->sor.work; in MatSOR_SeqKAIJ()
888 t = kaij->sor.t; in MatSOR_SeqKAIJ()
889 arr = kaij->sor.arr; in MatSOR_SeqKAIJ()
912 } else if (kaij->isTI) { in MatSOR_SeqKAIJ()
932 idiag = kaij->ibdiag + bs2 * (m - 1); in MatSOR_SeqKAIJ()
958 } else if (kaij->isTI) { in MatSOR_SeqKAIJ()
978 idiag = kaij->ibdiag; in MatSOR_SeqKAIJ()
998 } else if (kaij->isTI) { in MatSOR_SeqKAIJ()
1024 } else if (kaij->isTI) { in MatSOR_SeqKAIJ()
1042 idiag = kaij->ibdiag + bs2 * (m - 1); in MatSOR_SeqKAIJ()
1064 } else if (kaij->isTI) { in MatSOR_SeqKAIJ()
1089 } else if (kaij->isTI) { in MatSOR_SeqKAIJ()
1120 } else if (kaij->isTI) { in MatSOR_SeqKAIJ()
1419 …tCreateKAIJ(Mat A, PetscInt p, PetscInt q, const PetscScalar S[], const PetscScalar T[], Mat *kaij) in MatCreateKAIJ() argument
1422 PetscCall(MatCreate(PetscObjectComm((PetscObject)A), kaij)); in MatCreateKAIJ()
1423 PetscCall(MatSetType(*kaij, MATKAIJ)); in MatCreateKAIJ()
1424 PetscCall(MatKAIJSetAIJ(*kaij, A)); in MatCreateKAIJ()
1425 PetscCall(MatKAIJSetS(*kaij, p, q, S)); in MatCreateKAIJ()
1426 PetscCall(MatKAIJSetT(*kaij, p, q, T)); in MatCreateKAIJ()
1427 PetscCall(MatSetUp(*kaij)); in MatCreateKAIJ()