Home
last modified time | relevance | path

Searched refs:alen (Results 1 – 4 of 4) sorted by relevance

/petsc/src/mat/impls/aij/mpi/kokkos/
H A Dmpiaijkok.kokkos.cxx451 PetscInt alen = Ai[i + 1] - Ai[i], blen = Bi[i + 1] - Bi[i]; in MatMPIAIJKokkosReduceBegin() local
452 for (PetscInt j = 0; j < alen + blen; j++) { in MatMPIAIJKokkosReduceBegin()
455 } else if (j < nzLeft + alen) { in MatMPIAIJKokkosReduceBegin()
458 buf[j] = garray1[Bj[Bi[i] + j - alen]]; // right B, in global in MatMPIAIJKokkosReduceBegin()
652 PetscInt alen = Ai(i + 1) - Ai(i); in MatMPIAIJKokkosReduceBegin() local
656 Kokkos::parallel_for(Kokkos::ThreadVectorRange(t, alen + blen), [&](PetscInt j) { in MatMPIAIJKokkosReduceBegin()
660 } else if (j < nzleft + alen) { // diag A in MatMPIAIJKokkosReduceBegin()
663 val = Ba(Bi(i) + j - alen); in MatMPIAIJKokkosReduceBegin()
846 PetscInt alen = Ai[i + 1] - Ai[i], blen = Bi[i + 1] - Bi[i]; in MatMPIAIJKokkosBcastBegin() local
847 for (PetscInt j = 0; j < alen + blen; j++) { in MatMPIAIJKokkosBcastBegin()
[all …]
/petsc/src/vec/is/is/utils/
H A Disdiff.c749 PetscInt alen, clen, *cindices, *cindices2; in ISEmbed() local
757 PetscCall(ISGetLocalSize(a, &alen)); in ISEmbed()
759 PetscCall(PetscMalloc1(alen, &cindices)); in ISEmbed()
761 PetscCall(ISGlobalToLocalMappingApply(ltog, gtoltype, alen, aindices, &clen, cindices)); in ISEmbed()
764 if (clen != alen) { in ISEmbed()
/petsc/src/mat/impls/aij/seq/kokkos/
H A Daijkok.kokkos.cxx754 PetscInt coffset = ai(i) + bi(i), alen = ai(i + 1) - ai(i), blen = bi(i + 1) - bi(i); in MatSeqAIJKokkosMergeMats()
761 Kokkos::parallel_for(Kokkos::TeamThreadRange(t, alen + blen), [&](PetscInt k) { in MatSeqAIJKokkosMergeMats()
762 if (k < alen) { in MatSeqAIJKokkosMergeMats()
766 ca(coffset + k) = ba(bi(i) + k - alen); in MatSeqAIJKokkosMergeMats()
767 … cj(coffset + k) = bj(bi(i) + k - alen) + aN; /* Entries in B get new column indices in C */ in MatSeqAIJKokkosMergeMats()
783 PetscInt alen = ai(i + 1) - ai(i), blen = bi(i + 1) - bi(i); in MatSeqAIJKokkosMergeMats()
784 Kokkos::parallel_for(Kokkos::TeamThreadRange(t, alen + blen), [&](PetscInt k) { in MatSeqAIJKokkosMergeMats()
785 if (k < alen) ca(ci(i) + k) = aa(ai(i) + k); in MatSeqAIJKokkosMergeMats()
786 else ca(ci(i) + k) = ba(bi(i) + k - alen); in MatSeqAIJKokkosMergeMats()
/petsc/src/sys/objects/
H A Doptions.c2518 size_t alen, len = 0, tlen = 0; in PetscOptionsGetEList() local
2526 PetscCall(PetscStrlen(list[i], &alen)); in PetscOptionsGetEList()
2527 if (alen > len) len = alen; in PetscOptionsGetEList()