Lines Matching refs:abt

2058   MatProductCtx_MatTransMultDense *abt = *(MatProductCtx_MatTransMultDense **)data;  in MatProductCtxDestroy_MatMatTransMult_MPIDense_MPIDense()  local
2061 PetscCall(PetscFree2(abt->buf[0], abt->buf[1])); in MatProductCtxDestroy_MatMatTransMult_MPIDense_MPIDense()
2062 PetscCall(PetscFree2(abt->recvcounts, abt->recvdispls)); in MatProductCtxDestroy_MatMatTransMult_MPIDense_MPIDense()
2063 PetscCall(PetscFree(abt)); in MatProductCtxDestroy_MatMatTransMult_MPIDense_MPIDense()
2218 MatProductCtx_MatTransMultDense *abt; in MatMatTransposeMultSymbolic_MPIDense_MPIDense() local
2240 PetscCall(PetscNew(&abt)); in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2241 abt->tag = tag; in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2242 abt->alg = alg; in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2247 PetscCall(PetscMalloc2(bufsiz, &abt->buf[0], bufsiz, &abt->buf[1])); in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2250 …PetscCall(PetscMalloc2(B->rmap->n * B->cmap->N, &abt->buf[0], B->rmap->N * B->cmap->N, &abt->buf[1… in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2251 PetscCall(PetscMalloc2(size, &abt->recvcounts, size + 1, &abt->recvdispls)); in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2252 …for (i = 0; i <= size; i++) PetscCall(PetscMPIIntCast(B->rmap->range[i] * A->cmap->N, &abt->recvdi… in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2253 …or (i = 0; i < size; i++) PetscCall(PetscMPIIntCast(abt->recvdispls[i + 1] - abt->recvdispls[i], & in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2257 C->product->data = abt; in MatMatTransposeMultSymbolic_MPIDense_MPIDense()
2266 MatProductCtx_MatTransMultDense *abt; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic() local
2280 abt = (MatProductCtx_MatTransMultDense *)C->product->data; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2298 sendbuf = abt->buf[0]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2321 recvbuf = (i & 1) ? abt->buf[0] : abt->buf[1]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2322 PetscCallMPI(MPIU_Isend(sendbuf, sendsiz, MPIU_SCALAR, sendto, abt->tag, comm, &reqs[0])); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2323 PetscCallMPI(MPIU_Irecv(recvbuf, recvsiz, MPIU_SCALAR, recvfrom, abt->tag, comm, &reqs[1])); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2350 MatProductCtx_MatTransMultDense *abt; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv() local
2362 abt = (MatProductCtx_MatTransMultDense *)C->product->data; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2375 sendbuf = abt->buf[0]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2376 recvbuf = abt->buf[1]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2382 …PetscCallMPI(MPI_Allgatherv(sendbuf, ibn, MPIU_SCALAR, recvbuf, abt->recvcounts, abt->recvdispls, … in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2398 MatProductCtx_MatTransMultDense *abt; in MatMatTransposeMultNumeric_MPIDense_MPIDense() local
2403 abt = (MatProductCtx_MatTransMultDense *)C->product->data; in MatMatTransposeMultNumeric_MPIDense_MPIDense()
2404 switch (abt->alg) { in MatMatTransposeMultNumeric_MPIDense_MPIDense()