Lines Matching refs:mat_coo

64 static PetscErrorCode MatCeedAssemblePointBlockDiagonalCOO(Mat mat_ceed, Mat mat_coo) {  in MatCeedAssemblePointBlockDiagonalCOO()  argument
75 if (ctx->mats_assembled_pbd[i] == mat_coo) index = i; in MatCeedAssemblePointBlockDiagonalCOO()
89 PetscCall(PetscLogEventBegin(MATCEED_SETUP_PBDIAGONAL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
90 PetscCall(PetscLogEventBegin(MATCEED_SETUP_PBDIAGONAL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
92 PetscCall(PetscLogEventEnd(MATCEED_SETUP_PBDIAGONAL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
95 PetscCall(MatSetPreallocationCOOLocal(mat_coo, num_entries, rows_petsc, cols_petsc)); in MatCeedAssemblePointBlockDiagonalCOO()
100 ctx->mats_assembled_pbd[ctx->num_mats_assembled_pbd - 1] = mat_coo; in MatCeedAssemblePointBlockDiagonalCOO()
101 PetscCall(PetscLogEventEnd(MATCEED_SETUP_PBDIAGONAL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
107 PetscCall(PetscLogEventBegin(MATCEED_ASSEMBLE_PBDIAGONAL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
108 PetscCall(MatAssemblyBegin(mat_coo, MAT_FINAL_ASSEMBLY)); in MatCeedAssemblePointBlockDiagonalCOO()
115 PetscCall(MatGetType(mat_coo, &mat_type)); in MatCeedAssemblePointBlockDiagonalCOO()
120 … PetscCall(PetscLogEventBegin(MATCEED_ASSEMBLE_PBDIAGONAL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
122 PetscCall(PetscLogEventEnd(MATCEED_ASSEMBLE_PBDIAGONAL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
124 PetscCall(MatSetValuesCOO(mat_coo, values, INSERT_VALUES)); in MatCeedAssemblePointBlockDiagonalCOO()
126 if (is_spd_known) PetscCall(MatSetOption(mat_coo, MAT_SPD, is_spd)); in MatCeedAssemblePointBlockDiagonalCOO()
129 PetscCall(MatAssemblyEnd(mat_coo, MAT_FINAL_ASSEMBLY)); in MatCeedAssemblePointBlockDiagonalCOO()
130 PetscCall(PetscLogEventEnd(MATCEED_ASSEMBLE_PBDIAGONAL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
598 PetscErrorCode MatCeedCreateMatCOO(Mat mat_ceed, Mat *mat_coo) { in MatCeedCreateMatCOO() argument
634 PetscCall(DMCreateMatrix(ctx->dm_x, mat_coo)); in MatCeedCreateMatCOO()
652 PetscErrorCode MatCeedSetPreallocationCOO(Mat mat_ceed, Mat mat_coo) { in MatCeedSetPreallocationCOO() argument
670 PetscCall(PetscLogEventBegin(MATCEED_SETUP_FULL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedSetPreallocationCOO()
671 PetscCall(PetscLogEventBegin(MATCEED_SETUP_FULL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedSetPreallocationCOO()
673 PetscCall(PetscLogEventEnd(MATCEED_SETUP_FULL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedSetPreallocationCOO()
676 PetscCall(MatSetPreallocationCOOLocal(mat_coo, num_entries, rows_petsc, cols_petsc)); in MatCeedSetPreallocationCOO()
681 ctx->mats_assembled_full[ctx->num_mats_assembled_full - 1] = mat_coo; in MatCeedSetPreallocationCOO()
682 PetscCall(PetscLogEventEnd(MATCEED_SETUP_FULL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedSetPreallocationCOO()
700 PetscErrorCode MatCeedAssembleCOO(Mat mat_ceed, Mat mat_coo) { in MatCeedAssembleCOO() argument
711 if (ctx->mats_assembled_full[i] == mat_coo) index = i; in MatCeedAssembleCOO()
713 if (index == -1) PetscCall(MatCeedSetPreallocationCOO(mat_ceed, mat_coo)); in MatCeedAssembleCOO()
717 PetscCall(PetscLogEventBegin(MATCEED_ASSEMBLE_FULL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssembleCOO()
718 PetscCall(MatAssemblyBegin(mat_coo, MAT_FINAL_ASSEMBLY)); in MatCeedAssembleCOO()
725 PetscCall(MatGetType(mat_coo, &mat_type)); in MatCeedAssembleCOO()
730 PetscCall(PetscLogEventBegin(MATCEED_ASSEMBLE_FULL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssembleCOO()
732 PetscCall(PetscLogEventEnd(MATCEED_ASSEMBLE_FULL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssembleCOO()
734 PetscCall(MatSetValuesCOO(mat_coo, values, INSERT_VALUES)); in MatCeedAssembleCOO()
736 if (is_spd_known) PetscCall(MatSetOption(mat_coo, MAT_SPD, is_spd)); in MatCeedAssembleCOO()
739 PetscCall(MatAssemblyEnd(mat_coo, MAT_FINAL_ASSEMBLY)); in MatCeedAssembleCOO()
740 PetscCall(PetscLogEventEnd(MATCEED_ASSEMBLE_FULL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssembleCOO()