Fix crash on systems with cuda but not libnvidia-ml.so.1

starpu-runtime · Apr 2, 2024 · 7d883b4 · 7d883b4
1 parent 0818b90
commit 7d883b4
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/drivers/cuda/driver_cuda.c b/src/drivers/cuda/driver_cuda.c
@@ -243,9 +243,12 @@ void _starpu_cuda_discover_devices(struct _starpu_machine_config *config)
 #ifdef STARPU_HAVE_NVML_H
 	void *nvml = dlopen("libnvidia-ml.so.1", RTLD_LAZY);
 
-	_starpu_nvmlInit = dlsym(nvml, "nvmlInit_v2");
-	if (!_starpu_nvmlInit)
-		_starpu_nvmlInit = dlsym(nvml, "nvmlInit");
+	if (nvml)
+	{
+		_starpu_nvmlInit = dlsym(nvml, "nvmlInit_v2");
+		if (!_starpu_nvmlInit)
+			_starpu_nvmlInit = dlsym(nvml, "nvmlInit");
+	}
 	if (_starpu_nvmlInit)
 	{
 		_starpu_nvmlDeviceGetNvLinkState = dlsym(nvml, "nvmlDeviceGetNvLinkState");