Lines Matching refs:Ba
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()
233 PetscCall(MatSeqAIJHIPSPARSERestoreArray(B, &Ba)); in MatSetValuesCOO_MPIAIJHIPSPARSE()