Cédric Augonnet 15 anni fa
parent
commit
c1c1070153
2 ha cambiato i file con 6 aggiunte e 7 eliminazioni
  1. 3 3
      src/drivers/cpu/driver_cpu.c
  2. 3 4
      src/drivers/cuda/driver_cuda.c

+ 3 - 3
src/drivers/cpu/driver_cpu.c

@@ -124,11 +124,12 @@ void *_starpu_cpu_worker(void *arg)
 {
 	struct starpu_worker_s *cpu_arg = arg;
 	struct starpu_jobq_s *jobq = cpu_arg->jobq;
+	unsigned memnode = cpu_arg->memory_node;
 
 #ifdef STARPU_USE_FXT
 	_starpu_fxt_register_thread(cpu_arg->bindid);
 #endif
-	STARPU_TRACE_WORKER_INIT_START(STARPU_FUT_CPU_KEY, cpu_arg->memory_node);
+	STARPU_TRACE_WORKER_INIT_START(STARPU_FUT_CPU_KEY, memnode);
 
 	_starpu_bind_thread_on_cpu(cpu_arg->config, cpu_arg->bindid);
 
@@ -136,7 +137,7 @@ void *_starpu_cpu_worker(void *arg)
         fprintf(stderr, "cpu worker %d is ready on logical cpu %d\n", cpu_arg->devid, cpu_arg->bindid);
 #endif
 
-	_starpu_set_local_memory_node_key(&cpu_arg->memory_node);
+	_starpu_set_local_memory_node_key(&memnode);
 
 	_starpu_set_local_queue(jobq);
 
@@ -159,7 +160,6 @@ void *_starpu_cpu_worker(void *arg)
 
 	struct starpu_sched_policy_s *policy = _starpu_get_sched_policy();
 	struct starpu_jobq_s *queue = policy->starpu_get_local_queue(policy);
-	unsigned memnode = cpu_arg->memory_node;
 
 	while (_starpu_machine_is_running())
 	{

+ 3 - 4
src/drivers/cuda/driver_cuda.c

@@ -198,16 +198,16 @@ void *_starpu_cuda_worker(void *arg)
 	struct starpu_jobq_s *jobq = args->jobq;
 
 	int devid = args->devid;
-	unsigned memory_node = args->memory_node;
+	unsigned memnode = args->memory_node;
 
 #ifdef STARPU_USE_FXT
 	_starpu_fxt_register_thread(args->bindid);
 #endif
-	STARPU_TRACE_WORKER_INIT_START(STARPU_FUT_CUDA_KEY, memory_node);
+	STARPU_TRACE_WORKER_INIT_START(STARPU_FUT_CUDA_KEY, memnode);
 
 	_starpu_bind_thread_on_cpu(args->config, args->bindid);
 
-	_starpu_set_local_memory_node_key(&(args->memory_node));
+	_starpu_set_local_memory_node_key(&memnode);
 
 	_starpu_set_local_queue(jobq);
 
@@ -244,7 +244,6 @@ void *_starpu_cuda_worker(void *arg)
 
 	struct starpu_sched_policy_s *policy = _starpu_get_sched_policy();
 	struct starpu_jobq_s *queue = policy->starpu_get_local_queue(policy);
-	unsigned memnode = args->memory_node;
 	
 	while (_starpu_machine_is_running())
 	{