c - PetscMalloc 与 PetscMallocX

Question

使用 PetscMalloc2 (PetscMallocX) 代替 PetscMalloc 两次(X 次)的经验法则是什么？ block 是否应该具有相似的大小，或者将它们分配在一起/同时分配总是更有效？手册中的定义是“分配 2 (X) 个与 PETSC_MEMALIGN 对齐的内存块”，但这对我作为 PETSc 和精细 HPC 问题的新手来说意义不大。我一直认为编译器会处理此类问题。

Answer 1

参见http://www.mcs.anl.gov/petsc/petsc-current/include/petscsys.h.html#PetscMalloc2 ，线566:

#if defined(PETSC_USE_DEBUG)
#define PetscMalloc2(m1,t1,r1,m2,t2,r2) (PetscMalloc((m1)*sizeof(t1),r1) || PetscMalloc((m2)*sizeof(t2),r2))
#else
#define PetscMalloc2(m1,t1,r1,m2,t2,r2) ((*(r2) = 0,PetscMalloc((m1)*sizeof(t1) (m2)*sizeof(t2)+(PETSC_MEMALIGN-1),r1)) || (*(r2) = (t2*)PetscAddrAlign(*(r1)+m1),0))
#endif

如果处于 Debug模式，PetscMalloc2 相当于两个 PetscMalloc。

否则，PetscMalloc2 确保两个缓冲区在内存中是一个接一个的，由于内存对齐而有一点空间。 http://en.wikipedia.org/wiki/Data_structure_alignment通过 PetscMalloc 进行的分配仅被调用一次，如果您在代码中多次调用此函数，效果会更好。如果你偶尔调用一次，它不会有太大变化!

再见，