Searched refs:nLocalRows (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/ksp/pc/impls/amgx/ |
| H A D | amgx.cxx | 134 PetscInt nLocalRows; member
265 …PetscCall(MatGetRowIJ(amgx->localA, 0, PETSC_FALSE, PETSC_FALSE, &amgx->nLocalRows, &rowOffsets, &… in PCSetUp_AMGX()
267 …nLocalRows < std::numeric_limits<int>::max(), PETSC_COMM_SELF, PETSC_ERR_PLIB, "AmgX restricted to… in PCSetUp_AMGX()
270 …PetscCallCUDA(cudaMemcpy(&amgx->nnz, &rowOffsets[amgx->nLocalRows], sizeof(int), cudaMemcpyDefault… in PCSetUp_AMGX()
272 amgx->nnz = rowOffsets[amgx->nLocalRows]; in PCSetUp_AMGX()
282 …PetscCallMPI(MPI_Allgather(&amgx->nLocalRows, 1, MPIU_INT, partitionOffsets.data() + 1, 1, MPIU_IN… in PCSetUp_AMGX()
296 …PetscCallAmgX(AMGX_matrix_upload_distributed(amgx->A, amgx->nGlobalRows, (int)amgx->nLocalRows, (i… in PCSetUp_AMGX()
305 …PetscCallAmgX(AMGX_matrix_replace_coefficients(amgx->A, amgx->nLocalRows, amgx->nnz, amgx->values,… in PCSetUp_AMGX()
348 PetscCallAmgX(AMGX_vector_upload(amgx->sol, amgx->nLocalRows, 1, x_)); in PCApply_AMGX()
349 PetscCallAmgX(AMGX_vector_upload(amgx->rhs, amgx->nLocalRows, 1, b_)); in PCApply_AMGX()
|