Home
last modified time | relevance | path

Searched refs:atb (Results 1 – 7 of 7) sorted by relevance

/petsc/src/mat/impls/aij/mpi/
H A Dmpimattransposematmult.c12 MatProductCtx_MatTransMatMult *atb = *(MatProductCtx_MatTransMatMult **)data; in MatProductCtxDestroy_MPIDense_MatTransMatMult() local
15 PetscCall(MatDestroy(&atb->mA)); in MatProductCtxDestroy_MPIDense_MatTransMatMult()
16 PetscCall(VecDestroy(&atb->bt)); in MatProductCtxDestroy_MPIDense_MatTransMatMult()
17 PetscCall(VecDestroy(&atb->ct)); in MatProductCtxDestroy_MPIDense_MatTransMatMult()
18 PetscCall(PetscFree(atb)); in MatProductCtxDestroy_MPIDense_MatTransMatMult()
26 MatProductCtx_MatTransMatMult *atb; in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense() local
40 PetscCall(PetscNew(&atb)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
42 PetscCall(MatCreateMAIJ(A, B->cmap->N, &atb->mA)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
43 if (!atb->mA->assembled) { in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
44 PetscCall(MatAssemblyBegin(atb->mA, MAT_FINAL_ASSEMBLY)); in MatTransposeMatMultSymbolic_MPIAIJ_MPIDense()
[all …]
/petsc/src/mat/impls/aij/seq/
H A Dmattransposematmult.c12 MatProductCtx_MatTransMatMult *atb = *(MatProductCtx_MatTransMatMult **)data; in MatProductCtxDestroy_SeqDense_MatTransMatMult() local
15 PetscCall(MatDestroy(&atb->mA)); in MatProductCtxDestroy_SeqDense_MatTransMatMult()
16 PetscCall(VecDestroy(&atb->bt)); in MatProductCtxDestroy_SeqDense_MatTransMatMult()
17 PetscCall(VecDestroy(&atb->ct)); in MatProductCtxDestroy_SeqDense_MatTransMatMult()
18 PetscCall(PetscFree(atb)); in MatProductCtxDestroy_SeqDense_MatTransMatMult()
26 MatProductCtx_MatTransMatMult *atb; in MatTMatTMultSymbolic_SeqAIJ_SeqDense() local
48 PetscCall(PetscNew(&atb)); in MatTMatTMultSymbolic_SeqAIJ_SeqDense()
49 PetscCall(MatCreateMAIJ(A, dofm, &atb->mA)); in MatTMatTMultSymbolic_SeqAIJ_SeqDense()
50 PetscCall(MatCreateVecs(atb->mA, &atb->ct, &atb->bt)); in MatTMatTMultSymbolic_SeqAIJ_SeqDense()
51 C->product->data = atb; in MatTMatTMultSymbolic_SeqAIJ_SeqDense()
[all …]
H A Dmatptap.c36 MatProductCtx_MatTransMatMult *atb; in MatProductSymbolic_PtAP_SeqAIJ_SeqAIJ() local
38 PetscCall(PetscNew(&atb)); in MatProductSymbolic_PtAP_SeqAIJ_SeqAIJ()
42 atb->At = Pt; in MatProductSymbolic_PtAP_SeqAIJ_SeqAIJ()
43 atb->data = C->product->data; in MatProductSymbolic_PtAP_SeqAIJ_SeqAIJ()
44 atb->destroy = C->product->destroy; in MatProductSymbolic_PtAP_SeqAIJ_SeqAIJ()
45 C->product->data = atb; in MatProductSymbolic_PtAP_SeqAIJ_SeqAIJ()
280 MatProductCtx_MatTransMatMult *atb; in MatPtAPNumeric_SeqAIJ_SeqAIJ() local
284 atb = (MatProductCtx_MatTransMatMult *)C->product->data; in MatPtAPNumeric_SeqAIJ_SeqAIJ()
285 PetscCheck(atb, PetscObjectComm((PetscObject)C), PETSC_ERR_PLIB, "Missing data structure"); in MatPtAPNumeric_SeqAIJ_SeqAIJ()
286 PetscCall(MatTranspose(P, MAT_REUSE_MATRIX, &atb->At)); in MatPtAPNumeric_SeqAIJ_SeqAIJ()
[all …]
H A Dmatmatmult.c1376 MatProductCtx_MatTransMatMult *atb = *(MatProductCtx_MatTransMatMult **)data; in MatProductCtxDestroy_SeqAIJ_MatTransMatMult() local
1379 PetscCall(MatDestroy(&atb->At)); in MatProductCtxDestroy_SeqAIJ_MatTransMatMult()
1380 if (atb->destroy) PetscCall((*atb->destroy)(&atb->data)); in MatProductCtxDestroy_SeqAIJ_MatTransMatMult()
1381 PetscCall(PetscFree(atb)); in MatProductCtxDestroy_SeqAIJ_MatTransMatMult()
1419 MatProductCtx_MatTransMatMult *atb; in MatTransposeMatMultSymbolic_SeqAIJ_SeqAIJ() local
1422 PetscCall(PetscNew(&atb)); in MatTransposeMatMultSymbolic_SeqAIJ_SeqAIJ()
1427 product->data = atb; in MatTransposeMatMultSymbolic_SeqAIJ_SeqAIJ()
1429 atb->At = At; in MatTransposeMatMultSymbolic_SeqAIJ_SeqAIJ()
1958 MatProductCtx_MatTransMatMult *atb = (MatProductCtx_MatTransMatMult *)product->data; in MatProductNumeric_AtB_SeqAIJ_SeqAIJ() local
1960 PetscCheck(atb, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Missing product struct"); in MatProductNumeric_AtB_SeqAIJ_SeqAIJ()
[all …]
/petsc/src/mat/impls/dense/mpi/
H A Dmpidense.c2047 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()
[all …]
H A Dmpidense.h14 Mat atb; member
/petsc/src/mat/tests/
H A Dex237.c37 …PetscBool flg, cuda, maij = PETSC_FALSE, check = PETSC_FALSE, abt = PETSC_FALSE, atb = PETSC_FAL… in main() local
71 PetscCall(PetscOptionsGetBool(NULL, NULL, "-AtB", &atb, NULL)); in main()
237 if (atb) { in main()
385 if (!mkl && atb) { in main()
414 if (!mkl && atb && N[k] > 1 && check) { in main()