Lines Matching refs:Kokkos

10 using HostMirrorMemorySpace = Kokkos::DualView<PetscScalar *>::host_mirror_space::memory_space;
73     Kokkos::View<PetscScalar *> a_local_d;
74     Kokkos::View<PetscInt *>    i_local_d;
75     Kokkos::View<PetscInt *>    j_local_d;
78     Kokkos::View<PetscScalar *> a_nonlocal_d;
79     Kokkos::View<PetscInt *>    i_nonlocal_d;
80     Kokkos::View<PetscInt *>    j_nonlocal_d;
83     PetscCallCXX(a_local_d = Kokkos::View<PetscScalar *>("a_local_d", di[5]));
84     PetscCallCXX(i_local_d = Kokkos::View<PetscInt *>("i_local_d", m + 1));
85     PetscCallCXX(j_local_d = Kokkos::View<PetscInt *>("j_local_d", di[5]));
88     PetscCallCXX(a_nonlocal_d = Kokkos::View<PetscScalar *>("a_nonlocal_d", oi[5]));
89     PetscCallCXX(i_nonlocal_d = Kokkos::View<PetscInt *>("i_nonlocal_d", m + 1));
90     PetscCallCXX(j_nonlocal_d = Kokkos::View<PetscInt *>("j_nonlocal_d", oi[5]));
96     Kokkos::View<PetscScalar *, HostMirrorMemorySpace> a_local_h;
97     Kokkos::View<PetscInt *, HostMirrorMemorySpace>    i_local_h;
98     Kokkos::View<PetscInt *, HostMirrorMemorySpace>    j_local_h;
99     Kokkos::View<PetscScalar *, HostMirrorMemorySpace> a_nonlocal_h;
100     Kokkos::View<PetscInt *, HostMirrorMemorySpace>    i_nonlocal_h;
101     Kokkos::View<PetscInt *, HostMirrorMemorySpace>    j_nonlocal_h;
103     PetscCallCXX(a_local_h = Kokkos::View<PetscScalar *, HostMirrorMemorySpace>(da, di[5]));
104     PetscCallCXX(i_local_h = Kokkos::View<PetscInt *, HostMirrorMemorySpace>(di, m + 1));
105     PetscCallCXX(j_local_h = Kokkos::View<PetscInt *, HostMirrorMemorySpace>(dj, di[5]));
106     PetscCallCXX(a_nonlocal_h = Kokkos::View<PetscScalar *, HostMirrorMemorySpace>(oa, oi[5]));
107     PetscCallCXX(i_nonlocal_h = Kokkos::View<PetscInt *, HostMirrorMemorySpace>(oi, m + 1));
108     PetscCallCXX(j_nonlocal_h = Kokkos::View<PetscInt *, HostMirrorMemorySpace>(oj, oi[5]));
112     PetscCallCXX(Kokkos::deep_copy(a_local_d, a_local_h));
113     PetscCallCXX(Kokkos::deep_copy(i_local_d, i_local_h));
114     PetscCallCXX(Kokkos::deep_copy(j_local_d, j_local_h));
115     PetscCallCXX(Kokkos::deep_copy(a_nonlocal_d, a_nonlocal_h));
116     PetscCallCXX(Kokkos::deep_copy(i_nonlocal_d, i_nonlocal_h));
117     PetscCallCXX(Kokkos::deep_copy(j_nonlocal_d, j_nonlocal_h));