Lines Matching refs:kv
30 PetscScalarKokkosViewType<MemorySpace> kv;
38 if (overwrite) PetscCall(VecGetKokkosViewWrite(vec, &kv));
39 else PetscCall(VecGetKokkosView(vec, &kv));
41 *ov = PetscScalarKokkosOffsetView1DType<MemorySpace>(kv.data(), {gxs * dof}, {(gxs + gxm) * dof});
48 PetscScalarKokkosViewType<MemorySpace> kv;
54 kv = ov->view(); /* OffsetView to View */
55 if (overwrite) PetscCall(VecRestoreKokkosViewWrite(vec, &kv));
56 else PetscCall(VecRestoreKokkosView(vec, &kv));
64 ConstPetscScalarKokkosViewType<MemorySpace> kv;
72 PetscCall(VecGetKokkosView(vec, &kv));
73 *ov = ConstPetscScalarKokkosOffsetView1DType<MemorySpace>(kv.data(), {gxs * dof}, {(gxs + gxm) * dof});
80 ConstPetscScalarKokkosViewType<MemorySpace> kv;
86 kv = ov->view();
87 PetscCall(VecRestoreKokkosView(vec, &kv));
96 PetscScalarKokkosViewType<MemorySpace> kv;
104 if (overwrite) PetscCall(VecGetKokkosViewWrite(vec, &kv));
105 else PetscCall(VecGetKokkosView(vec, &kv));
106 *ov = PetscScalarKokkosOffsetView2DType<MemorySpace>(kv.data(), {gys * dof, gxs * dof}, {(gys + gym) * dof, (gxs + gxm) * dof});
113 PetscScalarKokkosViewType<MemorySpace> kv;
119 // kv = ov->view(); /* 2D OffsetView => 2D View => 1D View. Why does it not work? */
120 kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1));
121 if (overwrite) PetscCall(VecRestoreKokkosViewWrite(vec, &kv));
122 else PetscCall(VecRestoreKokkosView(vec, &kv));
130 ConstPetscScalarKokkosViewType<MemorySpace> kv;
138 PetscCall(VecGetKokkosView(vec, &kv));
139 *ov = ConstPetscScalarKokkosOffsetView2DType<MemorySpace>(kv.data(), {gys * dof, gxs * dof}, {(gys + gym) * dof, (gxs + gxm) * dof});
146 ConstPetscScalarKokkosViewType<MemorySpace> kv;
152 kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1));
153 PetscCall(VecRestoreKokkosView(vec, &kv));
162 PetscScalarKokkosViewType<MemorySpace> kv;
170 if (overwrite) PetscCall(VecGetKokkosViewWrite(vec, &kv));
171 else PetscCall(VecGetKokkosView(vec, &kv));
172 *ov = PetscScalarKokkosOffsetView3DType<MemorySpace>(kv.data(), {gzs * dof, gys * dof, gxs * dof}, {(gzs + gzm) * dof, (gys + gym) * dof, (gxs + gxm) * dof});
179 PetscScalarKokkosViewType<MemorySpace> kv;
185 kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent(2));
186 if (overwrite) PetscCall(VecRestoreKokkosViewWrite(vec, &kv));
187 else PetscCall(VecRestoreKokkosView(vec, &kv));
195 ConstPetscScalarKokkosViewType<MemorySpace> kv;
203 PetscCall(VecGetKokkosView(vec, &kv));
204 *ov = ConstPetscScalarKokkosOffsetView3DType<MemorySpace>(kv.data(), {gzs * dof, gys * dof, gxs * dof}, {(gzs + gzm) * dof, (gys + gym) * dof, (gxs + gxm) * dof});
211 ConstPetscScalarKokkosViewType<MemorySpace> kv;
217 kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent(2));
218 PetscCall(VecRestoreKokkosView(vec, &kv));
368 PetscScalarKokkosViewType<MemorySpace> kv;
376 if (overwrite) PetscCall(VecGetKokkosViewWrite(vec, &kv));
377 else PetscCall(VecGetKokkosView(vec, &kv));
378 *ov = PetscScalarKokkosOffsetView2DType<MemorySpace>(kv.data(), {gxs, 0}, {gxs + gxm, dof});
385 PetscScalarKokkosViewType<MemorySpace> kv;
391 kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1));
392 if (overwrite) PetscCall(VecRestoreKokkosViewWrite(vec, &kv));
393 else PetscCall(VecRestoreKokkosView(vec, &kv));
401 ConstPetscScalarKokkosViewType<MemorySpace> kv;
409 PetscCall(VecGetKokkosView(vec, &kv));
410 *ov = ConstPetscScalarKokkosOffsetView2DType<MemorySpace>(kv.data(), {gxs, 0}, {gxs + gxm, dof});
417 ConstPetscScalarKokkosViewType<MemorySpace> kv;
423 kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1));
424 PetscCall(VecRestoreKokkosView(vec, &kv));
433 PetscScalarKokkosViewType<MemorySpace> kv;
441 if (overwrite) PetscCall(VecGetKokkosViewWrite(vec, &kv));
442 else PetscCall(VecGetKokkosView(vec, &kv));
443 *ov = PetscScalarKokkosOffsetView3DType<MemorySpace>(kv.data(), {gys, gxs, 0}, {gys + gym, gxs + gxm, dof});
450 PetscScalarKokkosViewType<MemorySpace> kv;
456 kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent(2));
457 if (overwrite) PetscCall(VecRestoreKokkosViewWrite(vec, &kv));
458 else PetscCall(VecRestoreKokkosView(vec, &kv));
466 ConstPetscScalarKokkosViewType<MemorySpace> kv;
474 PetscCall(VecGetKokkosView(vec, &kv));
475 *ov = ConstPetscScalarKokkosOffsetView3DType<MemorySpace>(kv.data(), {gys, gxs, 0}, {gys + gym, gxs + gxm, dof});
482 ConstPetscScalarKokkosViewType<MemorySpace> kv;
488 kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent(2));
489 PetscCall(VecRestoreKokkosView(vec, &kv));
498 PetscScalarKokkosViewType<MemorySpace> kv;
506 if (overwrite) PetscCall(VecGetKokkosViewWrite(vec, &kv));
507 else PetscCall(VecGetKokkosView(vec, &kv));
508 *ov = PetscScalarKokkosOffsetView4DType<MemorySpace>(kv.data(), {gzs, gys, gxs, 0}, {gzs + gzm, gys + gym, gxs + gxm, dof});
515 PetscScalarKokkosViewType<MemorySpace> kv;
521 kv = PetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent(2) * ov->extent(3));
522 if (overwrite) PetscCall(VecRestoreKokkosViewWrite(vec, &kv));
523 else PetscCall(VecRestoreKokkosView(vec, &kv));
531 ConstPetscScalarKokkosViewType<MemorySpace> kv;
539 PetscCall(VecGetKokkosView(vec, &kv));
540 *ov = ConstPetscScalarKokkosOffsetView4DType<MemorySpace>(kv.data(), {gzs, gys, gxs, 0}, {gzs + gzm, gys + gym, gxs + gxm, dof});
547 ConstPetscScalarKokkosViewType<MemorySpace> kv;
553 kv = ConstPetscScalarKokkosViewType<MemorySpace>(ov->data(), ov->extent(0) * ov->extent(1) * ov->extent(2) * ov->extent(3));
554 PetscCall(VecRestoreKokkosView(vec, &kv));