Lines Matching refs:mat

29 PetscErrorCode RegisterMatScaleUserImpl(Mat mat)  in RegisterMatScaleUserImpl()  argument
34 PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)mat), &size)); in RegisterMatScaleUserImpl()
36 …PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatScaleUserImpl_C", MatScaleUserImpl_SeqA… in RegisterMatScaleUserImpl()
39 PetscCall(MatMPIAIJGetSeqAIJ(mat, &AA, &AB, NULL)); in RegisterMatScaleUserImpl()
40 …PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatScaleUserImpl_C", MatScaleUserImpl_MPIA… in RegisterMatScaleUserImpl()
50 PetscErrorCode DeRegisterMatScaleUserImpl(Mat mat) in DeRegisterMatScaleUserImpl() argument
55 PetscCallMPI(MPI_Comm_size(PetscObjectComm((PetscObject)mat), &size)); in DeRegisterMatScaleUserImpl()
57 PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatScaleUserImpl_C", NULL)); in DeRegisterMatScaleUserImpl()
60 PetscCall(MatMPIAIJGetSeqAIJ(mat, &AA, &AB, NULL)); in DeRegisterMatScaleUserImpl()
61 PetscCall(PetscObjectComposeFunction((PetscObject)mat, "MatScaleUserImpl_C", NULL)); in DeRegisterMatScaleUserImpl()
73 PetscErrorCode MatScaleUserImpl(Mat mat, PetscScalar a) in MatScaleUserImpl() argument
78 PetscCall(PetscObjectQueryFunction((PetscObject)mat, "MatScaleUserImpl_C", &f)); in MatScaleUserImpl()
79 if (f) PetscCall((*f)(mat, a)); in MatScaleUserImpl()
87 Mat mat; in main() local
99 PetscCall(MatCreate(PETSC_COMM_WORLD, &mat)); in main()
100 PetscCall(MatSetSizes(mat, PETSC_DECIDE, PETSC_DECIDE, m * n, m * n)); in main()
101 PetscCall(MatSetType(mat, MATAIJ)); in main()
102 PetscCall(MatSetUp(mat)); in main()
106 PetscCall(RegisterMatScaleUserImpl(mat)); in main()
115 PetscCall(MatSetValues(mat, 1, &Ii, 1, &J, &v, INSERT_VALUES)); in main()
119 PetscCall(MatSetValues(mat, 1, &Ii, 1, &J, &v, INSERT_VALUES)); in main()
123 PetscCall(MatSetValues(mat, 1, &Ii, 1, &J, &v, INSERT_VALUES)); in main()
127 PetscCall(MatSetValues(mat, 1, &Ii, 1, &J, &v, INSERT_VALUES)); in main()
130 PetscCall(MatSetValues(mat, 1, &Ii, 1, &Ii, &v, INSERT_VALUES)); in main()
133 PetscCall(MatAssemblyBegin(mat, MAT_FINAL_ASSEMBLY)); in main()
134 PetscCall(MatAssemblyEnd(mat, MAT_FINAL_ASSEMBLY)); in main()
138 PetscCall(MatView(mat, PETSC_VIEWER_STDOUT_WORLD)); in main()
139 PetscCall(MatScaleUserImpl(mat, none)); in main()
141 PetscCall(MatView(mat, PETSC_VIEWER_STDOUT_WORLD)); in main()
145 PetscCall(DeRegisterMatScaleUserImpl(mat)); in main()
146 PetscCall(MatDestroy(&mat)); in main()