Lines Matching refs:An
95 PetscInt An, i, k; in main() local
104 PetscCall(MatGetRowIJ(A, 0, PETSC_FALSE, PETSC_FALSE, &An, &Ai, &Aj, &done)); in main()
105 PetscCheck(done && An == m, PETSC_COMM_SELF, PETSC_ERR_ARG_SIZ, "Inconsistent sizes"); in main()
109 PetscCall(MatSetSizes(B, bs[j] * An, bs[j] * An, PETSC_DECIDE, PETSC_DECIDE)); in main()
110 PetscCall(PetscMalloc1(Ai[An] * bs[j] * bs[j], &val)); in main()
111 for (i = 0; i < Ai[An]; ++i) in main()
117 PetscCall(MatRestoreRowIJ(A, 0, PETSC_FALSE, PETSC_FALSE, &An, &Ai, &Aj, &done)); in main()
161 PetscInt An; in main() local
167 … PetscCall(MatGetRowIJ(A, 0, PETSC_FALSE, flg ? PETSC_FALSE : PETSC_TRUE, &An, &Ai, &Aj, &done)); in main()
169 PetscCall(PetscMalloc1(An + 1, &ia_ptr)); in main()
170 PetscCall(PetscMalloc1(Ai[An], &ja_ptr)); in main()
172 for (k = 0; k < An + 1; ++k) ia_ptr[k] = Ai[k]; in main()
173 for (k = 0; k < Ai[An]; ++k) ja_ptr[k] = Aj[k]; in main()
175 …PetscCallMKLSparse(mkl_sparse_d_create_csr, (&spr, SPARSE_INDEX_BASE_ZERO, An, An, ia_ptr, ia_ptr … in main()
179 …for (k = 0; k < An + 1; ++k) ia_ptr[k] = Ai[k] + 1; /* Fortran indexing to maximize cases covered … in main()
180 …for (k = 0; k < Ai[An]; ++k) ja_ptr[k] = Aj[k] + 1; /* Fortran indexing to maximize cases covered … in main()
182 …_d_create_bsr, (&spr, SPARSE_INDEX_BASE_ONE, SPARSE_LAYOUT_COLUMN_MAJOR, An, An, bs[j], ia_ptr, ia… in main()
186 …for (k = 0; k < An + 1; ++k) ia_ptr[k] = Ai[k] + 1; /* Fortran indexing to maximize cases covered … in main()
187 …for (k = 0; k < Ai[An]; ++k) ja_ptr[k] = Aj[k] + 1; /* Fortran indexing to maximize cases covered … in main()
189 …_d_create_bsr, (&spr, SPARSE_INDEX_BASE_ONE, SPARSE_LAYOUT_COLUMN_MAJOR, An, An, bs[j], ia_ptr, ia… in main()
199 …PetscCall(MatRestoreRowIJ(A, 0, PETSC_FALSE, flg ? PETSC_FALSE : PETSC_TRUE, &An, &Ai, &Aj, &done)… in main()