Lines Matching refs:ec

10   PetscInt        i, j, *aj = B->j, ec = 0, *garray, *sgarray;  in MatSetUpMultiply_MPISBAIJ()  local
34 if (!data) PetscCall(PetscHMapISet(gid1_lid1, gid1, ++ec)); in MatSetUpMultiply_MPISBAIJ()
38 PetscCall(PetscMalloc1(ec, &garray)); in MatSetUpMultiply_MPISBAIJ()
48 PetscCall(PetscSortInt(ec, garray)); in MatSetUpMultiply_MPISBAIJ()
50 for (i = 0; i < ec; i++) PetscCall(PetscHMapISet(gid1_lid1, garray[i] + 1, i + 1)); in MatSetUpMultiply_MPISBAIJ()
61 PetscCall(PetscMalloc2(2 * ec, &sgarray, ec, &ec_owner)); in MatSetUpMultiply_MPISBAIJ()
62 for (i = j = 0; i < ec; i++) { in MatSetUpMultiply_MPISBAIJ()
72 if (!indices[aj[B->i[i] + j]]) ec++; in MatSetUpMultiply_MPISBAIJ()
78 PetscCall(PetscMalloc1(ec, &garray)); in MatSetUpMultiply_MPISBAIJ()
79 PetscCall(PetscMalloc2(2 * ec, &sgarray, ec, &ec_owner)); in MatSetUpMultiply_MPISBAIJ()
81 ec = 0; in MatSetUpMultiply_MPISBAIJ()
85 garray[ec] = i; in MatSetUpMultiply_MPISBAIJ()
86 ec_owner[ec] = j; in MatSetUpMultiply_MPISBAIJ()
87 ec++; in MatSetUpMultiply_MPISBAIJ()
93 for (i = 0; i < ec; i++) indices[garray[i]] = i; in MatSetUpMultiply_MPISBAIJ()
101 B->nbs = ec; in MatSetUpMultiply_MPISBAIJ()
103 …tscLayoutCreateFromSizes(PetscObjectComm((PetscObject)sbaij->B), ec * mat->rmap->bs, ec * mat->rma… in MatSetUpMultiply_MPISBAIJ()
107 PetscCall(VecCreateSeq(PETSC_COMM_SELF, ec * bs, &sbaij->lvec)); in MatSetUpMultiply_MPISBAIJ()
110 PetscCall(PetscMalloc1(2 * ec, &stmp)); in MatSetUpMultiply_MPISBAIJ()
111 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, ec, garray, PETSC_COPY_VALUES, &from)); in MatSetUpMultiply_MPISBAIJ()
112 for (i = 0; i < ec; i++) stmp[i] = i; in MatSetUpMultiply_MPISBAIJ()
113 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, ec, stmp, PETSC_COPY_VALUES, &to)); in MatSetUpMultiply_MPISBAIJ()
127 …PetscCall(VecCreateMPI(PetscObjectComm((PetscObject)mat), (mbs + ec) * bs, PETSC_DETERMINE, &sbaij… in MatSetUpMultiply_MPISBAIJ()
134 for (i = 0; i < ec; i++) { in MatSetUpMultiply_MPISBAIJ()
140 for (i = ec, j = 0; i < 2 * ec; i++, j++) sgarray[i] = k + j; in MatSetUpMultiply_MPISBAIJ()
141 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, 2 * ec, sgarray, PETSC_COPY_VALUES, &from)); in MatSetUpMultiply_MPISBAIJ()
145 for (i = 0; i < ec; i++) stmp[i] = (k + i); in MatSetUpMultiply_MPISBAIJ()
147 for (i = ec; i < 2 * ec; i++) stmp[i] = sgarray[i - ec]; in MatSetUpMultiply_MPISBAIJ()
149 PetscCall(ISCreateBlock(PETSC_COMM_SELF, bs, 2 * ec, stmp, PETSC_COPY_VALUES, &to)); in MatSetUpMultiply_MPISBAIJ()