Lines Matching refs:sendbuf

2050   PetscCall(PetscFree2(atb->sendbuf, atb->recvcounts));  in MatProductCtxDestroy_MatTransMatMult_MPIDense_MPIDense()
2073 PetscScalar *carray, *sendbuf; in MatTransposeMatMultNumeric_MPIDense_MPIDense() local
2083 sendbuf = atb->sendbuf; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2157 for (i = ranges[proc]; i < ranges[proc + 1]; i++) sendbuf[k++] = atbarray[i + j * lda]; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2164 PetscCallMPI(MPI_Reduce_scatter(sendbuf, carray, recvcounts, MPIU_SCALAR, MPIU_SUM, comm)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2203 PetscCall(PetscMalloc2(cM * cN, &atb->sendbuf, size, &atb->recvcounts)); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2269 PetscScalar *sendbuf, *recvbuf = NULL, *cv; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic() local
2296 sendbuf = (PetscScalar *)bv; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2298 sendbuf = abt->buf[0]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2300 for (j = 0; j < bn; j++, k++) sendbuf[k] = bv[i * blda + j]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2322 PetscCallMPI(MPIU_Isend(sendbuf, sendsiz, MPIU_SCALAR, sendto, abt->tag, comm, &reqs[0])); in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2328 …llBLAS("BLASgemm", BLASgemm_("N", "T", &cm, &cn, &ck, &_DOne, av, &alda, sendbuf, &cn, &_DZero, cv… in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2336 sendbuf = recvbuf; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Cyclic()
2353 PetscScalar *cv, *sendbuf, *recvbuf; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv() local
2375 sendbuf = abt->buf[0]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2378 for (i = 0; i < cK; i++, k++) sendbuf[k] = bv[i * blda + j]; in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()
2382 …PetscCallMPI(MPI_Allgatherv(sendbuf, ibn, MPIU_SCALAR, recvbuf, abt->recvcounts, abt->recvdispls, … in MatMatTransposeMultNumeric_MPIDense_MPIDense_Allgatherv()