소스 검색

Always reset the local worker key to worker0 for datawizard to get the proper wait queue when using multiple workers in the same driver thread

Samuel Thibault 8 년 전
부모
커밋
5a5eb593d1
1개의 변경된 파일1개의 추가작업 그리고 0개의 파일을 삭제
  1. 1 0
      src/drivers/cuda/driver_cuda.c

+ 1 - 0
src/drivers/cuda/driver_cuda.c

@@ -727,6 +727,7 @@ int _starpu_cuda_driver_run_once(struct _starpu_worker_set *worker_set)
 #ifdef STARPU_SIMGRID
 	starpu_pthread_wait_reset(&worker0->wait);
 #endif
+	_starpu_set_local_worker_key(worker0);
 
 	/* First poll for completed jobs */
 	idle_tasks = 0;