Lines Matching refs:ai
118 PetscInt *aj, *ai; in MatSeqAIJMKL_create_mkl_handle()
144 ai = a->i; /* ai[k] is the position in aa and aj where row k starts. */ in MatSeqAIJMKL_create_mkl_handle()
148 …->csrA, SPARSE_INDEX_BASE_ZERO, (MKL_INT)m, (MKL_INT)n, (MKL_INT *)ai, (MKL_INT *)(ai + 1), (MKL_I… in MatSeqAIJMKL_create_mkl_handle()
167 PetscInt *aj, *ai, *unused; in MatSeqAIJMKL_setup_structure_from_mkl_handle() local
174 …_x_export_csr, csrA, &indexing, (MKL_INT *)&m, (MKL_INT *)&n, (MKL_INT **)&ai, (MKL_INT **)&unused… in MatSeqAIJMKL_setup_structure_from_mkl_handle()
177 aj = ai = NULL; in MatSeqAIJMKL_setup_structure_from_mkl_handle()
188 PetscCall(MatSeqAIJSetPreallocationCSR(A, ai, aj, NULL)); in MatSeqAIJMKL_setup_structure_from_mkl_handle()
227 PetscInt *ai, *aj, *unused; in MatSeqAIJMKL_update_from_mkl_handle() local
237 …ijmkl->csrA, &indexing, (MKL_INT *)&nrows, (MKL_INT *)&ncols, (MKL_INT **)&ai, (MKL_INT **)&unused… in MatSeqAIJMKL_update_from_mkl_handle()
242 nz = ai[i + 1] - ai[i]; in MatSeqAIJMKL_update_from_mkl_handle()
243 PetscCall(MatSetValues_SeqAIJ(A, 1, &i, nz, aj + ai[i], aa + ai[i], INSERT_VALUES)); in MatSeqAIJMKL_update_from_mkl_handle()
264 PetscInt *ai, *aj, *unused; in MatSeqAIJMKL_view_mkl_handle() local
279 …ijmkl->csrA, &indexing, (MKL_INT *)&nrows, (MKL_INT *)&ncols, (MKL_INT **)&ai, (MKL_INT **)&unused… in MatSeqAIJMKL_view_mkl_handle()
284 nz = ai[i + 1] - ai[i]; in MatSeqAIJMKL_view_mkl_handle()
347 const PetscInt *aj, *ai; in MatMult_SeqAIJMKL() local
360 ai = a->i; /* ai[k] is the position in aa and aj where row k starts. */ in MatMult_SeqAIJMKL()
363 mkl_xcsrmv(&transa, &m, &n, &alpha, matdescra, aa, aj, ai, ai + 1, x, &beta, y); in MatMult_SeqAIJMKL()
420 const PetscInt *aj, *ai; in MatMultTranspose_SeqAIJMKL() local
433 ai = a->i; /* ai[k] is the position in aa and aj where row k starts. */ in MatMultTranspose_SeqAIJMKL()
436 mkl_xcsrmv(&transa, &m, &n, &alpha, matdescra, aa, aj, ai, ai + 1, x, &beta, y); in MatMultTranspose_SeqAIJMKL()
491 const PetscInt *aj, *ai; in MatMultAdd_SeqAIJMKL() local
508 ai = a->i; /* ai[k] is the position in aa and aj where row k starts. */ in MatMultAdd_SeqAIJMKL()
514 mkl_xcsrmv(&transa, &m, &n, &alpha, matdescra, aa, aj, ai, ai + 1, x, &beta, z); in MatMultAdd_SeqAIJMKL()
519 mkl_xcsrmv(&transa, &m, &n, &alpha, matdescra, aa, aj, ai, ai + 1, x, &beta, z); in MatMultAdd_SeqAIJMKL()
589 const PetscInt *aj, *ai; in MatMultTransposeAdd_SeqAIJMKL() local
606 ai = a->i; /* ai[k] is the position in aa and aj where row k starts. */ in MatMultTransposeAdd_SeqAIJMKL()
612 mkl_xcsrmv(&transa, &m, &n, &alpha, matdescra, aa, aj, ai, ai + 1, x, &beta, z); in MatMultTransposeAdd_SeqAIJMKL()
617 mkl_xcsrmv(&transa, &m, &n, &alpha, matdescra, aa, aj, ai, ai + 1, x, &beta, z); in MatMultTransposeAdd_SeqAIJMKL()