Samuel Thibault 8 年 前
コミット
6a9541027d
共有1 個のファイルを変更した2 個の追加1 個の削除を含む
  1. 2 1
      src/drivers/cuda/driver_cuda.c

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

@@ -444,6 +444,7 @@ unsigned _starpu_get_cuda_device_count(void)
 	return (unsigned)cnt;
 }
 
+/* This is run from initialize to determine the number of CUDA devices */
 void _starpu_init_cuda(void)
 {
 	if (ncudagpus < 0)
@@ -606,7 +607,7 @@ static void execute_job_on_cuda(struct starpu_task *task, struct _starpu_worker
 	}
 }
 
-/* XXX Should this be merged with _starpu_init_cuda ? */
+/* This is run from the driver to initialize the driver CUDA context */
 int _starpu_cuda_driver_init(struct _starpu_worker_set *worker_set)
 {
 	struct _starpu_worker *worker0 = &worker_set->workers[0];