Lines Matching refs:atb
2047 MatProductCtx_TransMatMultDense *atb = *(MatProductCtx_TransMatMultDense **)data; in MatProductCtxDestroy_MatTransMatMult_MPIDense_MPIDense() local
2050 PetscCall(PetscFree2(atb->sendbuf, atb->recvcounts)); in MatProductCtxDestroy_MatTransMatMult_MPIDense_MPIDense()
2051 PetscCall(MatDestroy(&atb->atb)); in MatProductCtxDestroy_MatTransMatMult_MPIDense_MPIDense()
2052 PetscCall(PetscFree(atb)); in MatProductCtxDestroy_MatTransMatMult_MPIDense_MPIDense()
2070 MatProductCtx_TransMatMultDense *atb; in MatTransposeMatMultNumeric_MPIDense_MPIDense() local
2081 atb = (MatProductCtx_TransMatMultDense *)C->product->data; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2082 recvcounts = atb->recvcounts; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2083 sendbuf = atb->sendbuf; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2089 …cCall(MatTransposeMatMult(a->A, b->A, atb->atb ? MAT_REUSE_MATRIX : MAT_INITIAL_MATRIX, PETSC_DETE… in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2096 Mat atb_local = atb->atb; in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2153 PetscCall(MatDenseGetArrayRead(atb->atb, &atbarray)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2154 PetscCall(MatDenseGetLDA(atb->atb, &lda)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2160 PetscCall(MatDenseRestoreArrayRead(atb->atb, &atbarray)); in MatTransposeMatMultNumeric_MPIDense_MPIDense()
2177 MatProductCtx_TransMatMultDense *atb; in MatTransposeMatMultSymbolic_MPIDense_MPIDense() local
2201 PetscCall(PetscNew(&atb)); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2203 PetscCall(PetscMalloc2(cM * cN, &atb->sendbuf, size, &atb->recvcounts)); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2205 …; i < size; i++) PetscCall(PetscMPIIntCast((ranges[i + 1] - ranges[i]) * cN, &atb->recvcounts[i])); in MatTransposeMatMultSymbolic_MPIDense_MPIDense()
2206 C->product->data = atb; in MatTransposeMatMultSymbolic_MPIDense_MPIDense()