c++ MPI_Scatter MPI_Gather计算平均数

void t6(int argumentcount, char* argumentvector[]) {

	int process_id, num_process;
	int namelen;
	char processor_name[MPI_MAX_PROCESSOR_NAME];
	MPI_Init(&argumentcount, &argumentvector);
	MPI_Comm_rank(MPI_COMM_WORLD, &process_id);
	MPI_Comm_size(MPI_COMM_WORLD, &num_process);
	MPI_Get_processor_name(processor_name, &namelen);
	fprintf(stderr, "Hello world! Processor %d of %d on %s \n", process_id, num_process, processor_name);
	MPI_Barrier(MPI_COMM_WORLD);
	srand(time(NULL) + process_id * 10); // 初始化随机数发生器,srand() 用来设置 rand() 产生随机数时的随机数种子。参数 seed 必须是个整数,如果每次 seed 都设相同值,rand() 所产生的随机数值每次就会一样。

	double *send_data1 = nullptr, send_data2;
	double *recv_data1 = nullptr, *recv_data2 = nullptr;
	int num_sendrecv = 3;

	if (process_id == 0) {
		send_data1 = new double[num_sendrecv * num_process];
		std::cout << "send_data1" << std::endl;
		for (int i = 0; i < num_sendrecv*num_process; ++i) {
			send_data1[i] = ((double)rand() / (double)RAND_MAX)*100.0;
			std::cout << send_data1[i] << std::endl;
		}
	}

	recv_data1 = new double[num_sendrecv];

	MPI_Scatter(send_data1, num_sendrecv, MPI_DOUBLE, recv_data1, num_sendrecv, MPI_DOUBLE, 0, MPI_COMM_WORLD);

	std::string out_info = "process" + std::to_string(process_id) + "_recv_data1: ";
	for (int i = 0; i < num_sendrecv; ++i) {
		out_info += std::to_string(recv_data1[i]) +  ", ";
	}
	std::cout << out_info << std::endl;
	send_data2 = cal_avg(recv_data1, num_sendrecv);
	std::cout << "process" + std::to_string(process_id) + "_avg: " + std::to_string(send_data2) << std::endl;

	delete recv_data1;
	delete send_data1;

	if (process_id == 0) {
		recv_data2 = new double[num_process];
	}

	MPI_Gather(&send_data2, 1, MPI_DOUBLE, recv_data2, 1, MPI_DOUBLE, 0, MPI_COMM_WORLD);

	if (process_id == 0) {
		std::cout << "the following averages were calculated for each set of data" << std::endl;
		for (int i = 0; i < num_process; ++i) {
			std::cout << "process" << i << " " << recv_data2[i] << std::endl;
		}
	}

	delete recv_data2;
	

	MPI_Finalize();

}

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
好的,下面是使用MPI_Scatterv和MPI_Gatherv来计算矩阵乘法的示例代码: ```c #include <stdio.h> #include <stdlib.h> #include <mpi.h> #define MATRIX_SIZE 4 int main(int argc, char* argv[]) { int rank, size; MPI_Init(&argc, &argv); MPI_Comm_rank(MPI_COMM_WORLD, &rank); MPI_Comm_size(MPI_COMM_WORLD, &size); if (size != MATRIX_SIZE) { printf("Error: the number of processes must be %d\n", MATRIX_SIZE); MPI_Abort(MPI_COMM_WORLD, 1); } int matrix_a[MATRIX_SIZE][MATRIX_SIZE] = { {1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16} }; int matrix_b[MATRIX_SIZE][MATRIX_SIZE] = { {1, 2, 3, 4}, {5, 6, 7, 8}, {9, 10, 11, 12}, {13, 14, 15, 16} }; int local_matrix_a[MATRIX_SIZE][MATRIX_SIZE/MATRIX_SIZE]; int local_matrix_b[MATRIX_SIZE/MATRIX_SIZE][MATRIX_SIZE]; int local_matrix_c[MATRIX_SIZE/MATRIX_SIZE][MATRIX_SIZE/MATRIX_SIZE]; int sendcounts[MATRIX_SIZE]; int displs[MATRIX_SIZE]; // scatter matrix_a for (int i = 0; i < MATRIX_SIZE; i++) { int local_cols = MATRIX_SIZE / size; sendcounts[i] = local_cols; displs[i] = i * local_cols; } MPI_Scatterv(matrix_a, sendcounts, displs, MPI_INT, local_matrix_a, MATRIX_SIZE * MATRIX_SIZE / size, MPI_INT, 0, MPI_COMM_WORLD); // scatter matrix_b for (int i = 0; i < MATRIX_SIZE; i++) { int local_rows = MATRIX_SIZE / size; sendcounts[i] = local_rows; displs[i] = i * local_rows; } MPI_Scatterv(matrix_b, sendcounts, displs, MPI_INT, local_matrix_b, MATRIX_SIZE * MATRIX_SIZE / size, MPI_INT, 0, MPI_COMM_WORLD); // calculate local_matrix_c for (int i = 0; i < MATRIX_SIZE/MATRIX_SIZE; i++) { for (int j = 0; j < MATRIX_SIZE/MATRIX_SIZE; j++) { local_matrix_c[i][j] = 0; for (int k = 0; k < MATRIX_SIZE; k++) { local_matrix_c[i][j] += local_matrix_a[i][k] * local_matrix_b[k][j]; } } } // gather matrix_c MPI_Gatherv(local_matrix_c, MATRIX_SIZE * MATRIX_SIZE / size, MPI_INT, matrix_c, sendcounts, displs, MPI_INT, 0, MPI_COMM_WORLD); if (rank == 0) { printf("Matrix C:\n"); for (int i = 0; i < MATRIX_SIZE; i++) { for (int j = 0; j < MATRIX_SIZE; j++) { printf("%d ", matrix_c[i][j]); } printf("\n"); } } MPI_Finalize(); return 0; } ``` 在这个示例中,矩阵A和矩阵B都是4x4的矩阵。我们将使用4个进程来计算矩阵乘积,每个进程计算结果的一部分。使用MPI_Scatterv将矩阵A和矩阵B分发到各个进程中,并使用MPI_Gatherv将结果收集回主进程。在每个进程中,我们使用双重循环计算局部矩阵的乘积,并将结果存储在局部矩阵C中。最后,我们在主进程中输出完整的矩阵C。
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值