Searched refs:dstmtype (Results 1 – 3 of 3) sorted by relevance
| /petsc/src/vec/is/sf/impls/basic/kokkos/ |
| H A D | sfkok.kokkos.cxx | 575 static PetscErrorCode PetscSFLinkMemcpy_Kokkos(PetscSFLink PETSC_UNUSED link, PetscMemType dstmtype… in PetscSFLinkMemcpy_Kokkos() argument
581 if (PetscMemTypeHost(dstmtype) && PetscMemTypeHost(srcmtype)) { // H2H in PetscSFLinkMemcpy_Kokkos()
585 if (PetscMemTypeDevice(dstmtype) && PetscMemTypeHost(srcmtype)) { // H2D in PetscSFLinkMemcpy_Kokkos()
590 } else if (PetscMemTypeHost(dstmtype) && PetscMemTypeDevice(srcmtype)) { // D2H in PetscSFLinkMemcpy_Kokkos()
596 } else if (PetscMemTypeDevice(dstmtype) && PetscMemTypeDevice(srcmtype)) { // D2D in PetscSFLinkMemcpy_Kokkos()
|
| /petsc/src/vec/is/sf/impls/basic/ |
| H A D | sfpack.c | 501 static PetscErrorCode PetscSFLinkMemcpy_Host(PetscSFLink link, PetscMemType dstmtype, void *dst, Pe… in PetscSFLinkMemcpy_Host() argument
1087 PetscMemType rootmtype = link->rootmtype, leafmtype = link->leafmtype, srcmtype, dstmtype; in PetscSFLinkScatterLocal() local
1101 dstmtype = leafmtype; in PetscSFLinkScatterLocal()
1107 dstmtype = rootmtype; in PetscSFLinkScatterLocal()
1110 PetscCall((*link->Memcpy)(link, dstmtype, dstbuf, srcmtype, srcbuf, buflen * link->unitbytes)); in PetscSFLinkScatterLocal()
1112 if (PetscMemTypeHost(dstmtype)) PetscCall((*link->SyncStream)(link)); in PetscSFLinkScatterLocal()
1120 dstmtype = (direction == PETSCSF_ROOT2LEAF) ? link->leafmtype : link->rootmtype; in PetscSFLinkScatterLocal()
1121 PetscCall(PetscSFLinkGetScatterAndOp(link, dstmtype, op, dstdups, &ScatterAndOp)); in PetscSFLinkScatterLocal()
|
| /petsc/src/vec/is/sf/impls/basic/cupm/ |
| H A D | sfcupm_impl.hpp | 582 inline PetscErrorCode SfInterface<T>::LinkMemcpy(PetscSFLink link, PetscMemType dstmtype, void *dst… in LinkMemcpy() argument
591 …if (PetscMemTypeHost(dstmtype) && PetscMemTypeHost(srcmtype)) { /* Separate HostToHost so that pur… in LinkMemcpy()
595 int dtype = PetscMemTypeDevice(dstmtype) ? 1 : 0; in LinkMemcpy()
|