Lines Matching refs:baij
2315 Mat_SeqBAIJ *baij = (Mat_SeqBAIJ *)A->data; in MatZeroRows_SeqBAIJ() local
2317 PetscInt bs = A->rmap->bs, bs2 = baij->bs2, *sizes, row, bs_max; in MatZeroRows_SeqBAIJ()
2341 if (baij->keepnonzeropattern) { in MatZeroRows_SeqBAIJ()
2352 count = (baij->i[row / bs + 1] - baij->i[row / bs]) * bs; in MatZeroRows_SeqBAIJ()
2353 aa = baij->a + baij->i[row / bs] * bs2 + (row % bs); in MatZeroRows_SeqBAIJ()
2354 if (sizes[i] == bs && !baij->keepnonzeropattern) { in MatZeroRows_SeqBAIJ()
2356 if (baij->ilen[row / bs] > 0) { in MatZeroRows_SeqBAIJ()
2357 baij->ilen[row / bs] = 1; in MatZeroRows_SeqBAIJ()
2358 baij->j[baij->i[row / bs]] = row / bs; in MatZeroRows_SeqBAIJ()
2365 baij->ilen[row / bs] = 0; in MatZeroRows_SeqBAIJ()
2384 Mat_SeqBAIJ *baij = (Mat_SeqBAIJ *)A->data; in MatZeroRowsColumns_SeqBAIJ() local
2386 PetscInt bs = A->rmap->bs, bs2 = baij->bs2, row, col; in MatZeroRowsColumns_SeqBAIJ()
2410 for (j = baij->i[row]; j < baij->i[row + 1]; j++) { in MatZeroRowsColumns_SeqBAIJ()
2412 col = bs * baij->j[j] + k; in MatZeroRowsColumns_SeqBAIJ()
2414 aa = baij->a + j * bs2 + (i % bs) + bs * k; in MatZeroRowsColumns_SeqBAIJ()
2431 count = (baij->i[row / bs + 1] - baij->i[row / bs]) * bs; in MatZeroRowsColumns_SeqBAIJ()
2432 aa = baij->a + baij->i[row / bs] * bs2 + (row % bs); in MatZeroRowsColumns_SeqBAIJ()
2560 Mat_SeqBAIJ *baij = (Mat_SeqBAIJ *)mat->data; in MatSeqBAIJSetColumnIndices_SeqBAIJ() local
2563 baij->nz = baij->maxnz; in MatSeqBAIJSetColumnIndices_SeqBAIJ()
2564 PetscCall(PetscArraycpy(baij->j, indices, baij->nz)); in MatSeqBAIJSetColumnIndices_SeqBAIJ()
2565 PetscCall(PetscArraycpy(baij->ilen, baij->imax, baij->mbs)); in MatSeqBAIJSetColumnIndices_SeqBAIJ()
3916 Mat_SeqBAIJ *baij; in MatCreateSeqBAIJWithArrays() local
3926 baij = (Mat_SeqBAIJ *)(*mat)->data; in MatCreateSeqBAIJWithArrays()
3927 PetscCall(PetscMalloc2(m, &baij->imax, m, &baij->ilen)); in MatCreateSeqBAIJWithArrays()
3929 baij->i = i; in MatCreateSeqBAIJWithArrays()
3930 baij->j = j; in MatCreateSeqBAIJWithArrays()
3931 baij->a = a; in MatCreateSeqBAIJWithArrays()
3933 …baij->nonew = -1; /*this indicates that inserting a new value in the matrix that generate… in MatCreateSeqBAIJWithArrays()
3934 baij->free_a = PETSC_FALSE; in MatCreateSeqBAIJWithArrays()
3935 baij->free_ij = PETSC_FALSE; in MatCreateSeqBAIJWithArrays()
3936 baij->free_imax_ilen = PETSC_TRUE; in MatCreateSeqBAIJWithArrays()
3941 baij->ilen[ii] = baij->imax[ii] = row_len; in MatCreateSeqBAIJWithArrays()
3945 for (PetscInt ii = 0; ii < baij->i[m]; ii++) { in MatCreateSeqBAIJWithArrays()