We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
2 parents 75795c0 + 53b9f4a commit 49ca7e6Copy full SHA for 49ca7e6
opal/mca/accelerator/cuda/accelerator_cuda_component.c
@@ -36,6 +36,7 @@
36
/* Define global variables, used in accelerator_cuda.c */
37
CUstream opal_accelerator_cuda_memcpy_stream = NULL;
38
opal_mutex_t opal_accelerator_cuda_stream_lock = {0};
39
+int opal_accelerator_cuda_num_devices = 0;
40
41
/* Initialization lock for delayed cuda initialization */
42
static opal_mutex_t accelerator_cuda_init_lock;
@@ -137,6 +138,8 @@ int opal_accelerator_cuda_delayed_init()
137
138
goto out;
139
}
140
141
+ cuDeviceGetCount(&opal_accelerator_cuda_num_devices);
142
+
143
/* Check to see if this process is running in a CUDA context. If
144
* so, all is good. If not, then disable registration of memory. */
145
result = cuCtxGetCurrent(&cuContext);
0 commit comments