Home
last modified time | relevance | path

Searched refs:Atot (Results 1 – 6 of 6) sorted by relevance

/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx1345 PetscCount Atot; member
1354 Atot = coo_h->Atot; in MatCOOStruct_SeqAIJKokkos()
1356 …os::create_mirror_view_and_copy(DefaultMemorySpace(), PetscCountKokkosViewHost(coo_h->perm, Atot)); in MatCOOStruct_SeqAIJKokkos()
/petsc/src/mat/impls/aij/seq/
H A Daij.h173 …PetscCount Atot; /* Total number of valid (i.e., w/ non-negative indices) entries in the COO arra… member
H A Daij.c4732 coo->Atot = coo_n - nneg; // Annz is seqaij->nz, so no need to record that again in MatSetPreallocationCOO_SeqAIJ()
/petsc/src/mat/impls/aij/mpi/
H A Dmpiaij.c6174 …PetscCount Atot = 0, Btot = 0; /* Total number of nonzeros in the diagonal and off-diagonal block… in MatSplitEntries_Internal() local
6209 Atot += mid - k; in MatSplitEntries_Internal()
6234 PetscCall(PetscMalloc1(Atot, &Aperm)); in MatSplitEntries_Internal()
6240 Ajmap[0] = Bjmap[0] = Atot = Btot = Annz = Bnnz = 0; in MatSplitEntries_Internal()
6245 …PetscCall(PetscArraycpy(PetscSafePointerPlusOffset(Aperm, Atot), PetscSafePointerPlusOffset(perm, … in MatSplitEntries_Internal()
6247 Atot += mid - k; in MatSplitEntries_Internal()
6274 *Atot_ = Atot; in MatSplitEntries_Internal()
/petsc/src/mat/impls/aij/seq/seqhipsparse/
H A Daijhipsparse.hip.cxx3762 PetscCallHIP(hipMalloc((void **)&coo_d->perm, coo_h->Atot * sizeof(PetscCount))); in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
3763 …PetscCallHIP(hipMemcpy(coo_d->perm, coo_h->perm, coo_h->Atot * sizeof(PetscCount), hipMemcpyHostTo… in MatSetPreallocationCOO_SeqAIJHIPSPARSE()
/petsc/src/mat/impls/aij/seq/seqcusparse/
H A Daijcusparse.cu4405 PetscCallCUDA(cudaMalloc((void **)&coo_d->perm, coo_h->Atot * sizeof(PetscCount))); in MatSetPreallocationCOO_SeqAIJCUSPARSE()
4406 …PetscCallCUDA(cudaMemcpy(coo_d->perm, coo_h->perm, coo_h->Atot * sizeof(PetscCount), cudaMemcpyHos… in MatSetPreallocationCOO_SeqAIJCUSPARSE()