Pārlūkot izejas kodu

Add MemoryNode field to tasks.rec

Samuel Thibault 9 gadi atpakaļ
vecāks
revīzija
a19d9e2e7f
1 mainītis faili ar 6 papildinājumiem un 0 dzēšanām
  1. 6 0
      src/debug/traces/starpu_fxt.c

+ 6 - 0
src/debug/traces/starpu_fxt.c

@@ -78,6 +78,7 @@ struct task_info {
 	unsigned long job_id;
 	uint64_t tag;
 	int workerid;
+	int node;
 	double submit_time;
 	double start_time;
 	double end_time;
@@ -106,6 +107,7 @@ static struct task_info *get_task(unsigned long job_id, int mpi_rank)
 		task->job_id = job_id;
 		task->tag = 0;
 		task->workerid = -1;
+		task->node = -1;
 		task->submit_time = 0.;
 		task->start_time = 0.;
 		task->end_time = 0.;
@@ -155,6 +157,8 @@ static void task_dump(unsigned long job_id, int mpi_rank)
 	fprintf(tasks_file, "Tag: %"PRIx64"\n", task->tag);
 	if (task->workerid >= 0)
 		fprintf(tasks_file, "WorkerId: %d\n", task->workerid);
+	if (task->node >= 0)
+		fprintf(tasks_file, "MemoryNode: %d\n", task->node);
 	if (task->submit_time != 0.)
 		fprintf(tasks_file, "SubmitTime: %f\n", task->submit_time);
 	if (task->start_time != 0.)
@@ -1137,6 +1141,7 @@ static void create_paje_state_if_not_found(char *name, struct starpu_fxt_options
 static void handle_start_codelet_body(struct fxt_ev_64 *ev, struct starpu_fxt_options *options)
 {
 	int worker = ev->param[2];
+	int node = ev->param[3];
 
 	if (worker < 0) return;
 
@@ -1155,6 +1160,7 @@ static void handle_start_codelet_body(struct fxt_ev_64 *ev, struct starpu_fxt_op
 	task->start_time = start_codelet_time;
 	task->workerid = worker;
 	task->name = strdup(name);
+	task->node = node;
 
 #ifndef STARPU_ENABLE_PAJE_CODELET_DETAILS
 	if (out_paje_file)