| /petsc/src/mat/impls/aij/mpi/mpihipsparse/ |
| H A D | mpiaijhipsparse.hip.cxx | 127 …alar Aa[], PetscCount Bnnz, const PetscCount Bjmap1[], const PetscCount Bperm1[], PetscScalar Ba[]) in MatAddLocalCOOValues() argument
139 Ba[i] = (imode == INSERT_VALUES ? 0.0 : Ba[i]) + sum; in MatAddLocalCOOValues()
144 … const PetscCount Bimap2[], const PetscCount Bjmap2[], const PetscCount Bperm2[], PetscScalar Ba[]) in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
162 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJHIPSPARSE() local
199 PetscCall(MatSeqAIJHIPSPARSEGetArrayWrite(B, &Ba)); in MatSetValuesCOO_MPIAIJHIPSPARSE()
202 PetscCall(MatSeqAIJHIPSPARSEGetArray(B, &Ba)); in MatSetValuesCOO_MPIAIJHIPSPARSE()
216 …im3(256), 0, PetscDefaultHipStream, v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm1, Ba); in MatSetValuesCOO_MPIAIJHIPSPARSE()
223 …, PetscDefaultHipStream, v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjmap2, Bperm2, Ba); in MatSetValuesCOO_MPIAIJHIPSPARSE()
230 PetscCall(MatSeqAIJHIPSPARSERestoreArrayWrite(B, &Ba)); in MatSetValuesCOO_MPIAIJHIPSPARSE()
[all …]
|
| /petsc/src/mat/impls/aij/mpi/mpicusparse/ |
| H A D | mpiaijcusparse.cu | 127 …alar Aa[], PetscCount Bnnz, const PetscCount Bjmap1[], const PetscCount Bperm1[], PetscScalar Ba[]) in MatAddLocalCOOValues() argument
139 Ba[i] = (imode == INSERT_VALUES ? 0.0 : Ba[i]) + sum; in MatAddLocalCOOValues()
144 … const PetscCount Bimap2[], const PetscCount Bjmap2[], const PetscCount Bperm2[], PetscScalar Ba[]) in MatAddRemoteCOOValues() argument
153 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += kv[Bperm2[k]]; in MatAddRemoteCOOValues()
162 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJCUSPARSE() local
199 PetscCall(MatSeqAIJCUSPARSEGetArrayWrite(B, &Ba)); in MatSetValuesCOO_MPIAIJCUSPARSE()
202 PetscCall(MatSeqAIJCUSPARSEGetArray(B, &Ba)); in MatSetValuesCOO_MPIAIJCUSPARSE()
216 …((Annz + Bnnz + 255) / 256), 256>>>(v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm1, Ba); in MatSetValuesCOO_MPIAIJCUSPARSE()
223 …z2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjmap2, Bperm2, Ba); in MatSetValuesCOO_MPIAIJCUSPARSE()
230 PetscCall(MatSeqAIJCUSPARSERestoreArrayWrite(B, &Ba)); in MatSetValuesCOO_MPIAIJCUSPARSE()
[all …]
|
| /petsc/src/mat/impls/sbaij/mpi/ |
| H A D | mpiaijsbaij.c | 55 Mat_SeqBAIJ *Aa = (Mat_SeqBAIJ *)mpimat->A->data, *Ba = (Mat_SeqBAIJ *)mpimat->B->data; in MatConvert_MPIBAIJ_MPISBAIJ() local
74 o_nnz[i] = Ba->i[i + 1] - Ba->i[i]; in MatConvert_MPIBAIJ_MPISBAIJ()
|
| H A D | mpisbaij.c | 1188 Mat_SeqBAIJ *Ba = (Mat_SeqBAIJ *)mat->B->data; in MatGetRow_MPISBAIJ() local
1191 tmp = Aa->i[i + 1] - Aa->i[i] + Ba->i[i + 1] - Ba->i[i]; /* row length */ in MatGetRow_MPISBAIJ()
|
| /petsc/src/mat/impls/aij/mpi/kokkos/ |
| H A D | mpiaijkok.kokkos.cxx | 632 auto &Ba = B.values; in MatMPIAIJKokkosReduceBegin() local
659 val = Ba(Bi(i) + j); in MatMPIAIJKokkosReduceBegin()
663 val = Ba(Bi(i) + j - alen); in MatMPIAIJKokkosReduceBegin()
955 const auto &Ba = bkok->a_dual.view_device(); in MatMPIAIJKokkosBcastBegin() local
986 rootBuf(disp + j) = Ba(Bi(i) + j); in MatMPIAIJKokkosBcastBegin()
990 rootBuf(disp + j) = Ba(Bi(i) + j - alen); in MatMPIAIJKokkosBcastBegin()
1581 MatScalarKokkosView Aa, Ba; in MatSetValuesCOO_MPIAIJKokkos() local
1620 PetscCall(MatSeqAIJGetKokkosViewWrite(B, &Ba)); in MatSetValuesCOO_MPIAIJKokkos()
1623 PetscCall(MatSeqAIJGetKokkosView(B, &Ba)); in MatSetValuesCOO_MPIAIJKokkos()
1642 Ba(i) = (imode == INSERT_VALUES ? 0.0 : Ba(i)) + sum; in MatSetValuesCOO_MPIAIJKokkos()
[all …]
|
| /petsc/src/mat/impls/aij/seq/kokkos/ |
| H A D | aijkok.kokkos.cxx | 1942 const MatScalar *Ba = b->a; in MatLUFactorNumeric_SeqAIJKokkos() local
2030 PetscCall(PetscArraycpy(La + Li[i], Ba + Bi[i], llen)); // entries of L in MatLUFactorNumeric_SeqAIJKokkos()
2033 Ua[Ui[i]] = 1.0 / Ba[Bdiag[i]]; // diagonal entry in MatLUFactorNumeric_SeqAIJKokkos()
2034 PetscCall(PetscArraycpy(Ua + Ui[i] + 1, Ba + Bdiag[i + 1] + 1, ulen - 1)); // entries of U in MatLUFactorNumeric_SeqAIJKokkos()
2186 const MatScalar *Ba = b->a; in MatCholeskyFactorNumeric_SeqAIJKokkos() local
2241 D[i] = Ba[Bdiag[i]]; // actually Aa[Adiag[i]] is the inverse of the diagonal in MatCholeskyFactorNumeric_SeqAIJKokkos()
2243 for (PetscInt k = 0; k < Bi[i + 1] - Bi[i] - 1; k++) Ua[Bi[i] + 1 + k] = -Ba[Bi[i] + k]; in MatCholeskyFactorNumeric_SeqAIJKokkos()
|
| /petsc/src/mat/impls/sell/mpi/ |
| H A D | mpisell.c | 1680 Mat_SeqAIJ *Aa = (Mat_SeqAIJ *)a->A->data, *Ba = (Mat_SeqAIJ *)a->B->data; in MatConvert_MPIAIJ_MPISELL() local
1686 o_nnz[i] = Ba->i[i + 1] - Ba->i[i]; in MatConvert_MPIAIJ_MPISELL()
|
| /petsc/src/mat/impls/aij/mpi/ |
| H A D | mpiaij.c | 1728 Mat_SeqAIJ *Aa = (Mat_SeqAIJ *)mat->A->data, *Ba = (Mat_SeqAIJ *)mat->B->data; in MatGetRow_MPIAIJ() local
1731 tmp = Aa->i[i + 1] - Aa->i[i] + Ba->i[i + 1] - Ba->i[i]; in MatGetRow_MPIAIJ()
6625 PetscScalar *Aa, *Ba; in MatSetPreallocationCOO_MPIAIJ() local
6630 PetscCall(PetscCalloc1(Bnnz, &Ba)); in MatSetPreallocationCOO_MPIAIJ()
6646 PetscCall(MatCreateSeqAIJWithArrays(PETSC_COMM_SELF, m, mat->cmap->N, Bi, Bj, Ba, &mpiaij->B)); in MatSetPreallocationCOO_MPIAIJ()
6709 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJ() local
6737 PetscCall(MatSeqAIJGetArray(B, &Ba)); in MatSetValuesCOO_MPIAIJ()
6753 Ba[i] = (imode == INSERT_VALUES ? 0.0 : Ba[i]) + sum; in MatSetValuesCOO_MPIAIJ()
6762 for (PetscCount k = Bjmap2[i]; k < Bjmap2[i + 1]; k++) Ba[Bimap2[i]] += recvbuf[Bperm2[k]]; in MatSetValuesCOO_MPIAIJ()
6765 PetscCall(MatSeqAIJRestoreArray(B, &Ba)); in MatSetValuesCOO_MPIAIJ()
|
| /petsc/src/mat/impls/baij/mpi/ |
| H A D | mpibaij.c | 1316 Mat_SeqBAIJ *Aa = (Mat_SeqBAIJ *)mat->A->data, *Ba = (Mat_SeqBAIJ *)mat->B->data; in MatGetRow_MPIBAIJ() local
1319 tmp = Aa->i[i + 1] - Aa->i[i] + Ba->i[i + 1] - Ba->i[i]; in MatGetRow_MPIBAIJ()
|
| /petsc/doc/ |
| H A D | petsc.bib | 9034 author = {Ibrahima Ba and Barry~F. Smith},
|