Lines Matching refs:kv
1177 PetscErrorCode MatSeqAIJGetKokkosView(Mat A, ConstMatScalarKokkosView *kv)
1183 PetscAssertPointer(kv, 2);
1187 *kv = aijkok->a_dual.view_device();
1191 PetscErrorCode MatSeqAIJRestoreKokkosView(Mat A, ConstMatScalarKokkosView *kv)
1195 PetscAssertPointer(kv, 2);
1200 PetscErrorCode MatSeqAIJGetKokkosView(Mat A, MatScalarKokkosView *kv)
1206 PetscAssertPointer(kv, 2);
1210 *kv = aijkok->a_dual.view_device();
1214 PetscErrorCode MatSeqAIJRestoreKokkosView(Mat A, MatScalarKokkosView *kv)
1218 PetscAssertPointer(kv, 2);
1224 PetscErrorCode MatSeqAIJGetKokkosViewWrite(Mat A, MatScalarKokkosView *kv)
1230 PetscAssertPointer(kv, 2);
1233 *kv = aijkok->a_dual.view_device();
1237 PetscErrorCode MatSeqAIJRestoreKokkosViewWrite(Mat A, MatScalarKokkosView *kv)
1241 PetscAssertPointer(kv, 2);
1396 ConstMatScalarKokkosView kv;
1412 kv = Kokkos::create_mirror_view_and_copy(DefaultMemorySpace(), ConstMatScalarKokkosViewHost(v, n));
1414 kv = ConstMatScalarKokkosView(v, n); /* Directly use v[]'s memory */
1424 for (PetscCount k = jmap(i); k < jmap(i + 1); k++) sum += kv(perm(k));