/* registered_models_rwlock must be taken in write mode before calling this
* function */
void _starpu_register_model(struct starpu_perfmodel_t *model)
{
/* add the model to a linked list */
struct starpu_model_list_t *node = malloc(sizeof(struct starpu_model_list_t));
node->model = model;
//model->debug_modelid = debug_modelid++;
/* put this model at the beginning of the list */
node->next = registered_models;
registered_models = node;
#ifdef STARPU_MODEL_DEBUG
_starpu_create_sampling_directory_if_needed();
unsigned arch;
for (arch = 0; arch < NARCH_VARIATIONS; arch++)
{
char debugpath[256];
starpu_perfmodel_debugfilepath(model, arch, debugpath, 256);
model->per_arch[arch].debug_file = fopen(debugpath, "a+");
STARPU_ASSERT(model->per_arch[arch].debug_file);
}
#endif
return;
}
void starpu_perfmodel_print(struct starpu_perfmodel *model, struct starpu_perfmodel_arch* arch, unsigned nimpl, char *parameter, uint32_t *footprint, FILE *output)
{
int comb = starpu_perfmodel_arch_comb_get(arch->ndevices, arch->devices);
STARPU_ASSERT(comb != -1);
struct starpu_perfmodel_per_arch *arch_model = &model->state->per_arch[comb][nimpl];
char archname[32];
if (arch_model->regression.nsample || arch_model->regression.valid || arch_model->regression.nl_valid || arch_model->list)
{
starpu_perfmodel_get_arch_name(arch, archname, 32, nimpl);
fprintf(output, "# performance model for %s\n", archname);
}
if (parameter == NULL)
{
/* no specific parameter was requested, so we display everything */
if (arch_model->regression.nsample)
{
fprintf(output, "\tRegression : #sample = %u\n", arch_model->regression.nsample);
}
/* Only display the regression model if we could actually build a model */
if (arch_model->regression.valid)
{
fprintf(output, "\tLinear: y = alpha size ^ beta\n");
fprintf(output, "\t\talpha = %e\n", arch_model->regression.alpha);
fprintf(output, "\t\tbeta = %e\n", arch_model->regression.beta);
}
else
{
//fprintf(output, "\tLinear model is INVALID\n");
}
if (arch_model->regression.nl_valid)
{
fprintf(output, "\tNon-Linear: y = a size ^b + c\n");
fprintf(output, "\t\ta = %e\n", arch_model->regression.a);
fprintf(output, "\t\tb = %e\n", arch_model->regression.b);
fprintf(output, "\t\tc = %e\n", arch_model->regression.c);
}
else
{
//fprintf(output, "\tNon-Linear model is INVALID\n");
}
_starpu_perfmodel_print_history_based(arch_model, parameter, footprint, output);
#if 0
char debugname[1024];
starpu_perfmodel_debugfilepath(model, arch, debugname, 1024, nimpl);
printf("\t debug file path : %s\n", debugname);
#endif
}
else
{
/* only display the parameter that was specifically requested */
if (strcmp(parameter, "a") == 0)
{
printf("%e\n", arch_model->regression.a);
return;
}
if (strcmp(parameter, "b") == 0)
{
printf("%e\n", arch_model->regression.b);
return;
}
if (strcmp(parameter, "c") == 0)
{
printf("%e\n", arch_model->regression.c);
return;
}
if (strcmp(parameter, "alpha") == 0)
{
printf("%e\n", arch_model->regression.alpha);
return;
}
if (strcmp(parameter, "beta") == 0)
{
printf("%e\n", arch_model->regression.beta);
return;
}
if (strcmp(parameter, "path-file-debug") == 0)
{
char debugname[256];
starpu_perfmodel_debugfilepath(model, arch, debugname, 1024, nimpl);
printf("%s\n", debugname);
return;
}
if ((strcmp(parameter, "mean") == 0) || (strcmp(parameter, "stddev") == 0))
{
_starpu_perfmodel_print_history_based(arch_model, parameter, footprint, output);
return;
}
/* TODO display if it's valid ? */
fprintf(output, "Unknown parameter requested, aborting.\n");
exit(-1);
}
}