MPI_Comm_split(MPI_COMM_WORLD, col_color, col_key, &col_comm); MPI_Comm_split(MPI_COMM_WORLD, row_color, row_key, &row_comm); MPI_Type_vector(n, block_size, n, MPI_DOUBLE, &col_type); MPI_Type_commit(&col_type); MPI_Type_vector(block_size, n, n, MPI_DOUBLE, &row_type); MPI_Type_commit(&row_type); MPI_Scatter(&B[0][0], 1, col_type, &B_col[0][0], block_size * n, MPI_DOUBLE, 0, row_comm);散播时,mpi似乎没有将每列发送给对应进程,因为地址不在下一列开始,如何解决?
时间: 2024-03-02 13:53:09 浏览: 23
从您提供的代码来看,您已经使用了 `MPI_Type_vector()` 函数来创建一个自定义的数据类型 `col_type` 和 `row_type`,用于将矩阵 `B` 的列和行发送到不同的进程中。但是,在使用 `MPI_Scatter()` 函数进行散播时,您似乎没有使用正确的参数传递自定义的数据类型。
在使用 `MPI_Scatter()` 函数时,第二个参数应该指定要发送的数据块的数量,而不是每个进程应该接收到的数据块的数量。因此,您应该将第二个参数设置为 `num_cols`,表示要发送的列数,而不是 `1`,表示每个进程应该接收到的列数。此外,您应该使用 `col_type` 而不是 `row_type` 来发送数据到列通信子中的进程。
以下是修改后的代码片段:
```
MPI_Comm_split(MPI_COMM_WORLD, col_color, col_key, &col_comm);
MPI_Comm_split(MPI_COMM_WORLD, row_color, row_key, &row_comm);
MPI_Type_vector(n, block_size, n, MPI_DOUBLE, &col_type);
MPI_Type_commit(&col_type);
MPI_Type_vector(block_size, n, n, MPI_DOUBLE, &row_type);
MPI_Type_commit(&row_type);
MPI_Scatter(&B[0][0], num_cols, col_type, &B_col[0][0], block_size * n, MPI_DOUBLE, 0, col_comm);
```
请注意,这里的 `num_cols` 应该是矩阵 `B` 中列的数量,而不是在创建列通信子时指定的进程数。因此,您需要在代码中设置一个新的变量来存储矩阵 `B` 中的列数,并将其传递给 `MPI_Scatter()` 函数。
另外,如果您发现每个进程接收到的数据仍然不正确,那么您可能需要检查自定义数据类型的定义是否正确,并确保每个进程在接收数据时使用正确的数据类型。