Lines Matching refs:bases
204 PetscInt xbase, *bases, *ldims, j, x_t, y_t, s_t, base, count; in DMSetUp_DA_2D() local
374 PetscCall(PetscMalloc2(size + 1, &bases, size, &ldims)); in DMSetUp_DA_2D()
376 bases[0] = 0; in DMSetUp_DA_2D()
377 for (i = 1; i <= size; i++) bases[i] = ldims[i - 1]; in DMSetUp_DA_2D()
378 for (i = 1; i <= size; i++) bases[i] += bases[i - 1]; in DMSetUp_DA_2D()
379 base = bases[rank] * dof; in DMSetUp_DA_2D()
530 xbase = bases[rank]; in DMSetUp_DA_2D()
535 s_t = bases[n0] + x_t * y_t - (s_y - i) * x_t - s_x; in DMSetUp_DA_2D()
542 s_t = bases[n1] + x_t * y_t - (s_y + 1 - i) * x_t; in DMSetUp_DA_2D()
545 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + x * (s_y - i + 1) + j; in DMSetUp_DA_2D()
551 s_t = bases[n2] + x_t * y_t - (s_y + 1 - i) * x_t; in DMSetUp_DA_2D()
560 s_t = bases[n3] + (i + 1) * x_t - s_x; in DMSetUp_DA_2D()
563 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + x * i + s_x - j; in DMSetUp_DA_2D()
571 s_t = bases[n5] + (i)*x_t; in DMSetUp_DA_2D()
574 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + x * (i + 1) - 2 - j; in DMSetUp_DA_2D()
582 s_t = bases[n6] + (i)*x_t - s_x; in DMSetUp_DA_2D()
589 s_t = bases[n7] + (i - 1) * x_t; in DMSetUp_DA_2D()
592 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + x * (y - i - 1) + j; in DMSetUp_DA_2D()
598 s_t = bases[n8] + (i - 1) * x_t; in DMSetUp_DA_2D()
622 xbase = bases[rank]; in DMSetUp_DA_2D()
627 s_t = bases[n0] + x_t * y_t - (s_y - i) * x_t - s_x; in DMSetUp_DA_2D()
635 s_t = bases[n1] + x_t * y_t - (s_y + 1 - i) * x_t; in DMSetUp_DA_2D()
639 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + x * (s_y - i + 1) + j; in DMSetUp_DA_2D()
647 s_t = bases[n2] + x_t * y_t - (s_y + 1 - i) * x_t; in DMSetUp_DA_2D()
658 s_t = bases[n3] + (i + 1) * x_t - s_x; in DMSetUp_DA_2D()
662 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + x * i + s_x - j; in DMSetUp_DA_2D()
673 s_t = bases[n5] + (i)*x_t; in DMSetUp_DA_2D()
677 for (j = 0; j < s_x; j++) idx[nn++] = bases[rank] + x * (i + 1) - 2 - j; in DMSetUp_DA_2D()
688 s_t = bases[n6] + (i)*x_t - s_x; in DMSetUp_DA_2D()
696 s_t = bases[n7] + (i - 1) * x_t; in DMSetUp_DA_2D()
700 for (j = 0; j < x; j++) idx[nn++] = bases[rank] + x * (y - i - 1) + j; in DMSetUp_DA_2D()
708 s_t = bases[n8] + (i - 1) * x_t; in DMSetUp_DA_2D()
721 PetscCall(PetscFree2(bases, ldims)); in DMSetUp_DA_2D()