From 53b9f4ab42ec7a64cf91998eed2f9f83fd1f8a5b Mon Sep 17 00:00:00 2001 From: Tomislav Janjusic Date: Fri, 27 Sep 2024 23:15:07 +0300 Subject: [PATCH] accelecrator/cuda: fix for unresolved symbol Signed-off-by: Tomislav Janjusic bot:notacherrypick --- opal/mca/accelerator/cuda/accelerator_cuda_component.c | 3 +++ 1 file changed, 3 insertions(+) diff --git a/opal/mca/accelerator/cuda/accelerator_cuda_component.c b/opal/mca/accelerator/cuda/accelerator_cuda_component.c index d48e29c9f65..3a4c5ef48b9 100644 --- a/opal/mca/accelerator/cuda/accelerator_cuda_component.c +++ b/opal/mca/accelerator/cuda/accelerator_cuda_component.c @@ -36,6 +36,7 @@ /* Define global variables, used in accelerator_cuda.c */ CUstream opal_accelerator_cuda_memcpy_stream = NULL; opal_mutex_t opal_accelerator_cuda_stream_lock = {0}; +int opal_accelerator_cuda_num_devices = 0; /* Initialization lock for delayed cuda initialization */ static opal_mutex_t accelerator_cuda_init_lock; @@ -137,6 +138,8 @@ int opal_accelerator_cuda_delayed_init() goto out; } + cuDeviceGetCount(&opal_accelerator_cuda_num_devices); + /* Check to see if this process is running in a CUDA context. If * so, all is good. If not, then disable registration of memory. */ result = cuCtxGetCurrent(&cuContext);