| /petsc/src/ksp/ksp/utils/lmvm/blas_cyclic/ |
| H A D | blas_cyclic.c | 42 PetscInt m, m_local; in VecAXPBYCyclic() local
54 PetscCall(VecGetLocalSize(x, &m_local)); in VecAXPBYCyclic()
61 if (m_local == m) PetscCall(AXPBYCyclic_CUPM_Private(m, oldest, next, alpha, x_, beta, y_, 1)); in VecAXPBYCyclic()
65 } else if (m_local == m) on_host = PETSC_TRUE; in VecAXPBYCyclic()
110 PetscInt m, m_local; in VecDMVCyclic() local
126 PetscCall(VecGetLocalSize(A, &m_local)); in VecDMVCyclic()
134 …if (m_local == m) PetscCall(DMVCyclic_CUPM_Private(hermitian_transpose, m, oldest, next, alpha, A_… in VecDMVCyclic()
138 } else if (m_local == m) on_host = PETSC_TRUE; in VecDMVCyclic()
194 PetscInt m, m_local; in VecDSVCyclic() local
210 PetscCall(VecGetLocalSize(A, &m_local)); in VecDSVCyclic()
[all …]
|
| /petsc/src/ksp/ksp/utils/lmvm/ |
| H A D | lmproducts.c | 12 PetscInt m, m_local; in LMProductsCreate() local
21 PetscCall(MatGetLocalSize(basis->vecs, NULL, &m_local)); in LMProductsCreate()
22 (*dots)->m_local = m_local; in LMProductsCreate()
37 …VecType(PetscObjectComm((PetscObject)basis->vecs), vec_type, m_local, m_local, m, m, m_local, NULL… in LMProductsCreate()
142 if (dots->m_local) { in LMProductsUpdate_Internal()
220 if (dots->m_local) dot = values[(j % dots->m) * lda + (i % dots->m)]; in LMProductsUpdate_Internal()
263 …PetscCheck(dest->m_local == src->m_local, PETSC_COMM_SELF, PETSC_ERR_PLIB, "Cannot copy to LMProdu… in LMProductsCopy()
266 dest->m_local = src->m_local; in LMProductsCopy()
291 if (local_is_nonempty) *local_is_nonempty = (dots->m_local == dots->m) ? PETSC_TRUE : PETSC_FALSE; in LMProductsGetLocalMatrix()
308 if (dots->m_local) PetscCall(VecCopy(dots->diagonal_dup, dots->diagonal_local)); in LMProductsGetUpdatedDiagonal()
[all …]
|
| H A D | lmproducts.h | 22 …PetscInt m_local; // rank 0 will have all values (m_local = m), others have none (m_local … member
|
| /petsc/src/ksp/ksp/utils/lmvm/dense/ |
| H A D | denseqn.c | 641 PetscInt m_local; in MatGetLDLT() local
647 PetscCall(MatGetLocalSize(result, &m_local, NULL)); in MatGetLDLT()
650 if (m_local) { in MatGetLDLT()
665 PetscInt m = lmvm->m, m_local; in MatLMVMDBFGSUpdateMultData() local
714 PetscCall(MatGetLocalSize(lbfgs->YtS_triu_strict, &m_local, NULL)); in MatLMVMDBFGSUpdateMultData()
729 if (m_local > 0) { in MatLMVMDBFGSUpdateMultData()
745 if (m_local) { in MatLMVMDBFGSUpdateMultData()
851 PetscInt idx, i, j, m_local, local_n; in MatMult_LMVMDBFGS() local
958 PetscCall(VecGetSize(lbfgs->rwork2_local, &m_local)); in MatMult_LMVMDBFGS()
959 if (m_local) { in MatMult_LMVMDBFGS()
[all …]
|
| /petsc/src/ksp/pc/impls/asm/ |
| H A D | asm.c | 157 PetscInt i, m, m_local; in PCSetUp_ASM() local
360 PetscCall(ISGetLocalSize(osm->is_local[i], &m_local)); in PCSetUp_ASM()
364 PetscCall(PetscMalloc1(m_local, &idx1)); in PCSetUp_ASM()
365 … PetscCall(ISGlobalToLocalMappingApply(ltog, IS_GTOLM_DROP, m_local, idx_local, &nout, idx1)); in PCSetUp_ASM()
367 PetscCheck(nout == m_local, PETSC_COMM_SELF, PETSC_ERR_PLIB, "is_local not a subset of is"); in PCSetUp_ASM()
368 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, m_local, idx1, PETSC_OWN_POINTER, &isll)); in PCSetUp_ASM()
371 PetscCall(PetscMalloc1(m_local, &idx2)); in PCSetUp_ASM()
372 … PetscCall(ISGlobalToLocalMappingApply(ltog, IS_GTOLM_DROP, m_local, idx_local, &nout, idx2)); in PCSetUp_ASM()
374 … PetscCheck(nout == m_local, PETSC_COMM_SELF, PETSC_ERR_PLIB, "is_local not a subset of lis"); in PCSetUp_ASM()
375 PetscCall(ISCreateGeneral(PETSC_COMM_SELF, m_local, idx2, PETSC_OWN_POINTER, &isll_local)); in PCSetUp_ASM()
|