Lines Matching refs:A
15 PetscCall(MatCreate(PETSC_COMM_SELF, &b->A)); in MatMPISELLSetPreallocation_MPISELLCUDA()
16 PetscCall(MatBindToCPU(b->A, B->boundtocpu)); in MatMPISELLSetPreallocation_MPISELLCUDA()
17 PetscCall(MatSetSizes(b->A, B->rmap->n, B->cmap->n, B->rmap->n, B->cmap->n)); in MatMPISELLSetPreallocation_MPISELLCUDA()
18 PetscCall(MatSetType(b->A, MATSEQSELLCUDA)); in MatMPISELLSetPreallocation_MPISELLCUDA()
24 PetscCall(MatSeqSELLSetPreallocation(b->A, d_rlenmax, d_rlen)); in MatMPISELLSetPreallocation_MPISELLCUDA()
37 static PetscErrorCode MatAssemblyEnd_MPISELLCUDA(Mat A, MatAssemblyType mode) in MatAssemblyEnd_MPISELLCUDA() argument
40 PetscCall(MatAssemblyEnd_MPISELL(A, mode)); in MatAssemblyEnd_MPISELLCUDA()
41 …if (!A->was_assembled && mode == MAT_FINAL_ASSEMBLY) PetscCall(VecSetType(((Mat_MPISELL *)A->data)… in MatAssemblyEnd_MPISELLCUDA()
45 static PetscErrorCode MatDestroy_MPISELLCUDA(Mat A) in MatDestroy_MPISELLCUDA() argument
48 PetscCall(MatDestroy_MPISELL(A)); in MatDestroy_MPISELLCUDA()
49 PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMPISELLSetPreallocation_C", NULL)); in MatDestroy_MPISELLCUDA()
56 Mat A; in MatConvert_MPISELL_MPISELLCUDA() local
62 A = *newmat; in MatConvert_MPISELL_MPISELLCUDA()
63 A->boundtocpu = PETSC_FALSE; in MatConvert_MPISELL_MPISELLCUDA()
64 PetscCall(PetscFree(A->defaultvectype)); in MatConvert_MPISELL_MPISELLCUDA()
65 PetscCall(PetscStrallocpy(VECCUDA, &A->defaultvectype)); in MatConvert_MPISELL_MPISELLCUDA()
67 a = (Mat_MPISELL *)A->data; in MatConvert_MPISELL_MPISELLCUDA()
68 if (a->A) PetscCall(MatSetType(a->A, MATSEQSELLCUDA)); in MatConvert_MPISELL_MPISELLCUDA()
72 A->ops->assemblyend = MatAssemblyEnd_MPISELLCUDA; in MatConvert_MPISELL_MPISELLCUDA()
73 A->ops->setfromoptions = MatSetFromOptions_MPISELLCUDA; in MatConvert_MPISELL_MPISELLCUDA()
74 A->ops->destroy = MatDestroy_MPISELLCUDA; in MatConvert_MPISELL_MPISELLCUDA()
76 PetscCall(PetscObjectChangeTypeName((PetscObject)A, MATMPISELLCUDA)); in MatConvert_MPISELL_MPISELLCUDA()
77 …PetscCall(PetscObjectComposeFunction((PetscObject)A, "MatMPISELLSetPreallocation_C", MatMPISELLSet… in MatConvert_MPISELL_MPISELLCUDA()
81 PETSC_EXTERN PetscErrorCode MatCreate_MPISELLCUDA(Mat A) in MatCreate_MPISELLCUDA() argument
85 PetscCall(MatCreate_MPISELL(A)); in MatCreate_MPISELLCUDA()
86 PetscCall(MatConvert_MPISELL_MPISELLCUDA(A, MATMPISELLCUDA, MAT_INPLACE_MATRIX, &A)); in MatCreate_MPISELLCUDA()
136 …, PetscInt N, PetscInt d_nz, const PetscInt d_nnz[], PetscInt o_nz, const PetscInt o_nnz[], Mat *A) in MatCreateSELLCUDA() argument
141 PetscCall(MatCreate(comm, A)); in MatCreateSELLCUDA()
142 PetscCall(MatSetSizes(*A, m, n, M, N)); in MatCreateSELLCUDA()
145 PetscCall(MatSetType(*A, MATMPISELLCUDA)); in MatCreateSELLCUDA()
146 PetscCall(MatMPISELLSetPreallocation(*A, d_nz, d_nnz, o_nz, o_nnz)); in MatCreateSELLCUDA()
148 PetscCall(MatSetType(*A, MATSEQSELLCUDA)); in MatCreateSELLCUDA()
149 PetscCall(MatSeqSELLSetPreallocation(*A, d_nz, d_nnz)); in MatCreateSELLCUDA()