|
|
@@ -144,6 +144,15 @@ void _starpu_handle_job_termination(struct _starpu_job *j)
|
|
|
|
|
|
_STARPU_PTHREAD_MUTEX_UNLOCK(&j->sync_mutex);
|
|
|
|
|
|
+#ifdef STARPU_USE_SCHED_CTX_HYPERVISOR
|
|
|
+ int workerid = starpu_worker_get_id();
|
|
|
+ int i;
|
|
|
+ size_t data_size = 0;
|
|
|
+ for(i = 0; i < STARPU_NMAXBUFS; i++)
|
|
|
+ if(task->handles[i] != NULL)
|
|
|
+ data_size += _starpu_data_get_size(task->handles[i]);
|
|
|
+#endif //STARPU_USE_SCHED_CTX_HYPERVISOR
|
|
|
+
|
|
|
/* We release handle reference count */
|
|
|
if (task->cl)
|
|
|
{
|
|
|
@@ -210,13 +219,6 @@ void _starpu_handle_job_termination(struct _starpu_job *j)
|
|
|
{
|
|
|
_starpu_sched_post_exec_hook(task);
|
|
|
#ifdef STARPU_USE_SCHED_CTX_HYPERVISOR
|
|
|
- int workerid = starpu_worker_get_id();
|
|
|
- int i;
|
|
|
- size_t data_size = 0;
|
|
|
- for(i = 0; i < STARPU_NMAXBUFS; i++)
|
|
|
- if(task->handles[i] != NULL)
|
|
|
- data_size += _starpu_data_get_size(task->handles[i]);
|
|
|
-
|
|
|
starpu_call_poped_task_cb(workerid, task->sched_ctx, task->flops, data_size);
|
|
|
#endif //STARPU_USE_SCHED_CTX_HYPERVISOR
|
|
|
}
|