Bläddra i källkod

mpi/tests: use dynamic memory

Nathalie Furmento 9 år sedan
förälder
incheckning
ed7d000e1b
2 ändrade filer med 10 tillägg och 2 borttagningar
  1. 5 1
      mpi/tests/insert_task_recv_cache.c
  2. 5 1
      mpi/tests/insert_task_sent_cache.c

+ 5 - 1
mpi/tests/insert_task_recv_cache.c

@@ -63,7 +63,7 @@ void test_cache(int rank, char *enabled, size_t *comm_amount)
 {
 	int i;
 	int ret;
-	unsigned v[2][N];
+	unsigned **v;
 	starpu_data_handle_t data_handles[2];
 
 	FPRINTF_MPI(stderr, "Testing with STARPU_MPI_CACHE=%s\n", enabled);
@@ -74,9 +74,11 @@ void test_cache(int rank, char *enabled, size_t *comm_amount)
 	ret = starpu_mpi_init(NULL, NULL, 0);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
 
+	v = malloc(2 * sizeof(unsigned *));
 	for(i = 0; i < 2; i++)
 	{
 		int j;
+		v[i] = malloc(N * sizeof(unsigned));
 		for(j=0 ; j<N ; j++)
 		{
 			v[i][j] = 12;
@@ -121,7 +123,9 @@ void test_cache(int rank, char *enabled, size_t *comm_amount)
 	for(i = 0; i < 2; i++)
 	{
 		starpu_data_unregister(data_handles[i]);
+		free(v[i]);
 	}
+	free(v);
 
 	starpu_mpi_comm_amounts_retrieve(comm_amount);
 	starpu_mpi_shutdown();

+ 5 - 1
mpi/tests/insert_task_sent_cache.c

@@ -63,7 +63,7 @@ void test_cache(int rank, char *enabled, size_t *comm_amount)
 {
 	int i;
 	int ret;
-	unsigned v[2][N];
+	unsigned **v;
 	starpu_data_handle_t data_handles[2];
 
 	setenv("STARPU_MPI_CACHE", enabled, 1);
@@ -73,9 +73,11 @@ void test_cache(int rank, char *enabled, size_t *comm_amount)
 	ret = starpu_mpi_init(NULL, NULL, 0);
 	STARPU_CHECK_RETURN_VALUE(ret, "starpu_mpi_init");
 
+	v = malloc(2 * sizeof(unsigned *));
 	for(i = 0; i < 2; i++)
 	{
 		int j;
+		v[i] = malloc(N * sizeof(unsigned));
 		for(j=0 ; j<N ; j++)
 		{
 			v[i][j] = 12;
@@ -127,7 +129,9 @@ void test_cache(int rank, char *enabled, size_t *comm_amount)
 	for(i = 0; i < 2; i++)
 	{
 		starpu_data_unregister(data_handles[i]);
+		free(v[i]);
 	}
+	free(v);
 
 	starpu_mpi_comm_amounts_retrieve(comm_amount);
 	starpu_mpi_shutdown();