Forráskód Böngészése

mpi/src/starpu_mpi_cache.c: free dynamic memory when unsetting cache

Nathalie Furmento 10 éve
szülő
commit
1afd0e4f02
1 módosított fájl, 3 hozzáadás és 0 törlés
  1. 3 0
      mpi/src/starpu_mpi_cache.c

+ 3 - 0
mpi/src/starpu_mpi_cache.c

@@ -52,7 +52,10 @@ int starpu_mpi_cache_set(int enabled)
 		if (_starpu_cache_enabled)
 		{
 			// We need to clean the cache
+			int world_size;
 			starpu_mpi_cache_flush_all_data(MPI_COMM_WORLD);
+			starpu_mpi_comm_size(MPI_COMM_WORLD, &world_size);
+			_starpu_mpi_cache_free(world_size);
 		}
 		_starpu_cache_enabled = 0;
 	}