Просмотр исходного кода

tests/perfmodels/valid_model.c: unload models to clear memory

Nathalie Furmento лет назад: 12
Родитель
Сommit
6e338441c8
1 измененных файлов с 8 добавлено и 0 удалено
  1. 8 0
      tests/perfmodels/valid_model.c

+ 8 - 0
tests/perfmodels/valid_model.c

@@ -86,6 +86,7 @@ static int submit(struct starpu_codelet *codelet, struct starpu_perfmodel *model
 		STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
 	}
         starpu_data_unregister(handle);
+	starpu_perfmodel_unload_model(&lmodel);
 	starpu_shutdown(); // To force dumping perf models on disk
 
 	ret = starpu_perfmodel_load_symbol(codelet->model->symbol, &lmodel);
@@ -99,6 +100,13 @@ static int submit(struct starpu_codelet *codelet, struct starpu_perfmodel *model
 	for (archid = 0; archid < STARPU_NARCH_VARIATIONS; archid++)
 		new_nsamples += lmodel.per_arch[archid][0].regression.nsample;
 
+	ret = starpu_perfmodel_unload_model(&lmodel);
+	if (ret == 1)
+	{
+		FPRINTF(stderr, "The performance model for the symbol <%s> could not be UNloaded\n", codelet->model->symbol);
+		return 1;
+	}
+
 	if (old_nsamples + nloops == new_nsamples)
 	{
 		FPRINTF(stderr, "Sampling for <%s> OK %d + %d == %d\n", codelet->model->symbol, old_nsamples, nloops, new_nsamples);