Lines Matching full:if
33 if (registered) PetscFunctionReturn(PETSC_SUCCESS); in MatCeedRegisterLogEvents()
70 // Check if COO pattern set in MatCeedAssemblePointBlockDiagonalCOO()
75 if (ctx->mats_assembled_pbd[i] == mat_coo) index = i; in MatCeedAssemblePointBlockDiagonalCOO()
77 if (index == -1) { in MatCeedAssemblePointBlockDiagonalCOO()
83 // -- Assemble sparsity pattern if mat hasn't been assembled before in MatCeedAssemblePointBlockDiagonalCOO()
85 if (stage_amg_setup == -1) { in MatCeedAssemblePointBlockDiagonalCOO()
98 …if (!ctx->coo_values_pbd) PetscCallCeed(ctx->ceed, CeedVectorCreate(ctx->ceed, num_entries, &ctx->… in MatCeedAssemblePointBlockDiagonalCOO()
116 if (strstr(mat_type, "cusparse")) mem_type = CEED_MEM_DEVICE; in MatCeedAssemblePointBlockDiagonalCOO()
117 else if (strstr(mat_type, "kokkos")) mem_type = CEED_MEM_DEVICE; in MatCeedAssemblePointBlockDiagonalCOO()
126 if (is_spd_known) PetscCall(MatSetOption(mat_coo, MAT_SPD, is_spd)); in MatCeedAssemblePointBlockDiagonalCOO()
150 if (use_ceed_pbd) { in MatCeedAssembleInnerBlockDiagonalMat()
151 // Check if COO pattern set in MatCeedAssembleInnerBlockDiagonalMat()
152 …if (!ctx->mat_assembled_pbd_internal) PetscCall(MatCeedCreateMatCOO(mat_ceed, &ctx->mat_assembled_… in MatCeedAssembleInnerBlockDiagonalMat()
156 if (mat_inner) *mat_inner = ctx->mat_assembled_pbd_internal; in MatCeedAssembleInnerBlockDiagonalMat()
158 // Check if COO pattern set in MatCeedAssembleInnerBlockDiagonalMat()
159 …if (!ctx->mat_assembled_full_internal) PetscCall(MatCeedCreateMatCOO(mat_ceed, &ctx->mat_assembled… in MatCeedAssembleInnerBlockDiagonalMat()
163 if (mat_inner) *mat_inner = ctx->mat_assembled_full_internal; in MatCeedAssembleInnerBlockDiagonalMat()
184 // Assemble inner mat if needed in MatGetVariableBlockDiagonal_Ceed()
206 // Assemble inner mat if needed in MatGetBlockDiagonal_Ceed()
231 // Check if COO pattern set in MatGetDiagonalBlock_Ceed()
232 …if (!ctx->mat_assembled_full_internal) PetscCall(MatCeedCreateMatCOO(mat_ceed, &ctx->mat_assembled… in MatGetDiagonalBlock_Ceed()
261 …if (!viewer) PetscCall(PetscViewerASCIIGetStdout(PetscObjectComm((PetscObject)mat_ceed), &viewer)); in MatView_Ceed()
265 if (size == 1 && format == PETSC_VIEWER_LOAD_BALANCE) PetscFunctionReturn(PETSC_SUCCESS); in MatView_Ceed()
268 if (rank != 0) PetscFunctionReturn(PETSC_SUCCESS); in MatView_Ceed()
285 if (is_detailed) PetscCallCeed(ctx->ceed, CeedOperatorView(ctx->op_mult, file)); in MatView_Ceed()
288 if (ctx->op_mult_transpose) { in MatView_Ceed()
291 if (is_detailed) PetscCallCeed(ctx->ceed, CeedOperatorView(ctx->op_mult_transpose, file)); in MatView_Ceed()
335 if (dm_y) { in MatCreateCeed()
354 if (op_mult_transpose) { in MatCreateCeed()
368 if (dm_x == dm_y) { in MatCreateCeed()
395 if (block_size > 1) PetscCall(MatSetBlockSize(*mat, block_size)); in MatCreateCeed()
396 … if (num_blocks) PetscCall(MatSetVariableBlockSizes(*mat, num_blocks, (PetscInt *)vblock_sizes)); in MatCreateCeed()
405 if (is_composite) { in MatCreateCeed()
419 if (num_bases > 1) { in MatCreateCeed()
423 if (num_comp != block_size) ctx->is_ceed_pbd_valid = PETSC_FALSE; in MatCreateCeed()
424 if (num_comp < max_vblock_size) ctx->is_ceed_vpbd_valid = PETSC_FALSE; in MatCreateCeed()
435 if (num_bases > 1) { in MatCreateCeed()
439 if (num_comp != block_size) ctx->is_ceed_pbd_valid = PETSC_FALSE; in MatCreateCeed()
440 if (num_comp < max_vblock_size) ctx->is_ceed_vpbd_valid = PETSC_FALSE; in MatCreateCeed()
463 if (strstr(vec_type, VECCUDA)) coo_mat_type = MATAIJCUSPARSE; in MatCreateCeed()
464 else if (strstr(vec_type, VECKOKKOS)) coo_mat_type = MATAIJKOKKOS; in MatCreateCeed()
472 …if (op_mult_transpose) PetscCall(MatShellSetOperation(*mat, MATOP_MULT_TRANSPOSE, (void (*)(void))… in MatCreateCeed()
538 …if (ctx->op_mult_transpose) PetscCall(MatShellSetOperation(mat_other, MATOP_MULT_TRANSPOSE, (void … in MatCeedCopy()
547 if (block_size > 1) PetscCall(MatSetBlockSize(mat_other, block_size)); in MatCeedCopy()
554 …if (num_blocks) PetscCall(MatSetVariableBlockSizes(mat_other, num_blocks, (PetscInt *)block_sizes)… in MatCeedCopy()
578 if (ctx->op_mult_transpose) { in MatCeedSetAssemblyDataUpdateNeeded()
581 if (update_needed) { in MatCeedSetAssemblyDataUpdateNeeded()
620 if (is_coo_mat_type_cl) { in MatCeedCreateMatCOO()
664 // -- Assemble sparsity pattern if mat hasn't been assembled before in MatCeedSetPreallocationCOO()
666 if (stage_amg_setup == -1) { in MatCeedSetPreallocationCOO()
679 …if (!ctx->coo_values_full) PetscCallCeed(ctx->ceed, CeedVectorCreate(ctx->ceed, num_entries, &ctx-… in MatCeedSetPreallocationCOO()
706 // Set COO pattern if needed in MatCeedAssembleCOO()
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()
726 if (strstr(mat_type, "cusparse")) mem_type = CEED_MEM_DEVICE; in MatCeedAssembleCOO()
727 else if (strstr(mat_type, "kokkos")) mem_type = CEED_MEM_DEVICE; in MatCeedAssembleCOO()
736 if (is_spd_known) PetscCall(MatSetOption(mat_coo, MAT_SPD, is_spd)); in MatCeedAssembleCOO()
765 if (label) { in MatCeedSetContextDouble()
769 if (set_value != value) { in MatCeedSetContextDouble()
774 if (ctx->op_mult_transpose) { in MatCeedSetContextDouble()
777 if (label) { in MatCeedSetContextDouble()
781 if (set_value != value) { in MatCeedSetContextDouble()
788 if (was_updated) { in MatCeedSetContextDouble()
816 if (!label && ctx->op_mult_transpose) { in MatCeedGetContextDouble()
820 if (label) { in MatCeedGetContextDouble()
897 @param[out] time Current time, or -1.0 if the boundary evaluator has no time field
951 if (shift_a) { in MatCeedSetShifts()
974 if (ctx) { in MatCeedSetContext()
999 if (user_ctx) PetscCall(PetscContainerGetPointer(user_ctx, (void **)ctx)); in MatCeedGetContext()
1038 // Check if same in MatCeedSetCOOMatType()
1045 if (len_old == len_new) PetscCall(PetscStrncmp(ctx->coo_mat_type, type, len_old, &is_same)); in MatCeedSetCOOMatType()
1046 if (is_same) PetscFunctionReturn(PETSC_SUCCESS); in MatCeedSetCOOMatType()
1050 if (ctx->mat_assembled_full_internal) { in MatCeedSetCOOMatType()
1052 if (ctx->mats_assembled_full[i] == ctx->mat_assembled_full_internal) { in MatCeedSetCOOMatType()
1062 if (ctx->mat_assembled_pbd_internal) { in MatCeedSetCOOMatType()
1064 if (ctx->mats_assembled_pbd[i] == ctx->mat_assembled_pbd_internal) { in MatCeedSetCOOMatType()
1115 if (X_loc) { in MatCeedSetLocalVectors()
1124 if (Y_loc_transpose) { in MatCeedSetLocalVectors()
1152 if (X_loc) { in MatCeedGetLocalVectors()
1156 if (Y_loc_transpose) { in MatCeedGetLocalVectors()
1176 if (X_loc) PetscCall(VecDestroy(X_loc)); in MatCeedRestoreLocalVectors()
1177 if (Y_loc_transpose) PetscCall(VecDestroy(Y_loc_transpose)); in MatCeedRestoreLocalVectors()
1197 if (op_mult) { in MatCeedGetCeedOperators()
1201 if (op_mult_transpose) { in MatCeedGetCeedOperators()
1224 if (op_mult) PetscCallCeed(ctx->ceed, CeedOperatorDestroy(op_mult)); in MatCeedRestoreCeedOperators()
1225 if (op_mult_transpose) PetscCallCeed(ctx->ceed, CeedOperatorDestroy(op_mult_transpose)); in MatCeedRestoreCeedOperators()
1245 if (log_event_mult) ctx->log_event_mult = log_event_mult; in MatCeedSetLogEvents()
1246 if (log_event_mult_transpose) ctx->log_event_mult_transpose = log_event_mult_transpose; in MatCeedSetLogEvents()
1266 if (log_event_mult) *log_event_mult = ctx->log_event_mult; in MatCeedGetLogEvents()
1267 if (log_event_mult_transpose) *log_event_mult_transpose = ctx->log_event_mult_transpose; in MatCeedGetLogEvents()
1287 if (log_event_mult) ctx->log_event_ceed_mult = log_event_mult; in MatCeedSetCeedOperatorLogEvents()
1288 if (log_event_mult_transpose) ctx->log_event_ceed_mult_transpose = log_event_mult_transpose; in MatCeedSetCeedOperatorLogEvents()
1308 if (log_event_mult) *log_event_mult = ctx->log_event_ceed_mult; in MatCeedGetCeedOperatorLogEvents()
1309 if (log_event_mult_transpose) *log_event_mult_transpose = ctx->log_event_ceed_mult_transpose; in MatCeedGetCeedOperatorLogEvents()
1356 if (X_loc) PetscCall(VecReferenceCopy(X_loc, &(*ctx)->X_loc)); in MatCeedContextCreate()
1357 if (Y_loc_transpose) PetscCall(VecReferenceCopy(Y_loc_transpose, &(*ctx)->Y_loc_transpose)); in MatCeedContextCreate()
1375 …if (op_mult_transpose) PetscCallCeed((*ctx)->ceed, CeedOperatorReferenceCopy(op_mult_transpose, &(… in MatCeedContextCreate()
1385 if (op_mult_transpose) { in MatCeedContextCreate()
1392 if (x_loc_len > 0 || y_loc_len > 0) { in MatCeedContextCreate()
1402 if (X_loc) { in MatCeedContextCreate()
1421 if (Y_loc_transpose) { in MatCeedContextCreate()
1447 Note: If `ctx_copy` is non-null, it is assumed to be a valid pointer to a `MatCeedContext`.
1475 if (!ctx || --(*ctx)->ref_count > 0) PetscFunctionReturn(PETSC_SUCCESS); in MatCeedContextDestroy()
1565 if (!ctx->X_loc) PetscCall(DMGetLocalVector(ctx->dm_x, &X_loc)); in MatMult_Ceed()
1592 if (!ctx->X_loc) PetscCall(DMRestoreLocalVector(ctx->dm_x, &X_loc)); in MatMult_Ceed()
1597 if (PetscMemTypeDevice(ctx->mem_type)) PetscCall(PetscLogGpuFlops(ctx->flops_mult)); in MatMult_Ceed()
1626 if (!ctx->Y_loc_transpose) PetscCall(DMGetLocalVector(ctx->dm_y, &Y_loc)); in MatMultTranspose_Ceed()
1653 if (!ctx->Y_loc_transpose) PetscCall(DMRestoreLocalVector(ctx->dm_y, &Y_loc)); in MatMultTranspose_Ceed()
1658 if (PetscMemTypeDevice(ctx->mem_type)) PetscCall(PetscLogGpuFlops(ctx->flops_mult_transpose)); in MatMultTranspose_Ceed()