Lines Matching refs:aj

118   PetscInt      *aj, *ai;  in MatSeqAIJMKL_create_mkl_handle()
142 aj = a->j; /* aj[k] gives column index for element aa[k]. */ in MatSeqAIJMKL_create_mkl_handle()
148 …SE_INDEX_BASE_ZERO, (MKL_INT)m, (MKL_INT)n, (MKL_INT *)ai, (MKL_INT *)(ai + 1), (MKL_INT *)aj, aa); in MatSeqAIJMKL_create_mkl_handle()
167 PetscInt *aj, *ai, *unused; in MatSeqAIJMKL_setup_structure_from_mkl_handle() local
174 …ndexing, (MKL_INT *)&m, (MKL_INT *)&n, (MKL_INT **)&ai, (MKL_INT **)&unused, (MKL_INT **)&aj, &aa); 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 … (MKL_INT *)&nrows, (MKL_INT *)&ncols, (MKL_INT **)&ai, (MKL_INT **)&unused, (MKL_INT **)&aj, &aa); 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 … (MKL_INT *)&nrows, (MKL_INT *)&ncols, (MKL_INT **)&ai, (MKL_INT **)&unused, (MKL_INT **)&aj, &aa); in MatSeqAIJMKL_view_mkl_handle()
287 …PetscCall(PetscViewerASCIIPrintf(viewer, "(%" PetscInt_FMT ", %g) ", aj[k], PetscRealPart(aa[k]))… in MatSeqAIJMKL_view_mkl_handle()
289 PetscCall(PetscViewerASCIIPrintf(viewer, "(%" PetscInt_FMT ", NULL)", aj[k])); in MatSeqAIJMKL_view_mkl_handle()
347 const PetscInt *aj, *ai; in MatMult_SeqAIJMKL() local
358 aj = a->j; /* aj[k] gives column index for element aa[k]. */ 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
431 aj = a->j; /* aj[k] gives column index for element aa[k]. */ 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
506 aj = a->j; /* aj[k] gives column index for element aa[k]. */ 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
604 aj = a->j; /* aj[k] gives column index for element aa[k]. */ 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()