瀏覽代碼

Fix sendrecv_bench with more than 2 nodes

Samuel Thibault 5 年之前
父節點
當前提交
d491e56f9b
共有 1 個文件被更改,包括 14 次插入0 次删除
  1. 14 0
      mpi/tests/sendrecv_bench.c

+ 14 - 0
mpi/tests/sendrecv_bench.c

@@ -117,6 +117,20 @@ int main(int argc, char **argv)
 
 	if (rank >= 2)
 	{
+		starpu_pause();
+		for (uint64_t s = NX_MIN; s <= NX_MAX; s = _next(s, multiplier, increment))
+		{
+			iterations = _iterations(iterations, s);
+
+			starpu_mpi_barrier(MPI_COMM_WORLD);
+
+			for (uint64_t j = 0; j < iterations; j++)
+			{
+				starpu_mpi_barrier(MPI_COMM_WORLD);
+			}
+		}
+		starpu_resume();
+
 		starpu_mpi_shutdown();
 		if (!mpi_init)
 			MPI_Finalize();