Lines Matching refs:colmap

18   PetscCall(PetscHMapIDestroy(&baij->colmap));  in MatDestroy_MPIBAIJ()
20 PetscCall(PetscFree(baij->colmap)); in MatDestroy_MPIBAIJ()
151 PetscCall(PetscHMapICreateWithSize(baij->nbs, &baij->colmap)); in MatCreateColmap_MPIBAIJ_Private()
152 for (i = 0; i < nbs; i++) PetscCall(PetscHMapISet(baij->colmap, baij->garray[i] + 1, i * bs + 1)); in MatCreateColmap_MPIBAIJ_Private()
154 PetscCall(PetscCalloc1(baij->Nbs + 1, &baij->colmap)); in MatCreateColmap_MPIBAIJ_Private()
155 for (i = 0; i < nbs; i++) baij->colmap[baij->garray[i]] = i * bs + 1; in MatCreateColmap_MPIBAIJ_Private()
282 if (!baij->colmap) PetscCall(MatCreateColmap_MPIBAIJ_Private(mat)); in MatSetValues_MPIBAIJ()
284 PetscCall(PetscHMapIGetWithDefault(baij->colmap, in[j] / bs + 1, 0, &col)); in MatSetValues_MPIBAIJ()
287 col = baij->colmap[in[j] / bs] - 1; in MatSetValues_MPIBAIJ()
458 if (!baij->colmap) PetscCall(MatCreateColmap_MPIBAIJ_Private(mat)); in MatSetValuesBlocked_MPIBAIJ()
461 PetscCall(PetscHMapIGetWithDefault(baij->colmap, in[j] + 1, 0, &col)); in MatSetValuesBlocked_MPIBAIJ()
464 col = baij->colmap[in[j]] < 1 ? -1 : (baij->colmap[in[j]] - 1) / bs; in MatSetValuesBlocked_MPIBAIJ()
671 if (!baij->colmap) PetscCall(MatCreateColmap_MPIBAIJ_Private(mat)); in MatGetValues_MPIBAIJ()
673 PetscCall(PetscHMapIGetWithDefault(baij->colmap, idxn[j] / bs + 1, 0, &data)); in MatGetValues_MPIBAIJ()
676 data = baij->colmap[idxn[j] / bs] - 1; in MatGetValues_MPIBAIJ()
2690 PetscCall(PetscHMapIDestroy(&b->colmap)); in MatMPIBAIJSetPreallocation_MPIBAIJ()
2692 PetscCall(PetscFree(b->colmap)); in MatMPIBAIJSetPreallocation_MPIBAIJ()
2867 b->colmap = NULL; in MatCreate_MPIBAIJ()
3200 if (oldmat->colmap) { in MatDuplicate_MPIBAIJ()
3202 PetscCall(PetscHMapIDuplicate(oldmat->colmap, &a->colmap)); in MatDuplicate_MPIBAIJ()
3204 PetscCall(PetscMalloc1(a->Nbs, &a->colmap)); in MatDuplicate_MPIBAIJ()
3205 PetscCall(PetscArraycpy(a->colmap, oldmat->colmap, a->Nbs)); in MatDuplicate_MPIBAIJ()
3207 } else a->colmap = NULL; in MatDuplicate_MPIBAIJ()
3370 PetscErrorCode MatMPIBAIJGetSeqBAIJ(Mat A, Mat *Ad, Mat *Ao, const PetscInt *colmap[]) in MatMPIBAIJGetSeqBAIJ() argument
3380 if (colmap) *colmap = a->garray; in MatMPIBAIJGetSeqBAIJ()
3480 if (!baij->colmap) PetscCall(MatCreateColmap_MPIBAIJ_Private(mat)); in matmpibaijsetvaluesblocked_()
3486 PetscCall(PetscHMapIGetWithDefault(baij->colmap, in[j] + 1, 0, &data)); in matmpibaijsetvaluesblocked_()
3490 …PetscCheck((baij->colmap[in[j]] - 1) % bs == 0, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Incorrect colmap… in matmpibaijsetvaluesblocked_()
3494 PetscCall(PetscHMapIGetWithDefault(baij->colmap, in[j] + 1, 0, &col)); in matmpibaijsetvaluesblocked_()
3497 col = (baij->colmap[in[j]] - 1) / bs; in matmpibaijsetvaluesblocked_()