Home
last modified time | relevance | path

Searched refs:nPetscComplex (Results 1 – 3 of 3) sorted by relevance

/petsc/src/vec/is/sf/impls/basic/kokkos/
H A Dsfkok.kokkos.cxx643 PetscInt nPetscComplex = 0; in PetscSFLinkSetUp_Kokkos() local
656 PetscCall(MPIPetsc_Type_compare_contig(unit, MPIU_COMPLEX, &nPetscComplex)); in PetscSFLinkSetUp_Kokkos()
726 } else if (nPetscComplex) { in PetscSFLinkSetUp_Kokkos()
728 if (nPetscComplex == 8) PackInit_ComplexType<Kokkos::complex<PetscReal>, 8, 1>(link); in PetscSFLinkSetUp_Kokkos()
729 else if (nPetscComplex % 8 == 0) PackInit_ComplexType<Kokkos::complex<PetscReal>, 8, 0>(link); in PetscSFLinkSetUp_Kokkos()
730 else if (nPetscComplex == 4) PackInit_ComplexType<Kokkos::complex<PetscReal>, 4, 1>(link); in PetscSFLinkSetUp_Kokkos()
731 else if (nPetscComplex % 4 == 0) PackInit_ComplexType<Kokkos::complex<PetscReal>, 4, 0>(link); in PetscSFLinkSetUp_Kokkos()
732 else if (nPetscComplex == 2) PackInit_ComplexType<Kokkos::complex<PetscReal>, 2, 1>(link); in PetscSFLinkSetUp_Kokkos()
733 else if (nPetscComplex % 2 == 0) PackInit_ComplexType<Kokkos::complex<PetscReal>, 2, 0>(link); in PetscSFLinkSetUp_Kokkos()
734 else if (nPetscComplex == 1) PackInit_ComplexType<Kokkos::complex<PetscReal>, 1, 1>(link); in PetscSFLinkSetUp_Kokkos()
[all …]
/petsc/src/vec/is/sf/impls/basic/cupm/
H A Dsfcupm_impl.hpp647 PetscInt nPetscComplex = 0; in LinkSetUp() local
659 PetscCall(MPIPetsc_Type_compare_contig(unit, MPIU_COMPLEX, &nPetscComplex)); in LinkSetUp()
729 } else if (nPetscComplex) { in LinkSetUp()
731 if (nPetscComplex == 8) PackInit_ComplexType<PetscComplex, 8, 1>(link); in LinkSetUp()
732 else if (nPetscComplex % 8 == 0) PackInit_ComplexType<PetscComplex, 8, 0>(link); in LinkSetUp()
733 else if (nPetscComplex == 4) PackInit_ComplexType<PetscComplex, 4, 1>(link); in LinkSetUp()
734 else if (nPetscComplex % 4 == 0) PackInit_ComplexType<PetscComplex, 4, 0>(link); in LinkSetUp()
735 else if (nPetscComplex == 2) PackInit_ComplexType<PetscComplex, 2, 1>(link); in LinkSetUp()
736 else if (nPetscComplex % 2 == 0) PackInit_ComplexType<PetscComplex, 2, 0>(link); in LinkSetUp()
737 else if (nPetscComplex == 1) PackInit_ComplexType<PetscComplex, 1, 1>(link); in LinkSetUp()
[all …]
/petsc/src/vec/is/sf/impls/basic/
H A Dsfpack.c515 PetscInt nPetscComplex = 0; in PetscSFLinkSetUp_Host() local
526 PetscCall(MPIPetsc_Type_compare_contig(unit, MPIU_COMPLEX, &nPetscComplex)); in PetscSFLinkSetUp_Host()
632 } else if (nPetscComplex) { in PetscSFLinkSetUp_Host()
633 if (nPetscComplex == 8) PackInit_ComplexType_PetscComplex_8_1(link); in PetscSFLinkSetUp_Host()
634 else if (nPetscComplex % 8 == 0) PackInit_ComplexType_PetscComplex_8_0(link); in PetscSFLinkSetUp_Host()
635 else if (nPetscComplex == 4) PackInit_ComplexType_PetscComplex_4_1(link); in PetscSFLinkSetUp_Host()
636 else if (nPetscComplex % 4 == 0) PackInit_ComplexType_PetscComplex_4_0(link); in PetscSFLinkSetUp_Host()
637 else if (nPetscComplex == 2) PackInit_ComplexType_PetscComplex_2_1(link); in PetscSFLinkSetUp_Host()
638 else if (nPetscComplex % 2 == 0) PackInit_ComplexType_PetscComplex_2_0(link); in PetscSFLinkSetUp_Host()
639 else if (nPetscComplex == 1) PackInit_ComplexType_PetscComplex_1_1(link); in PetscSFLinkSetUp_Host()
[all …]