使用 DMDAVecGetArray

问题描述

我正在尝试使用具有 2 个自由度的 PETSc 的 DMDA 向量,并使用结构 like in the manual 访问它们。 但是,当我尝试使用 DMDAVecGetArray 时,即使有一个自由度(如下例所示),我也会收到内存 double free or corruption 错误。当我用 DMDAVecGetArray 替换 VecGetArray 时,一切正常。

导致此错误的原因是什么?


编译MWE

> gcc -Wall -Wextra -O0 -g $(pkg-config --cflags petsc mpi) -o mwe mwe.c $(pkg-config --libs petsc mpi)

并运行

> mpiexec --host localhost:$(nproc) -n $(nproc) mwe

MWE:

#include <petscdm.h>
#include <petscdmda.h>
#include <petscvec.h>

int
main(int argc,char **argv)
{
    PetscErrorCode ierr;
#define E(x)                                                                                                           \
    do {                                                                                                           \
        ierr = x;                                                                                              \
        CHKERRQ(ierr);                                                                                         \
    } while (0)

    PetscInt    N    = 10;
    Vec     fg,f;

    ierr = PetscInitialize(&argc,&argv,NULL,NULL);
    if (ierr) {
        return ierr;
    }

    DM domain;
    E(DMDACreate1d(PETSC_COMM_WORLD,DM_BOUNDARY_GHOSTED,N,1,&domain));
    E(DMSetUp(domain));

    E(DMCreateGlobalVector(domain,&fg));
    E(DMCreateLocalVector(domain,&f));
    E(VecZeroEntries(fg));

    E(VecAssemblyBegin(fg));
    E(VecAssemblyBegin(f));
    E(VecAssemblyEnd(fg));
    E(VecAssemblyEnd(f));

    PetscInt size;
    E(VecGetSize(f,&size));

    E(DMGlobalToLocal(domain,fg,INSERT_VALUES,f));

    PetscScalar *farr;
    E(VecGetArray(f,&farr)); // replace this
    // E(DMDAVecGetArrayWrite(domain,f,&farr)); // with this

    for (PetscInt i = 0; i < size; i++) {
        farr[i] += 1;
    }

    E(VecRestoreArray(f,&farr)); // replace this
    // E(DMDAVecRestoreArrayWrite(domain,&farr)); // with this
    E(DMLocalToGlobal(domain,fg));

    E(VecDestroy(&f));
    E(VecDestroy(&fg));
    E(DMDestroy(&domain));

    E(PetscFinalize());
    return ierr;
}

Debian 靶心; petsc-dev 3.14.4+dfsg1-1; libopenmpi-dev 4.1.0-7

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)