如何在C ++中使用MPI而不是C malloc / calloc动态分配内存?

时间:2019-07-05 16:15:49

标签: c++ c mpi

我正在尝试使用MPI在C ++中编写并行代码,但是我只知道如何使用C命令(如malloc / calloc)分配内存。目的是使用身份矩阵并在MPI流程之间分解它。

在本地工作空间上创建身份矩阵,然后将其从本地发送到等级0以进行打印。

我尝试过的代码段是:

使用C语言分配内存:

  • calloc
// N is the matrix size (N x N)
int* A=( int* )calloc( local_N*N, sizeof(int) );
  • malloc
typedef int row[N];
row *mat;

/* Dividing the number of rows to each processor*/

int n = N / size;

mat = (row *) malloc(n * sizeof(row));

使用C ++语言分配内存:

 int**  matrix = new int *[row];

我成功地使用C编程语言运行,无论如何我想重写C ++的代码。

1 个答案:

答案 0 :(得分:1)

MPI一起使用连续数组非常方便,特别是使用连续数据进行编码要容易得多,例如制作derived data types。我的建议是使用vector并展平您的数据:

const int N = 100;
const int M = 20;
const int size = 4;
int n = N / size;
std::vector<int> mat(n*M); // each process has a mat with length of n * M
for(uint32_t i = 0; i < n; ++i)
    for(uint32_t j = 0; j < M; ++j)
        mat.at(i * M + j) = j; // this is equivalent to mat[i][j] if mat was 2D

您还可以使用smart pointer s:

using ManagedInt = std::unique_ptr<int[]> ;
auto managedMat = std::unique_ptr<ManagedInt[]>(new ManagedInt[n]);
for (size_t i = 0; i < n; ++i)
    managedMat[i] = ManagedInt(new int[M]);

for(uint32_t i = 0; i < n; ++i)
    for(uint32_t j = 0; j < M; ++j)
       managedMat[i][j] = j;

不好的方法是:

警告,您将进入***程序员国。

// allocate mat
int **mat = new int *[n];
for (int i = 0; i < n; i++) {
    mat[i] = new int [M];
}

// now you can use mat[i][j]

// delete mat
for (int i = 0; i < n; i++) {
    delete[] mat[i];
}
delete[] mat;

godbolt