|
@@ -485,9 +485,7 @@ static void _starpu_initialize_workers_cuda_gpuid(struct _starpu_machine_config
|
|
|
struct _starpu_machine_topology *topology = &config->topology;
|
|
|
struct starpu_conf *uconf = &config->conf;
|
|
|
|
|
|
- _starpu_initialize_workers_deviceid(uconf->use_explicit_workers_cuda_gpuid == 0
|
|
|
- ? NULL
|
|
|
- : (int *)uconf->workers_cuda_gpuid,
|
|
|
+ _starpu_initialize_workers_deviceid(uconf->use_explicit_workers_cuda_gpuid == 0 ? NULL : (int *)uconf->workers_cuda_gpuid,
|
|
|
&(config->current_cuda_gpuid),
|
|
|
(int *)topology->workers_cuda_gpuid,
|
|
|
"STARPU_WORKERS_CUDAID",
|
|
@@ -1076,7 +1074,7 @@ unsigned _starpu_topology_get_nhwcpu(struct _starpu_machine_config *config)
|
|
|
#endif
|
|
|
#if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
|
|
|
if (config->conf.ncuda != 0)
|
|
|
- _starpu_init_cuda();
|
|
|
+ _starpu_cuda_init();
|
|
|
#endif
|
|
|
_starpu_init_topology(config);
|
|
|
|
|
@@ -1091,7 +1089,7 @@ unsigned _starpu_topology_get_nhwpu(struct _starpu_machine_config *config)
|
|
|
#endif
|
|
|
#if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
|
|
|
if (config->conf.ncuda != 0)
|
|
|
- _starpu_init_cuda();
|
|
|
+ _starpu_cuda_init();
|
|
|
#endif
|
|
|
_starpu_init_topology(config);
|
|
|
|
|
@@ -1106,7 +1104,7 @@ unsigned _starpu_topology_get_nnumanodes(struct _starpu_machine_config *config S
|
|
|
#endif
|
|
|
#if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
|
|
|
if (config->conf.ncuda != 0)
|
|
|
- _starpu_init_cuda();
|
|
|
+ _starpu_cuda_init();
|
|
|
#endif
|
|
|
_starpu_init_topology(config);
|
|
|
|
|
@@ -1459,7 +1457,7 @@ static int _starpu_init_machine_config(struct _starpu_machine_config *config, in
|
|
|
#endif
|
|
|
#if defined(STARPU_USE_CUDA) || defined(STARPU_SIMGRID)
|
|
|
if (config->conf.ncuda != 0)
|
|
|
- _starpu_init_cuda();
|
|
|
+ _starpu_cuda_init();
|
|
|
#endif
|
|
|
_starpu_init_topology(config);
|
|
|
|
|
@@ -1497,7 +1495,7 @@ static int _starpu_init_machine_config(struct _starpu_machine_config *config, in
|
|
|
{
|
|
|
/* The user did not disable CUDA. We need to initialize CUDA
|
|
|
* early to count the number of devices */
|
|
|
- _starpu_init_cuda();
|
|
|
+ _starpu_cuda_init();
|
|
|
int nb_devices = _starpu_get_cuda_device_count();
|
|
|
|
|
|
STARPU_ASSERT_MSG(ncuda >= -1, "ncuda can not be negative and different from -1 (is is %d)", ncuda);
|
|
@@ -1891,7 +1889,7 @@ int _starpu_bind_thread_on_cpu(int cpuid STARPU_ATTRIBUTE_UNUSED, int workerid S
|
|
|
#endif
|
|
|
#ifdef STARPU_USE_CUDA
|
|
|
if (config->conf.ncuda != 0)
|
|
|
- _starpu_init_cuda();
|
|
|
+ _starpu_cuda_init();
|
|
|
#endif
|
|
|
_starpu_init_topology(config);
|
|
|
|
|
@@ -2008,7 +2006,7 @@ void _starpu_bind_thread_on_cpus(struct _starpu_combined_worker *combined_worker
|
|
|
#endif
|
|
|
#ifdef STARPU_USE_CUDA
|
|
|
if (config->conf.ncuda != 0)
|
|
|
- _starpu_init_cuda();
|
|
|
+ _starpu_cuda_init();
|
|
|
#endif
|
|
|
_starpu_init_topology(config);
|
|
|
|