Home
last modified time | relevance | path

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

/petsc/src/vec/is/sf/impls/basic/nvshmem/
H A Dsfnvshmem.cu159 …tmtype, const void *rootdata, PetscMemType leafmtype, const void *leafdata, PetscBool *use_nvshmem) in PetscSFLinkNvshmemCheck() argument
171 if (sf->use_nvshmem && !sf->checked_nvshmem_eligibility) { in PetscSFLinkNvshmemCheck()
175 …if (!isBasic || (result != MPI_IDENT && result != MPI_CONGRUENT)) sf->use_nvshmem = PETSC_FALSE; /… in PetscSFLinkNvshmemCheck()
181 if (sf->use_nvshmem) { in PetscSFLinkNvshmemCheck()
184 if (hasNullRank) sf->use_nvshmem = PETSC_FALSE; in PetscSFLinkNvshmemCheck()
190 if (sf->use_nvshmem) { in PetscSFLinkNvshmemCheck()
203 *use_nvshmem = PETSC_TRUE; in PetscSFLinkNvshmemCheck()
205 *use_nvshmem = PETSC_FALSE; in PetscSFLinkNvshmemCheck()
208 *use_nvshmem = PETSC_FALSE; in PetscSFLinkNvshmemCheck()
722 if (link->use_nvshmem) { in PetscSFLinkCreate_NVSHMEM()
[all …]
/petsc/src/vec/vec/impls/mpi/
H A Dpvecimpl.h42 PetscBool use_nvshmem; /* Try to use NVSHMEM in communication of, for example, VecNorm */ member
/petsc/src/vec/is/sf/impls/basic/
H A Dsfmpi.c119 if (!link->use_nvshmem) { /* Only check with MPI links */ in PetscSFLinkCreate_MPI()
H A Dsfpack.c406 if (!link->use_nvshmem) { in PetscSFLinkDestroy()
426 PetscBool use_nvshmem; in PetscSFLinkCreate() local
427 PetscCall(PetscSFLinkNvshmemCheck(sf, rootmtype, rootdata, leafmtype, leafdata, &use_nvshmem)); in PetscSFLinkCreate()
428 if (use_nvshmem) { in PetscSFLinkCreate()
H A Dsfpack.h198 PetscBool use_nvshmem; /* Does this link use nvshem (vs. MPI) for communication? */ member
/petsc/include/petsc/private/
H A Dsfimpl.h115 …PetscBool use_nvshmem; /* TRY to use nvshmem on cuda devices with this SF when pos… member
/petsc/src/vec/is/sf/interface/
H A Dsf.c86 b->use_nvshmem = PETSC_FALSE; /* Default is not to try NVSHMEM */ in PetscSFCreate()
88 PetscCall(PetscOptionsGetBool(NULL, NULL, "-use_nvshmem", &b->use_nvshmem, NULL)); in PetscSFCreate()