使用 DMDAVecGetArray 时 PETSc 内存损坏
PETSc memory corruption on using DMDAVecGetArray
我正在尝试使用具有 2 个自由度的 PETSc 的 DMDA 向量并使用结构 like in the manual 访问它们。
但是,当我尝试使用 DMDAVecGetArray
即使只有一个自由度时(如下例所示),我也会收到内存 double free or corruption
错误。当我用 VecGetArray
替换 DMDAVecGetArray
时,一切正常。
导致此错误的原因是什么?
用
编译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, 1, NULL, &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, f, &farr)); // with this
E(DMLocalToGlobal(domain, f, INSERT_VALUES, 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
好的,找到问题了。这里的主要区别在于,当使用 VecGetArray
时,基础数组使用本地索引进行索引,而当使用 DMDAVecGetArray
时,使用全局索引。
因此,为了修复代码,我不仅要获取数组的大小,还要获取局部向量部分的第一个索引。
使用函数
PetscInt start, size;
E(DMDAGetCorners(domain, &start, NULL, NULL, &size, NULL, NULL));
然后从 start
循环到 start + size
修复了内存损坏错误。
我正在尝试使用具有 2 个自由度的 PETSc 的 DMDA 向量并使用结构 like in the manual 访问它们。
但是,当我尝试使用 DMDAVecGetArray
即使只有一个自由度时(如下例所示),我也会收到内存 double free or corruption
错误。当我用 VecGetArray
替换 DMDAVecGetArray
时,一切正常。
导致此错误的原因是什么?
用
编译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, 1, NULL, &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, f, &farr)); // with this
E(DMLocalToGlobal(domain, f, INSERT_VALUES, 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
好的,找到问题了。这里的主要区别在于,当使用 VecGetArray
时,基础数组使用本地索引进行索引,而当使用 DMDAVecGetArray
时,使用全局索引。
因此,为了修复代码,我不仅要获取数组的大小,还要获取局部向量部分的第一个索引。
使用函数
PetscInt start, size;
E(DMDAGetCorners(domain, &start, NULL, NULL, &size, NULL, NULL));
然后从 start
循环到 start + size
修复了内存损坏错误。