Browse Source

Do not set ncudagpus each time _starpu_init_cuda is called, it is useless

Samuel Thibault 8 years ago
parent
commit
03c9afe025
1 changed files with 5 additions and 2 deletions
  1. 5 2
      src/drivers/cuda/driver_cuda.c

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

@@ -446,8 +446,11 @@ unsigned _starpu_get_cuda_device_count(void)
 
 void _starpu_init_cuda(void)
 {
-	ncudagpus = _starpu_get_cuda_device_count();
-	STARPU_ASSERT(ncudagpus <= STARPU_MAXCUDADEVS);
+	if (ncudagpus < 0)
+	{
+		ncudagpus = _starpu_get_cuda_device_count();
+		STARPU_ASSERT(ncudagpus <= STARPU_MAXCUDADEVS);
+	}
 }
 
 static int start_job_on_cuda(struct _starpu_job *j, struct _starpu_worker *worker, unsigned char pipeline_idx STARPU_ATTRIBUTE_UNUSED)