python/featgraph/util/adj_partitioning.py [105:128]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            if row_idx < num_row_partitions - 1 and col_idx < num_col_partitions - 1:
                adj_partition_scipy_csr = adj_scipy_csr[row_idx*num_rows_per_partition:(row_idx+1)*num_rows_per_partition, \
                    col_idx*num_cols_per_partition:(col_idx+1)*num_cols_per_partition]
            elif row_idx < num_row_partitions - 1 and col_idx == num_col_partitions - 1:
                adj_partition_scipy_csr = adj_scipy_csr[row_idx*num_rows_per_partition:(row_idx+1)*num_rows_per_partition, \
                    col_idx*num_cols_per_partition::]
            elif row_idx == num_row_partitions - 1 and col_idx < num_col_partitions - 1:
                adj_partition_scipy_csr = adj_scipy_csr[row_idx*num_rows_per_partition::, \
                    col_idx*num_cols_per_partition:(col_idx+1)*num_cols_per_partition]
            elif row_idx == num_row_partitions - 1 and col_idx == num_col_partitions - 1:
                adj_partition_scipy_csr = adj_scipy_csr[row_idx*num_rows_per_partition::, \
                    col_idx*num_cols_per_partition::]
            else:
                raise RuntimeError("no condition is satisfied")
            adj_partition_scipy_coo = adj_partition_scipy_csr.tocoo()
            nnz_in_this_partition = adj_partition_scipy_coo.nnz
            edge_id_list_after_partition[counter:(counter+nnz_in_this_partition)] = adj_partition_scipy_coo.data
            adj_row_indices_after_partition[counter:(counter+nnz_in_this_partition)] = \
                adj_partition_scipy_coo.row + row_idx * num_rows_per_partition
            adj_col_indices_after_partition[counter:(counter+nnz_in_this_partition)] = \
                adj_partition_scipy_coo.col + col_idx * num_cols_per_partition
            counter += nnz_in_this_partition

    return edge_id_list_after_partition, adj_row_indices_after_partition, adj_col_indices_after_partition
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



python/featgraph/util/adj_partitioning.py [157:180]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        if row_idx < num_row_partitions - 1 and col_idx < num_col_partitions - 1:
            adj_partition_scipy_csr = adj_scipy_csr[row_idx*num_rows_per_partition:(row_idx+1)*num_rows_per_partition, \
                col_idx*num_cols_per_partition:(col_idx+1)*num_cols_per_partition]
        elif row_idx < num_row_partitions - 1 and col_idx == num_col_partitions - 1:
            adj_partition_scipy_csr = adj_scipy_csr[row_idx*num_rows_per_partition:(row_idx+1)*num_rows_per_partition, \
                col_idx*num_cols_per_partition::]
        elif row_idx == num_row_partitions - 1 and col_idx < num_col_partitions - 1:
            adj_partition_scipy_csr = adj_scipy_csr[row_idx*num_rows_per_partition::, \
                col_idx*num_cols_per_partition:(col_idx+1)*num_cols_per_partition]
        elif row_idx == num_row_partitions - 1 and col_idx == num_col_partitions - 1:
            adj_partition_scipy_csr = adj_scipy_csr[row_idx*num_rows_per_partition::, \
                col_idx*num_cols_per_partition::]
        else:
            raise RuntimeError("no condition is satisfied")
        adj_partition_scipy_coo = adj_partition_scipy_csr.tocoo()
        nnz_in_this_partition = adj_partition_scipy_coo.nnz
        edge_id_list_after_partition[counter:(counter+nnz_in_this_partition)] = adj_partition_scipy_coo.data
        adj_row_indices_after_partition[counter:(counter+nnz_in_this_partition)] = \
            adj_partition_scipy_coo.row + row_idx * num_rows_per_partition
        adj_col_indices_after_partition[counter:(counter+nnz_in_this_partition)] = \
            adj_partition_scipy_coo.col + col_idx * num_cols_per_partition
        counter += nnz_in_this_partition

    return edge_id_list_after_partition, adj_row_indices_after_partition, adj_col_indices_after_partition
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -