| /petsc/src/vec/vec/impls/mpi/kokkos/ |
| H A D | mpikok.kokkos.cxx | 379 …tAndArrays_Private(PetscLayout map, const PetscScalar harray[], const PetscScalar darray[], Vec *v) in VecCreateMPIKokkosWithLayoutAndArrays_Private() argument
384 if (map->n > 0) PetscCheck(darray, map->comm, PETSC_ERR_ARG_WRONG, "darray cannot be NULL"); in VecCreateMPIKokkosWithLayoutAndArrays_Private()
386 …PetscCheck(harray == darray, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "harray and darray must be the … in VecCreateMPIKokkosWithLayoutAndArrays_Private()
391 …tr = new Vec_Kokkos(map->n, const_cast<PetscScalar *>(harray), const_cast<PetscScalar *>(darray))); in VecCreateMPIKokkosWithLayoutAndArrays_Private()
429 …osWithArray(MPI_Comm comm, PetscInt bs, PetscInt n, PetscInt N, const PetscScalar darray[], Vec *v) in VecCreateMPIKokkosWithArray() argument
446 harray = const_cast<PetscScalar *>(darray); in VecCreateMPIKokkosWithArray()
456 veckok = new Vec_Kokkos(n, harray, const_cast<PetscScalar *>(darray)); in VecCreateMPIKokkosWithArray()
489 …etscInt bs, PetscInt n, PetscInt N, const PetscScalar harray[], const PetscScalar darray[], Vec *v) in VecCreateMPIKokkosWithArrays_Private() argument
497 PetscCheck(darray, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "darray cannot be NULL"); in VecCreateMPIKokkosWithArrays_Private()
499 …if (std::is_same<DefaultMemorySpace, HostMirrorMemorySpace>::value) PetscCheck(harray == darray, P… in VecCreateMPIKokkosWithArrays_Private()
[all …]
|
| /petsc/src/vec/vec/impls/seq/kokkos/ |
| H A D | veckok.kokkos.cxx | 1727 …reateSeqKokkosWithArray(MPI_Comm comm, PetscInt bs, PetscInt n, const PetscScalar darray[], Vec *v) in VecCreateSeqKokkosWithArray() argument
1742 if (!darray) { /* Allocate memory ourself if user provided NULL */ in VecCreateSeqKokkosWithArray()
1747 harray = const_cast<PetscScalar *>(darray); in VecCreateSeqKokkosWithArray()
1755 PetscCallCXX(veckok = new Vec_Kokkos{n, harray, const_cast<PetscScalar *>(darray)}); in VecCreateSeqKokkosWithArray()
1779 …tAndArrays_Private(PetscLayout map, const PetscScalar harray[], const PetscScalar darray[], Vec *v) in VecCreateSeqKokkosWithLayoutAndArrays_Private() argument
1784 if (map->n > 0) PetscCheck(darray, map->comm, PETSC_ERR_ARG_WRONG, "darray cannot be NULL"); in VecCreateSeqKokkosWithLayoutAndArrays_Private()
1786 …PetscCheck(harray == darray, PETSC_COMM_SELF, PETSC_ERR_ARG_WRONG, "harray and darray must be the … in VecCreateSeqKokkosWithLayoutAndArrays_Private()
1791 …tr = new Vec_Kokkos(map->n, const_cast<PetscScalar *>(harray), const_cast<PetscScalar *>(darray))); in VecCreateSeqKokkosWithLayoutAndArrays_Private()
1823 …Comm comm, PetscInt bs, PetscInt n, const PetscScalar harray[], const PetscScalar darray[], Vec *v) in VecCreateSeqKokkosWithArrays_Private() argument
1833 PetscCall(VecCreateSeqKokkosWithLayoutAndArrays_Private(map, harray, darray, v)); in VecCreateSeqKokkosWithArrays_Private()
|
| /petsc/src/mat/impls/scalapack/ |
| H A D | matscalapack.c | 433 PetscScalar *darray, *d2d, v; in MatGetDiagonal_ScaLAPACK() local
438 PetscCall(VecGetArray(D, &darray)); in MatGetDiagonal_ScaLAPACK()
465 …PetscCallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&a->M, &one, d2d, &one, &one, d2desc, darray, &o… in MatGetDiagonal_ScaLAPACK()
492 …PetscCallBLAS("SCALAPACKgemr2d", SCALAPACKgemr2d_(&one, &a->N, d2d, &one, &one, d2desc, darray, &o… in MatGetDiagonal_ScaLAPACK()
496 PetscCall(VecRestoreArray(D, &darray)); in MatGetDiagonal_ScaLAPACK()
|
| /petsc/src/mat/impls/aij/seq/seqcusparse/ |
| H A D | aijcusparse.cu | 3759 PetscScalar *darray; in MatGetDiagonal_SeqAIJCUSPARSE() local
3768 PetscCall(VecCUDAGetArrayWrite(diag, &darray)); in MatGetDiagonal_SeqAIJCUSPARSE()
3769 …ow_offsets->data().get(), mat->column_indices->data().get(), mat->values->data().get(), n, darray); in MatGetDiagonal_SeqAIJCUSPARSE()
3771 PetscCall(VecCUDARestoreArrayWrite(diag, &darray)); in MatGetDiagonal_SeqAIJCUSPARSE()
|