使用HYPRE库并行装配IJ稀疏矩阵指南: 矩阵预分配和重复利用-CSDN博客

本文链接：https://blog.csdn.net/weixin_42849849/article/details/147478042

使用HYPRE库并行装配IJ稀疏矩阵指南

HYPRE是一个流行的并行求解器库，特别适合大规模稀疏线性系统的求解。下面介绍如何并行装配IJ格式的稀疏矩阵，包括预先分配矩阵空间和循环使用。

1. 初始化矩阵

首先需要创建并初始化一个IJ矩阵：

#include "HYPRE.h"
#include "HYPRE_parcsr_ls.h"

HYPRE_IJMatrix A;
int ilower, iupper; // 本进程负责的行范围
int jlower, jupper; // 列范围（通常全局）
int n_procs, myid;

// 初始化MPI和HYPRE
MPI_Comm_size(MPI_COMM_WORLD, &n_procs);
MPI_Comm_rank(MPI_COMM_WORLD, &myid);

// 确定本进程负责的行范围（假设均匀划分）
int total_rows = ...; // 全局总行数
ilower = (total_rows / n_procs) * myid;
iupper = (total_rows / n_procs) * (myid + 1) - 1;
if (myid == n_procs - 1) iupper = total_rows - 1;

// 创建IJ矩阵
HYPRE_IJMatrixCreate(MPI_COMM_WORLD, ilower, iupper, jlower, jupper, &A);
HYPRE_IJMatrixSetObjectType(A, HYPRE_PARCSR);

2. 预先分配矩阵空间（已知稀疏模式）

如果矩阵的稀疏模式已知且相同，可以预先分配空间以提高效率：

// 假设每行的非零元数量已知
int *nnz_per_row = (int*)malloc((iupper - ilower + 1) * sizeof(int));
for (int i = 0; i <= iupper - ilower; i++) {
    nnz_per_row[i] = ...; // 设置每行的非零元数量
}

// 预先分配矩阵空间
HYPRE_IJMatrixSetRowSizes(A, nnz_per_row);
HYPRE_IJMatrixInitialize(A);

free(nnz_per_row);

3. 装配矩阵值

在计算循环中装配矩阵值：

for (int time_step = 0; time_step < max_steps; time_step++) {
    // 每次迭代前可以清除旧值（如果需要）
    // HYPRE_IJMatrixSetConstantValues(A, 0.0);
    
    for (int i = ilower; i <= iupper; i++) {
        int local_row = i - ilower;
        int ncols = ...; // 本行的非零元数
        int *cols = ...; // 列索引数组
        double *values = ...; // 值数组
        
        // 设置矩阵值
        HYPRE_IJMatrixSetValues(A, 1, &ncols, &i, cols, values);
    }
    
    // 完成装配
    HYPRE_IJMatrixAssemble(A);
    
    // 在这里可以使用矩阵进行求解等操作...
    
    // 如果需要获取ParCSR矩阵对象用于求解器
    HYPRE_ParCSRMatrix parcsr_A;
    HYPRE_IJMatrixGetObject(A, (void**)&parcsr_A);
}

4. 优化技巧

批量设置值：如果可能，批量设置多行值比逐行设置更高效：

int rows[10]; int nrows = 10;
int ncols_per_row[10]; 
int *cols[10]; 
double *values[10];
HYPRE_IJMatrixSetValues(A, nrows, ncols_per_row, rows, cols, values);

重用矩阵结构：如果只有值变化而稀疏模式不变，可以：

// 第一次装配
HYPRE_IJMatrixInitialize(A);
HYPRE_IJMatrixAssemble(A);

// 后续只更新值
HYPRE_IJMatrixUpdateValues(A, ...);

内存管理：预先分配所有内存，避免在时间循环中频繁分配释放。

5. 清理资源

计算完成后释放资源：

HYPRE_IJMatrixDestroy(A);

6. 完整示例框架

#include "HYPRE.h"
#include "HYPRE_parcsr_ls.h"
#include <mpi.h>

void assemble_matrix(HYPRE_IJMatrix A, int ilower, int iupper) {
    // 实现具体的矩阵装配逻辑
    for (int i = ilower; i <= iupper; i++) {
        int local_row = i - ilower;
        int ncols = ...;
        int *cols = ...;
        double *values = ...;
        HYPRE_IJMatrixSetValues(A, 1, &ncols, &i, cols, values);
    }
}

int main(int argc, char *argv[]) {
    MPI_Init(&argc, &argv);
    
    int n_procs, myid;
    MPI_Comm_size(MPI_COMM_WORLD, &n_procs);
    MPI_Comm_rank(MPI_COMM_WORLD, &myid);
    
    // 矩阵参数
    int total_rows = 1000; // 示例值
    int ilower = (total_rows / n_procs) * myid;
    int iupper = (total_rows / n_procs) * (myid + 1) - 1;
    if (myid == n_procs - 1) iupper = total_rows - 1;
    
    // 创建矩阵
    HYPRE_IJMatrix A;
    HYPRE_IJMatrixCreate(MPI_COMM_WORLD, ilower, iupper, 0, total_rows-1, &A);
    HYPRE_IJMatrixSetObjectType(A, HYPRE_PARCSR);
    
    // 预先分配
    int *nnz_per_row = (int*)malloc((iupper - ilower + 1) * sizeof(int));
    // 填充nnz_per_row...
    HYPRE_IJMatrixSetRowSizes(A, nnz_per_row);
    HYPRE_IJMatrixInitialize(A);
    free(nnz_per_row);
    
    // 时间循环
    for (int step = 0; step < 100; step++) {
        assemble_matrix(A, ilower, iupper);
        HYPRE_IJMatrixAssemble(A);
        
        // 使用矩阵求解...
    }
    
    HYPRE_IJMatrixDestroy(A);
    MPI_Finalize();
    return 0;
}