Searched refs:Atot1 (Results 1 – 5 of 5) sorted by relevance
| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 90 PetscCallHIP(hipMalloc((void **)&coo_d->Aperm1, coo_h->Atot1 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
104 …PetscCallHIP(hipMemcpy(coo_d->Aperm1, coo_h->Aperm1, coo_h->Atot1 * sizeof(PetscCount), hipMemcpyH… in MatSetPreallocationCOO_MPIAIJHIPSPARSE()
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 90 PetscCallCUDA(cudaMalloc((void **)&coo_d->Aperm1, coo_h->Atot1 * sizeof(PetscCount))); in MatSetPreallocationCOO_MPIAIJCUSPARSE()
104 …PetscCallCUDA(cudaMemcpy(coo_d->Aperm1, coo_h->Aperm1, coo_h->Atot1 * 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 | 6555 PetscCount Annz1, Bnnz1, Atot1, Btot1; in MatSetPreallocationCOO_MPIAIJ() local
6562 …PetscCall(MatSplitEntries_Internal(mat, rem, i1, j1, perm1, rowBegin1, rowMid1, rowEnd1, &Atot1, &… in MatSetPreallocationCOO_MPIAIJ()
6680 coo->Atot1 = Atot1; in MatSetPreallocationCOO_MPIAIJ()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 1525 …= Kokkos::create_mirror_view_and_copy(exec, PetscCountKokkosViewHost(coo_h->Aperm1, coo_h->Atot1)); in MatCOOStruct_MPIAIJKokkos()
|