Searched refs:akok (Results 1 – 2 of 2) sorted by relevance
313 Mat_SeqAIJKokkos *akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr); in MatSeqAIJKokkosGenerateTranspose_Private() local315 KokkosCsrMatrix &T = akok->csrmatT; in MatSeqAIJKokkosGenerateTranspose_Private()318 PetscCheck(akok, PETSC_COMM_WORLD, PETSC_ERR_PLIB, "Unexpected NULL (Mat_SeqAIJKokkos*)A->spptr"); in MatSeqAIJKokkosGenerateTranspose_Private()319 …PetscCall(KokkosDualViewSyncDevice(akok->a_dual, PetscGetKokkosExecutionSpace())); // Sync A's val… in MatSeqAIJKokkosGenerateTranspose_Private()321 const auto &Aa = akok->a_dual.view_device(); in MatSeqAIJKokkosGenerateTranspose_Private()324 *csrmatT = akok->csrmat; in MatSeqAIJKokkosGenerateTranspose_Private()328 …if (!akok->transpose_updated) { // if the value is out of date, update the cached versi… in MatSeqAIJKokkosGenerateTranspose_Private()329 const auto &perm = akok->transpose_perm; // get the permutation array in MatSeqAIJKokkosGenerateTranspose_Private()338 akok->transpose_perm = perm; // cache the perm in this matrix for reuse in MatSeqAIJKokkosGenerateTranspose_Private()341 akok->transpose_updated = PETSC_TRUE; in MatSeqAIJKokkosGenerateTranspose_Private()[all …]
743 …Mat_SeqAIJKokkos *akok = static_cast<Mat_SeqAIJKokkos *>(A->spptr), *bkok = static_cast<Mat_SeqAIJ… in MatMPIAIJKokkosBcastBegin() local950 PetscCall(KokkosDualViewSyncDevice(akok->a_dual, PetscGetKokkosExecutionSpace())); in MatMPIAIJKokkosBcastBegin()954 const auto &Aa = akok->a_dual.view_device(); in MatMPIAIJKokkosBcastBegin()956 const auto &Ai = akok->i_dual.view_device(); in MatMPIAIJKokkosBcastBegin()