37                   matrix.nrows * sizeof(PetscInt) +  /* row_nnz[*] */  in spbas_memory_requirement()
38                   matrix.nrows * sizeof(PetscInt *); /* icols[*] */  in spbas_memory_requirement()
41   if (matrix.col_idx_type == SPBAS_OFFSET_ARRAY) memreq += matrix.nrows * sizeof(PetscInt);  in spbas_memory_requirement()
48     memreq += matrix.nrows * sizeof(PetscScalar *); /* values[*] */  in spbas_memory_requirement()
62   PetscInt nrows        = result->nrows;  in spbas_allocate_pattern()  local
67   PetscCall(PetscMalloc1(nrows, &result->row_nnz));  in spbas_allocate_pattern()
68   PetscCall(PetscMalloc1(nrows, &result->icols));  in spbas_allocate_pattern()
72     PetscCall(PetscMalloc1(nrows, &result->icol0));  in spbas_allocate_pattern()
79     PetscCall(PetscMalloc1(nrows, &result->values));  in spbas_allocate_pattern()
98   PetscInt  nrows = result->nrows;  in spbas_allocate_data()  local
111     for (i = 1; i < nrows; i++) result->icols[i] = result->icols[i - 1] + result->row_nnz[i - 1];  in spbas_allocate_data()
120 …    for (i = 1; i < nrows; i++) result->values[i] = result->values[i - 1] + result->row_nnz[i - 1];  in spbas_allocate_data()
123     for (i = 0; i < nrows; i++) {  in spbas_allocate_data()
128       for (i = 0; i < nrows; i++) {  in spbas_allocate_data()
179 static PetscErrorCode spbas_mergesort_icols(PetscInt nrows, PetscInt *irow_in, PetscInt *icol_in, P…  in spbas_mergesort_icols()  argument
190   PetscCall(PetscMalloc1(nrows, &ialloc));  in spbas_mergesort_icols()
196   for (istep = 1; istep < nrows; istep *= 2) {  in spbas_mergesort_icols()
206     for (istart = 0; istart < nrows; istart += 2 * istep) {  in spbas_mergesort_icols()
210       if (i1end > nrows) i1end = nrows;  in spbas_mergesort_icols()
213       if (i2end > nrows) i2end = nrows;  in spbas_mergesort_icols()
238     for (i = 0; i < nrows; i++) isort[i] = ihlp2[i];  in spbas_mergesort_icols()
250 PetscErrorCode spbas_compress_pattern(PetscInt *irow_in, PetscInt *icol_in, PetscInt nrows, PetscIn…  in spbas_compress_pattern()  argument
252   PetscInt        nnz      = irow_in[nrows];  in spbas_compress_pattern()
253   size_t          mem_orig = (nrows + nnz) * sizeof(PetscInt);  in spbas_compress_pattern()
266   B->nrows        = nrows;  in spbas_compress_pattern()
276     for (i = 0; i < nrows; i++) B->icol0[i] = icol_in[irow_in[i]];  in spbas_compress_pattern()
280   PetscCall(PetscMalloc1(nrows, &isort));  in spbas_compress_pattern()
281   PetscCall(PetscMalloc1(nrows, &ipoint));  in spbas_compress_pattern()
282   PetscCall(PetscCalloc1(nrows, &used));  in spbas_compress_pattern()
284   for (i = 0; i < nrows; i++) {  in spbas_compress_pattern()
291   PetscCall(spbas_mergesort_icols(nrows, irow_in, icol_in, col_idx_type, isort));  in spbas_compress_pattern()
295   for (i = 1; i < nrows; i++) {  in spbas_compress_pattern()
300   for (i = 0; i < nrows; i++) used[ipoint[i]] = PETSC_TRUE;  in spbas_compress_pattern()
304   for (i = 0; i < nrows; i++) {  in spbas_compress_pattern()
311   for (i = 0; i < B->nrows; i++) {  in spbas_compress_pattern()
328   for (i = 0; i < nrows; i++) B->icols[i] = B->icols[ipoint[i]];  in spbas_compress_pattern()
330 …(PetscInfo(NULL, "         (%g nonzeros per row)\n", (double)((PetscReal)nnz / (PetscReal)nrows)));  in spbas_compress_pattern()
359     for (i = 0; i < matrix.nrows; i++) PetscCall(PetscFree(matrix.icols[i]));  in spbas_delete()
362       for (i = 0; i < matrix.nrows; i++) PetscCall(PetscFree(matrix.values[i]));  in spbas_delete()
379   PetscInt     nrows = matrix_A.nrows;  in spbas_matrix_to_crs()  local
391   PetscCall(PetscMalloc1(nrows + 1, &irow));  in spbas_matrix_to_crs()
403   for (i = 0; i < nrows; i++) {  in spbas_matrix_to_crs()
437   PetscInt     ncols        = in_matrix.nrows;  in spbas_transpose()
438   PetscInt     nrows        = in_matrix.ncols;  in spbas_transpose()  local
447   result->nrows        = nrows;  in spbas_transpose()
457   for (i = 0; i < nrows; i++) result->row_nnz[i] = 0;  in spbas_transpose()
476   for (i = 0; i < nrows; i++) result->row_nnz[i] = 0;  in spbas_transpose()
633   PetscInt      nrows = matrix_A->nrows;  in spbas_apply_reordering_rows()  local
642   if (do_values) PetscCall(PetscMalloc1(nrows, &vals));  in spbas_apply_reordering_rows()
643   PetscCall(PetscMalloc1(nrows, &row_nnz));  in spbas_apply_reordering_rows()
644   PetscCall(PetscMalloc1(nrows, &icols));  in spbas_apply_reordering_rows()
646   for (i = 0; i < nrows; i++) {  in spbas_apply_reordering_rows()
671   PetscInt     nrows = matrix_A->nrows;  in spbas_apply_reordering_cols()  local
680   for (i = 0; i < nrows; i++) {  in spbas_apply_reordering_cols()
703 PetscErrorCode spbas_pattern_only(PetscInt nrows, PetscInt ncols, PetscInt *ai, PetscInt *aj, spbas…  in spbas_pattern_only()  argument
710   retval.nrows = nrows;  in spbas_pattern_only()
712   retval.nnz   = ai[nrows];  in spbas_pattern_only()
719   for (i = 0; i < nrows; i++) retval.row_nnz[i] = ai[i + 1] - ai[i];  in spbas_pattern_only()
722   for (i = 0; i < retval.nrows; i++) {  in spbas_pattern_only()
778   PetscInt     nrows = in_matrix.nrows;  in spbas_power()  local
788   PetscCheck(ncols == nrows, PETSC_COMM_SELF, PETSC_ERR_ARG_INCOMP, "Dimension error");  in spbas_power()
793   retval.nrows        = ncols;  in spbas_power()
794   retval.ncols        = nrows;  in spbas_power()
803   PetscCall(PetscCalloc1(PetscMax(ncols, nrows), &iwork));  in spbas_power()
809   for (i = 0; i < nrows; i++) {  in spbas_power()
852   for (i = 0; i < inout_matrix->nrows; i++) {  in spbas_keep_upper()