Lines Matching refs:mat_coo

72 static PetscErrorCode MatCeedAssemblePointBlockDiagonalCOO(Mat mat_ceed, Mat mat_coo) {  in MatCeedAssemblePointBlockDiagonalCOO()  argument
83 if (ctx->mats_assembled_pbd[i] == mat_coo) index = i; in MatCeedAssemblePointBlockDiagonalCOO()
97 PetscCall(PetscLogEventBegin(MATCEED_SETUP_PBDIAGONAL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
98 PetscCall(PetscLogEventBegin(MATCEED_SETUP_PBDIAGONAL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
100 PetscCall(PetscLogEventEnd(MATCEED_SETUP_PBDIAGONAL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
103 PetscCall(MatSetPreallocationCOOLocal(mat_coo, num_entries, rows_petsc, cols_petsc)); in MatCeedAssemblePointBlockDiagonalCOO()
108 ctx->mats_assembled_pbd[ctx->num_mats_assembled_pbd - 1] = mat_coo; in MatCeedAssemblePointBlockDiagonalCOO()
109 PetscCall(PetscLogEventEnd(MATCEED_SETUP_PBDIAGONAL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
115 PetscCall(PetscLogEventBegin(MATCEED_ASSEMBLE_PBDIAGONAL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
116 PetscCall(MatAssemblyBegin(mat_coo, MAT_FINAL_ASSEMBLY)); in MatCeedAssemblePointBlockDiagonalCOO()
123 PetscCall(MatGetMemTypeFromVecType(mat_coo, &mem_type_petsc)); in MatCeedAssemblePointBlockDiagonalCOO()
125 … PetscCall(PetscLogEventBegin(MATCEED_ASSEMBLE_PBDIAGONAL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
127 PetscCall(PetscLogEventEnd(MATCEED_ASSEMBLE_PBDIAGONAL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
129 PetscCall(MatSetValuesCOO(mat_coo, values, INSERT_VALUES)); in MatCeedAssemblePointBlockDiagonalCOO()
131 if (is_spd_known) PetscCall(MatSetOption(mat_coo, MAT_SPD, is_spd)); in MatCeedAssemblePointBlockDiagonalCOO()
134 PetscCall(MatAssemblyEnd(mat_coo, MAT_FINAL_ASSEMBLY)); in MatCeedAssemblePointBlockDiagonalCOO()
135 PetscCall(PetscLogEventEnd(MATCEED_ASSEMBLE_PBDIAGONAL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssemblePointBlockDiagonalCOO()
649 PetscErrorCode MatCeedCreateMatCOO(Mat mat_ceed, Mat *mat_coo) { in MatCeedCreateMatCOO() argument
664 PetscCall(DMCreateMatrix(ctx->dm_x, mat_coo)); in MatCeedCreateMatCOO()
682 PetscErrorCode MatCeedSetPreallocationCOO(Mat mat_ceed, Mat mat_coo) { in MatCeedSetPreallocationCOO() argument
698 PetscCall(PetscLogEventBegin(MATCEED_SETUP_FULL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedSetPreallocationCOO()
699 …PetscCall(PetscObjectQuery((PetscObject)mat_coo, "__PETSc_MatCOOStruct_Host", (PetscObject *)&coo_… in MatCeedSetPreallocationCOO()
706 PetscCall(PetscLogEventBegin(MATCEED_SETUP_FULL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedSetPreallocationCOO()
708 PetscCall(PetscLogEventEnd(MATCEED_SETUP_FULL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedSetPreallocationCOO()
711 PetscCall(MatSetPreallocationCOOLocal(mat_coo, num_entries, rows_petsc, cols_petsc)); in MatCeedSetPreallocationCOO()
720 ctx->mats_assembled_full[ctx->num_mats_assembled_full - 1] = mat_coo; in MatCeedSetPreallocationCOO()
721 PetscCall(PetscLogEventEnd(MATCEED_SETUP_FULL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedSetPreallocationCOO()
739 PetscErrorCode MatCeedAssembleCOO(Mat mat_ceed, Mat mat_coo) { in MatCeedAssembleCOO() argument
750 if (ctx->mats_assembled_full[i] == mat_coo) index = i; in MatCeedAssembleCOO()
752 if (index == -1) PetscCall(MatCeedSetPreallocationCOO(mat_ceed, mat_coo)); in MatCeedAssembleCOO()
756 PetscCall(PetscLogEventBegin(MATCEED_ASSEMBLE_FULL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssembleCOO()
763 PetscCall(MatGetMemTypeFromVecType(mat_coo, &mem_type_petsc)); in MatCeedAssembleCOO()
765 PetscCall(PetscLogEventBegin(MATCEED_ASSEMBLE_FULL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssembleCOO()
767 PetscCall(PetscLogEventEnd(MATCEED_ASSEMBLE_FULL_CEEDOP, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssembleCOO()
769 PetscCall(MatSetValuesCOO(mat_coo, values, INSERT_VALUES)); in MatCeedAssembleCOO()
771 if (is_spd_known) PetscCall(MatSetOption(mat_coo, MAT_SPD, is_spd)); in MatCeedAssembleCOO()
774 PetscCall(MatAssemblyBegin(mat_coo, MAT_FINAL_ASSEMBLY)); in MatCeedAssembleCOO()
775 PetscCall(MatAssemblyEnd(mat_coo, MAT_FINAL_ASSEMBLY)); in MatCeedAssembleCOO()
776 PetscCall(PetscLogEventEnd(MATCEED_ASSEMBLE_FULL, mat_ceed, mat_coo, NULL, NULL)); in MatCeedAssembleCOO()