Lines Matching refs:nvars
351 PetscInt nvars = ex->nvars; in MatSetValuesLocal_HYPRESStruct_3d() local
355 PetscCall(PetscMalloc1(7 * nvars, &entries)); in MatSetValuesLocal_HYPRESStruct_3d()
361 grid_rank = irow[i] / nvars; in MatSetValuesLocal_HYPRESStruct_3d()
362 var_type = (irow[i] % nvars); in MatSetValuesLocal_HYPRESStruct_3d()
365 to_grid_rank = icol[j] / nvars; in MatSetValuesLocal_HYPRESStruct_3d()
366 to_var_type = (icol[j] % nvars); in MatSetValuesLocal_HYPRESStruct_3d()
453 PetscInt nvars = ex->nvars; in MatZeroRowsLocal_HYPRESStruct_3d() local
458 PetscCall(PetscMalloc1(7 * nvars, &entries)); in MatZeroRowsLocal_HYPRESStruct_3d()
460 PetscCall(PetscMalloc1(nvars, &values)); in MatZeroRowsLocal_HYPRESStruct_3d()
461 PetscCall(PetscMalloc1(7 * nvars * nvars, &values[0])); in MatZeroRowsLocal_HYPRESStruct_3d()
462 for (i = 1; i < nvars; i++) values[i] = values[i - 1] + nvars * 7; in MatZeroRowsLocal_HYPRESStruct_3d()
464 for (i = 0; i < nvars; i++) { in MatZeroRowsLocal_HYPRESStruct_3d()
465 PetscCall(PetscArrayzero(values[i], nvars * 7 * sizeof(HYPRE_Complex))); in MatZeroRowsLocal_HYPRESStruct_3d()
469 for (i = 0; i < nvars * 7; i++) entries[i] = (HYPRE_Int)i; in MatZeroRowsLocal_HYPRESStruct_3d()
473 grid_rank = irow[i] / nvars; in MatZeroRowsLocal_HYPRESStruct_3d()
474 var_type = (irow[i] % nvars); in MatZeroRowsLocal_HYPRESStruct_3d()
480 …rixSetValues(ex->ss_mat, part, index, (HYPRE_Int)var_type, 7 * (HYPRE_Int)nvars, entries, values[v… in MatZeroRowsLocal_HYPRESStruct_3d()
491 …rixSetValues(ex->ss_mat, part, index, (HYPRE_Int)var_type, 7 * (HYPRE_Int)nvars, entries, values[v… in MatZeroRowsLocal_HYPRESStruct_3d()
504 PetscInt nvars = ex->nvars; in MatZeroEntries_HYPRESStruct_3d() local
519 PetscCall(PetscMalloc2(nvars * 7, &entries, nvars * 7 * size, &values)); in MatZeroEntries_HYPRESStruct_3d()
520 for (i = 0; i < nvars * 7; i++) entries[i] = i; in MatZeroEntries_HYPRESStruct_3d()
521 PetscCall(PetscArrayzero(values, nvars * 7 * size)); in MatZeroEntries_HYPRESStruct_3d()
523 …or (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructMatrixSetBoxValues(ex->ss_mat, part, ilower… in MatZeroEntries_HYPRESStruct_3d()
563 ex->nvars = (int)dof; in MatSetUp_HYPRESStruct()
571 PetscCall(PetscMalloc1(ex->nvars, &vartypes)); in MatSetUp_HYPRESStruct()
572 for (i = 0; i < ex->nvars; i++) vartypes[i] = HYPRE_SSTRUCT_VARIABLE_CELL; in MatSetUp_HYPRESStruct()
573 … PetscCallHYPRE(HYPRE_SStructGridSetVariables(ex->ss_grid, part, (HYPRE_Int)ex->nvars, vartypes)); in MatSetUp_HYPRESStruct()
590 ssize = 3 * (ex->nvars); in MatSetUp_HYPRESStruct()
593 for (i = 0; i < (ex->nvars); i++) { in MatSetUp_HYPRESStruct()
609 ssize = 5 * (ex->nvars); in MatSetUp_HYPRESStruct()
612 for (i = 0; i < (ex->nvars); i++) { in MatSetUp_HYPRESStruct()
630 ssize = 7 * (ex->nvars); in MatSetUp_HYPRESStruct()
633 for (i = 0; i < (ex->nvars); i++) { in MatSetUp_HYPRESStruct()
646 …for (i = 0; i < (ex->nvars); i++) PetscCallHYPRE(HYPRE_SStructGraphSetStencil(ex->ss_graph, part, … in MatSetUp_HYPRESStruct()
706 PetscInt nvars = mx->nvars; in MatMult_HYPRESStruct() local
732 …for (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructVectorSetBoxValues(mx->ss_b, part, hlower,… in MatMult_HYPRESStruct()
739 …for (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructVectorGetBoxValues(mx->ss_x, part, hlower,… in MatMult_HYPRESStruct()
745 PetscCall(PetscMalloc1(nvars * size, &z)); in MatMult_HYPRESStruct()
751 k = i * nvars; in MatMult_HYPRESStruct()
752 for (j = 0; j < nvars; j++) z[j * size + i] = xx[k + j]; in MatMult_HYPRESStruct()
754 …for (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructVectorSetBoxValues(mx->ss_b, part, hlower,… in MatMult_HYPRESStruct()
761 …for (i = 0; i < nvars; i++) PetscCallHYPRE(HYPRE_SStructVectorGetBoxValues(mx->ss_x, part, hlower,… in MatMult_HYPRESStruct()
764 k = i * nvars; in MatMult_HYPRESStruct()
765 for (j = 0; j < nvars; j++) yy[k + j] = z[j * size + i]; in MatMult_HYPRESStruct()