Lines Matching refs:Aa
127 …nt Annz, const PetscCount Ajmap1[], const PetscCount Aperm1[], PetscScalar Aa[], PetscCount Bnnz, … in MatAddLocalCOOValues() argument
135 Aa[i] = (imode == INSERT_VALUES ? 0.0 : Aa[i]) + sum; in MatAddLocalCOOValues()
144 …imap2[], const PetscCount Ajmap2[], const PetscCount Aperm2[], PetscScalar Aa[], PetscCount Bnnz2,… in MatAddRemoteCOOValues() argument
150 for (PetscCount k = Ajmap2[i]; k < Ajmap2[i + 1]; k++) Aa[Aimap2[i]] += kv[Aperm2[k]]; in MatAddRemoteCOOValues()
162 PetscScalar *Aa, *Ba; in MatSetValuesCOO_MPIAIJCUSPARSE() local
198 PetscCall(MatSeqAIJCUSPARSEGetArrayWrite(A, &Aa)); /* write matrix values */ in MatSetValuesCOO_MPIAIJCUSPARSE()
201 PetscCall(MatSeqAIJCUSPARSEGetArray(A, &Aa)); /* read & write matrix values */ in MatSetValuesCOO_MPIAIJCUSPARSE()
216 …<(int)((Annz + Bnnz + 255) / 256), 256>>>(v1, imode, Annz, Ajmap1, Aperm1, Aa, Bnnz, Bjmap1, Bperm… in MatSetValuesCOO_MPIAIJCUSPARSE()
223 …t)((Annz2 + Bnnz2 + 255) / 256), 256>>>(v2, Annz2, Aimap2, Ajmap2, Aperm2, Aa, Bnnz2, Bimap2, Bjma… in MatSetValuesCOO_MPIAIJCUSPARSE()
229 PetscCall(MatSeqAIJCUSPARSERestoreArrayWrite(A, &Aa)); in MatSetValuesCOO_MPIAIJCUSPARSE()
232 PetscCall(MatSeqAIJCUSPARSERestoreArray(A, &Aa)); in MatSetValuesCOO_MPIAIJCUSPARSE()