Lines Matching refs:unit

405   if (!link->isbuiltin) PetscCallMPI(MPI_Type_free(&link->unit));  in PetscSFLinkDestroy()
420 PetscErrorCode PetscSFLinkCreate(PetscSF sf, MPI_Datatype unit, PetscMemType rootmtype, const void … in PetscSFLinkCreate() argument
423 PetscCall(PetscSFSetErrorOnUnsupportedOverlap(sf, unit, rootdata, leafdata)); in PetscSFLinkCreate()
429 …PetscCall(PetscSFLinkCreate_NVSHMEM(sf, unit, rootmtype, rootdata, leafmtype, leafdata, op, sfop, … in PetscSFLinkCreate()
434 …PetscCall(PetscSFLinkCreate_MPI(sf, unit, rootmtype, rootdata, leafmtype, leafdata, op, sfop, myli… in PetscSFLinkCreate()
438 PetscErrorCode PetscSFLinkGetInUse(PetscSF sf, MPI_Datatype unit, const void *rootdata, const void … in PetscSFLinkGetInUse() argument
447 PetscCall(MPIPetsc_Type_compare(unit, link->unit, &match)); in PetscSFLinkGetInUse()
480 PetscErrorCode PetscSFSetErrorOnUnsupportedOverlap(PetscSF sf, MPI_Datatype unit, const void *rootd… in PetscSFSetErrorOnUnsupportedOverlap() argument
493 PetscCall(MPIPetsc_Type_compare(unit, link->unit, &match)); in PetscSFSetErrorOnUnsupportedOverlap()
508 PetscErrorCode PetscSFLinkSetUp_Host(PetscSF sf, PetscSFLink link, MPI_Datatype unit) in PetscSFLinkSetUp_Host() argument
519 PetscCall(MPIPetsc_Type_compare_contig(unit, MPI_SIGNED_CHAR, &nSignedChar)); in PetscSFLinkSetUp_Host()
520 PetscCall(MPIPetsc_Type_compare_contig(unit, MPI_UNSIGNED_CHAR, &nUnsignedChar)); in PetscSFLinkSetUp_Host()
522 PetscCall(MPIPetsc_Type_compare_contig(unit, MPI_INT, &nInt)); in PetscSFLinkSetUp_Host()
523 PetscCall(MPIPetsc_Type_compare_contig(unit, MPIU_INT, &nPetscInt)); in PetscSFLinkSetUp_Host()
524 PetscCall(MPIPetsc_Type_compare_contig(unit, MPIU_REAL, &nPetscReal)); in PetscSFLinkSetUp_Host()
526 PetscCall(MPIPetsc_Type_compare_contig(unit, MPIU_COMPLEX, &nPetscComplex)); in PetscSFLinkSetUp_Host()
528 PetscCall(MPIPetsc_Type_compare(unit, MPI_2INT, &is2Int)); in PetscSFLinkSetUp_Host()
529 PetscCall(MPIPetsc_Type_compare(unit, MPIU_2INT, &is2PetscInt)); in PetscSFLinkSetUp_Host()
531 PetscCallMPI(MPIPetsc_Type_get_envelope(unit, &ni, &na, &nc, &nd, &combiner)); in PetscSFLinkSetUp_Host()
541 link->unit = MPI_2INT; in PetscSFLinkSetUp_Host()
548 link->unit = MPIU_2INT; in PetscSFLinkSetUp_Host()
563 link->unit = MPIU_REAL; in PetscSFLinkSetUp_Host()
579 link->unit = MPIU_INT; in PetscSFLinkSetUp_Host()
596 link->unit = MPI_INT; in PetscSFLinkSetUp_Host()
613 link->unit = MPI_SIGNED_CHAR; in PetscSFLinkSetUp_Host()
629 link->unit = MPI_UNSIGNED_CHAR; in PetscSFLinkSetUp_Host()
646 link->unit = MPIU_COMPLEX; in PetscSFLinkSetUp_Host()
652 PetscCallMPI(MPI_Type_get_extent(unit, &lb, &nbyte)); in PetscSFLinkSetUp_Host()
678 if (link->isbuiltin) link->unit = unit; in PetscSFLinkSetUp_Host()
681 if (!link->isbuiltin) PetscCallMPI(MPI_Type_dup(unit, &link->unit)); in PetscSFLinkSetUp_Host()
870 …char *)buf + i * link->unitbytes, (char *)data + indices[i] * link->unitbytes, 1, link->unit, op)); in PetscSFLinkUnpackDataWithMPIReduceLocal()
873 …PetscCallMPI(MPIU_Reduce_local(buf, (char *)data + start * link->unitbytes, count, link->unit, op)… in PetscSFLinkUnpackDataWithMPIReduceLocal()
894 …har *)src + srcIdx[i] * link->unitbytes, (char *)dst + disp * link->unitbytes, 1, link->unit, op)); in PetscSFLinkScatterDataWithMPIReduceLocal()