Skip to content

Commit

Permalink
Merge pull request #12828 from janjust/v5.0.x
Browse files Browse the repository at this point in the history
accelecrator/cuda: fix for unresolved symbol
  • Loading branch information
janjust authored Sep 30, 2024
2 parents 75795c0 + 53b9f4a commit 49ca7e6
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions opal/mca/accelerator/cuda/accelerator_cuda_component.c
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down Expand Up @@ -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);
Expand Down

0 comments on commit 49ca7e6

Please sign in to comment.