Lines Matching refs:bs2
131 PetscInt bs2, oldnmax; in MatStashScatterEnd_Ref() local
147 bs2 = stash->bs * stash->bs; in MatStashScatterEnd_Ref()
148 oldnmax = ((int)(stash->n * 1.1) + 5) * bs2; in MatStashScatterEnd_Ref()
182 PetscInt bs2 = stash->bs * stash->bs; in MatStashGetInfo_Private() local
185 if (nstash) *nstash = stash->n * bs2; in MatStashGetInfo_Private()
221 PetscInt newnmax, bs2 = stash->bs * stash->bs; in MatStashExpand_Private() local
227 if (stash->umax) cnewnmax = stash->umax / bs2; in MatStashExpand_Private()
228 else cnewnmax = DEFAULT_STASH_SIZE / bs2; in MatStashExpand_Private()
230 if (stash->umax > stash->oldnmax) cnewnmax = stash->umax / bs2; in MatStashExpand_Private()
231 else cnewnmax = stash->oldnmax / bs2; in MatStashExpand_Private()
237 PetscCall(PetscMatStashSpaceGet(bs2, newnmax, &stash->space)); in MatStashExpand_Private()
340 PetscInt i, j, k, bs2, bs = stash->bs, l; in MatStashValuesRowBlocked_Private() local
349 bs2 = bs * bs; in MatStashValuesRowBlocked_Private()
356 array = space->val + bs2 * l; in MatStashValuesRowBlocked_Private()
357 vals = values + idx * bs2 * n + bs * i; in MatStashValuesRowBlocked_Private()
391 PetscInt i, j, k, bs2, bs = stash->bs, l; in MatStashValuesColBlocked_Private() local
400 bs2 = bs * bs; in MatStashValuesColBlocked_Private()
407 array = space->val + bs2 * l; in MatStashValuesColBlocked_Private()
408 vals = values + idx * bs2 * n + bs * i; in MatStashValuesColBlocked_Private()
446 PetscInt *owner, *startv, *starti, bs2; in MatStashScatterBegin_Ref() local
465 bs2 = stash->bs * stash->bs; in MatStashScatterBegin_Ref()
512 …for (PetscMPIInt i = 0; i < nreceives; i++) PetscCall(PetscMPIIntCast(olengths[i] * bs2 / 2, &olen… in MatStashScatterBegin_Ref()
522 PetscCall(PetscMalloc2(bs2 * stash->n, &svalues, 2 * (stash->n + 1), &sindices)); in MatStashScatterBegin_Ref()
542 if (bs2 == 1) { in MatStashScatterBegin_Ref()
547 buf1 = svalues + bs2 * startv[j]; in MatStashScatterBegin_Ref()
548 buf2 = space->val + bs2 * l; in MatStashScatterBegin_Ref()
549 for (k = 0; k < bs2; k++) buf1[k] = buf2[k]; in MatStashScatterBegin_Ref()
565 …PetscCallMPI(MPIU_Isend(svalues + bs2 * startv[i], bs2 * nlengths[i], MPIU_SCALAR, i, tag2, comm, … in MatStashScatterBegin_Ref()
571 …tscInfo(NULL, "Mesg_to: %d: size: %zu bytes\n", i, (size_t)(nlengths[i] * (bs2 * sizeof(PetscScala… in MatStashScatterBegin_Ref()
630 PetscInt bs2; in MatStashScatterGetMesg_Ref() local
639 bs2 = stash->bs * stash->bs; in MatStashScatterGetMesg_Ref()
655 *nvals = *nvals / bs2; in MatStashScatterGetMesg_Ref()
687 …PetscInt n = stash->n, bs = stash->bs, bs2 = bs * bs, cnt, *row, *col, *perm, rowstart, … in MatStashSortCompress_Private() local
696 valptr[cnt] = &space->val[i * bs2]; in MatStashSortCompress_Private()
714 PetscCall(PetscArraycpy(block->vals, valptr[perm[colstart]], bs2)); in MatStashSortCompress_Private()
717 for (l = 0; l < bs2; l++) block->vals[l] += valptr[perm[j]][l]; in MatStashSortCompress_Private()
719 PetscCall(PetscArraycpy(block->vals, valptr[perm[j]], bs2)); in MatStashSortCompress_Private()
735 PetscInt bs2 = PetscSqr(stash->bs); in MatStashBlockTypeSetUp() local
761 stash->blocktype_size = offsetof(struct DummyBlock, vals) + bs2 * sizeof(PetscScalar); in MatStashBlockTypeSetUp()
769 blocklens[1] = (PetscMPIInt)bs2; in MatStashBlockTypeSetUp()
986 PetscInt bs2 = stash->bs * stash->bs; in MatStashScatterEnd_BTS() local
987 PetscInt oldnmax = ((int)(stash->n * 1.1) + 5) * bs2; in MatStashScatterEnd_BTS()