Searched refs:aijkok (Results 1 – 1 of 1) sorted by relevance
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 66 Mat_SeqAIJKokkos *aijkok; in MatAssemblyEnd_SeqAIJKokkos() local
73 aijkok = static_cast<Mat_SeqAIJKokkos *>(A->spptr); in MatAssemblyEnd_SeqAIJKokkos()
79 …if (!aijkok || aijkok->nonzerostate != A->nonzerostate) { /* aijkok might not exist yet or nonzero… in MatAssemblyEnd_SeqAIJKokkos()
80 …if (aijkok && aijkok->host_aij_allocated_by_kokkos) { /* Avoid accidentally freeing much needed … in MatAssemblyEnd_SeqAIJKokkos()
81 … PetscCall(PetscShmgetAllocateArray(aijkok->nrows() + 1, sizeof(PetscInt), (void **)&aijseq->i)); in MatAssemblyEnd_SeqAIJKokkos()
82 PetscCall(PetscShmgetAllocateArray(aijkok->nnz(), sizeof(PetscInt), (void **)&aijseq->j)); in MatAssemblyEnd_SeqAIJKokkos()
83 PetscCall(PetscShmgetAllocateArray(aijkok->nnz(), sizeof(PetscInt), (void **)&aijseq->a)); in MatAssemblyEnd_SeqAIJKokkos()
84 PetscCall(PetscArraycpy(aijseq->i, aijkok->i_host_data(), aijkok->nrows() + 1)); in MatAssemblyEnd_SeqAIJKokkos()
85 PetscCall(PetscArraycpy(aijseq->j, aijkok->j_host_data(), aijkok->nnz())); in MatAssemblyEnd_SeqAIJKokkos()
86 PetscCall(PetscArraycpy(aijseq->a, aijkok->a_host_data(), aijkok->nnz())); in MatAssemblyEnd_SeqAIJKokkos()
[all …]
|