Searched refs:d_z (Results 1 – 2 of 2) sorted by relevance
| /petsc/src/ts/utils/dmplexlandau/kokkos/ |
| H A D | landau.kokkos.cxx | 415 const PetscReal *d_z = (LANDAU_DIM == 3) ? d_z_k->data() : NULL; in LandauKokkosJacobian() local
607 …const PetscReal vj[3] = {d_x[jpidx_glb], d_y[jpidx_glb], d_z ? d_z[jpidx_glb] : … in LandauKokkosJacobian()
619 …PetscReal U[3][3], z = d_z[ipidx], mask = (PetscAbs(vj[0]-x) < 100*PETSC_SQRT_MACHINE_EPSILON && P… in LandauKokkosJacobian()
|
| /petsc/src/vec/vec/impls/seq/cupm/ |
| H A D | vecseqcupm_impl.hpp | 1272 PetscScalar *d_z; in MDot_() local
1278 PetscCall(PetscDeviceMalloc(dctx, PETSC_MEMTYPE_CUPM(), nv, &d_z)); in MDot_()
1289 …sXdot(handle, n, DeviceArrayRead(sub, yin[i]), 1, xptr.cupmdata(), 1, cupmScalarPtrCast(d_z + i))); in MDot_()
1294 PetscCall(PetscCUPMMemcpyAsync(z, d_z, nv, cupmMemcpyDeviceToHost, stream)); in MDot_()
1295 PetscCall(PetscDeviceFree(dctx, d_z)); in MDot_()
|