petsc 矩阵测试

最新推荐文章于 2024-08-25 10:36:12 发布

lizhengjiang

最新推荐文章于 2024-08-25 10:36:12 发布

阅读量2.8k

点赞数

本文链接：https://blog.csdn.net/lizhengjiang/article/details/16854583

版权

这篇博客介绍了如何在PETSc中进行矩阵的行划分和赋值操作，通过示例展示了矩阵的分布情况，强调了即使元素不在当前节点的管辖范围内，PETSc也会自动处理数据通信。内容包括矩阵元素的赋值模式，如（I, J+1）、（I-1, J）、（I+1, J）、（I, J-1），形成了一种条状结构，并解释了如何使用if语句进行相邻元素赋值。" 105721613,5114941,Rust FFI与libc crate详解,"['Rust编程', '系统编程', 'C语言接口', 'FFI', '底层开发']

摘要由CSDN通过智能技术生成

第一个PETSc矩阵测试:

static char help[] = " matrix parallel partition test\n\n";

 #include <petscksp.h>
 int main(int argc, char **args)
{
 Mat A;
 PetscInt dim, i, j, Ii, J, Istart, Iend, n=6;
 PetscScalar v ;

 PetscInitialize(&argc, &args, (char*)0,help);
 PetscOptionsGetReal(NULL, "-sigma1", &sigma1, NULL);
 PetscOptionsGetInt(NULL, "-n", &n, NULL);
 dim = n * n;

 MatCreate(PETSC_COMM_WORLD, &A);
 MatSetSizes(A,PETSC_DECIDE,PETSC_DECIDE,dim,dim);
 MatSetFromOptions(A);
 MatSetUp(A);

 MatGetOwnershipRange(A,&Istart,&Iend);

 for (Ii = Istart; Ii < Iend; Ii++) {
        v = 1.0; i = Ii/n; j = Ii - i * n;
        if(i > 0) {
                J = Ii - n; MatSetValues(A,1,&Ii,1,&J,&v,ADD_VALUES);
        }
        if(i < n - 1) {
                J = Ii + n; MatSetValues(A,1,&Ii,1,&J,&v,ADD_VALUES);