When defined the variable STARPU_PERF_MODEL_DIR will be used to dump …

…perfmodel files.

ChangeLog

@@ -24,6 +24,8 @@ Changes:
 
 Small changes:
   * Fix build system for StarPU Python interface
+  * When defined the variable STARPU_PERF_MODEL_DIR will be used to
+    dump perfmodel files.
 
 New features:
   * Add starpu_data_register_victim_selector to let schedulers select eviction

doc/doxygen/chapters/starpu_basics/starpu_applications.doxy

@@ -441,4 +441,15 @@ mv $HOME/.starpu/sampling /usr/local/starpu/sampling
 STARPU_PERF_MODEL_DIR=/usr/local/starpu/sampling ./application
 \endverbatim
 
+When defined the variable \ref STARPU_PERF_MODEL_DIR will be used to
+dump perfmodel files. The application will read bus performance files
+and codelet performance files from <c>$HOME/.starpu</c> when
+available, and will write in the directory \ref STARPU_PERF_MODEL_DIR.
+
+\verbatim
+$ STARPU_PERF_MODEL_DIR=$(mktemp -d)  STARPU_SCHED=dmda STARPU_CALIBRATE=1 ./examples/cholesky/cholesky_implicit
+...
+[starpu][starpu_save_history_based_model] Going to write performance model in file </tmp/tmp.wZSizZncuU/codelets/45/chol_model_potrf.vesubie> for model <chol_model_potrf>
+\endverbatim
+
 */

src/core/perfmodel/perfmodel.c

@@ -38,6 +38,7 @@
 #endif
 
 static int _starpu_expected_transfer_time_writeback;
+char *_starpu_perf_model_dir = NULL;
 
 void _starpu_init_perfmodel(void)
 {
@@ -772,10 +773,10 @@ void _starpu_set_perf_model_dirs()
 {
 	if (_perf_model_paths_nb != 0) return;
 
-	char *env = starpu_getenv("STARPU_PERF_MODEL_DIR");
-	if (env)
+	_starpu_perf_model_dir = starpu_getenv("STARPU_PERF_MODEL_DIR");
+	if (_starpu_perf_model_dir)
 	{
-		_perf_model_add_dir(env, 0, "by variable STARPU_PERF_MODEL_DIR");
+		_perf_model_add_dir(_starpu_perf_model_dir, 0, "by variable STARPU_PERF_MODEL_DIR");
 	}
 
 #ifdef STARPU_PERF_MODEL_DIR
@@ -786,7 +787,7 @@ void _starpu_set_perf_model_dirs()
 	_perf_model_add_dir(home, 0, "by STARPU_HOME directory");
 #endif
 
-	env = starpu_getenv("STARPU_PERF_MODEL_PATH");
+	char *env = starpu_getenv("STARPU_PERF_MODEL_PATH");
 	if (env)
 	{
 		char *saveptr, *token;

src/core/perfmodel/perfmodel.h

@@ -71,6 +71,7 @@ struct starpu_perfmodel_arch;
 
 extern unsigned _starpu_calibration_minimum;
 extern int _starpu_benchmarking_bus;
+extern char *_starpu_perf_model_dir;
 
 void _starpu_find_perf_model_codelet(const char *symbol, const char *hostname, char *path, size_t maxlen);
 void _starpu_find_perf_model_codelet_debug(const char *symbol, const char *hostname, const char *arch, char *path, size_t maxlen);

src/core/perfmodel/perfmodel_history.c

@@ -1236,12 +1236,14 @@ void starpu_save_history_based_model(struct starpu_perfmodel *model)
 	char path[STR_LONG_LENGTH];
 	starpu_perfmodel_get_model_path(model->symbol, path, sizeof(path));
 
-	if (path[0] == '\0')
+	if (path[0] == '\0' || _starpu_perf_model_dir)
+	{
 		starpu_perfmodel_get_model_path_default_location(model->symbol, path, sizeof(path));
+	}
 
 	free(model->path);
 	model->path = strdup(path);
-	_STARPU_DEBUG("Opening performance model file <%s> for model <%s>\n", path, model->symbol);
+	_STARPU_DEBUG("Going to write performance model in file <%s> for model <%s>\n", path, model->symbol);
 
 	/* overwrite existing file, or create it */
 	FILE *f;