|
@@ -26,6 +26,7 @@
|
|
|
#include <starpu_top.h>
|
|
|
#include <core/sched_policy.h>
|
|
|
#include <top/starpu_top_core.h>
|
|
|
+#include <core/debug.h>
|
|
|
|
|
|
void _starpu_driver_start_job(struct _starpu_worker *args, struct _starpu_job *j, struct timespec *codelet_start, int rank, int profiling)
|
|
|
{
|
|
@@ -64,6 +65,9 @@ void _starpu_driver_start_job(struct _starpu_worker *args, struct _starpu_job *j
|
|
|
if (starpu_top)
|
|
|
_starpu_top_task_started(task,workerid,codelet_start);
|
|
|
|
|
|
+#ifdef HAVE_AYUDAME_H
|
|
|
+ if (AYU_event) AYU_event(AYU_RUNTASK, j->job_id, NULL);
|
|
|
+#endif
|
|
|
_STARPU_TRACE_START_CODELET_BODY(j);
|
|
|
}
|
|
|
|
|
@@ -77,6 +81,9 @@ void _starpu_driver_end_job(struct _starpu_worker *args, struct _starpu_job *j,
|
|
|
unsigned calibrate_model = 0;
|
|
|
|
|
|
_STARPU_TRACE_END_CODELET_BODY(j, j->nimpl, perf_arch);
|
|
|
+#ifdef HAVE_AYUDAME_H
|
|
|
+ if (AYU_event) AYU_event(AYU_POSTRUNTASK, j->job_id, NULL);
|
|
|
+#endif
|
|
|
|
|
|
if (cl && cl->model && cl->model->benchmarking)
|
|
|
calibrate_model = 1;
|
|
@@ -181,5 +188,12 @@ struct starpu_task *_starpu_get_worker_task(struct _starpu_worker *args, int wor
|
|
|
_starpu_worker_set_status(workerid, STATUS_UNKNOWN);
|
|
|
}
|
|
|
|
|
|
+#ifdef HAVE_AYUDAME_H
|
|
|
+ if (AYU_event) {
|
|
|
+ int id = workerid;
|
|
|
+ AYU_event(AYU_PRERUNTASK, _starpu_get_job_associated_to_task(task)->job_id, &id);
|
|
|
+ }
|
|
|
+#endif
|
|
|
+
|
|
|
return task;
|
|
|
}
|