Searched refs:Atot2 (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 95 PetscCallHIP(hipMalloc((void **)&coo_d->Aperm2, coo_h->Atot2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
109 …PetscCallHIP(hipMemcpy(coo_d->Aperm2, coo_h->Aperm2, coo_h->Atot2 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 95 PetscCallCUDA(cudaMalloc((void **)&coo_d->Aperm2, coo_h->Atot2 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
109 …PetscCallCUDA(cudaMemcpy(coo_d->Aperm2, coo_h->Aperm2, coo_h->Atot2 * sizeof(PetscCount), cudaMemc… in MatSetPreallocationCOO_MPIAIJCUSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.h | 81 …PetscCount Atot1, Atot2, Btot1, Btot2; /* Total local (tot1) and remote (tot2) entries (which mi… member
|
| H A D | mpiaij.c | 6558 PetscCount Annz2, Bnnz2, Atot2, Btot2; in MatSetPreallocationCOO_MPIAIJ() local
6563 …PetscCall(MatSplitEntries_Internal(mat, n2, i2, j2, perm2, rowBegin2, rowMid2, rowEnd2, &Atot2, &A… in MatSetPreallocationCOO_MPIAIJ()
6681 coo->Atot2 = Atot2; in MatSetPreallocationCOO_MPIAIJ()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1530 …= Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aperm2, coo_h->Atot2)); in MatCOOStruct_MPIAIJKokkos()
|