Nvidia GPUs: Call nvmlInit() once (state-preservation)

LLNL · Jul 7, 2023 · c897425 · c897425
1 parent 6ea2644
commit c897425
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/src/variorum/Nvidia_GPU/config_nvidia.c b/src/variorum/Nvidia_GPU/config_nvidia.c
@@ -20,6 +20,7 @@ uint64_t *detect_gpu_arch(void)
 
 int set_nvidia_func_ptrs(int idx)
 {
+    static int init_complete = 0;
     int err = 0;
 
     if (*g_platform[idx].arch_id == VOLTA)
@@ -39,6 +40,10 @@ int set_nvidia_func_ptrs(int idx)
         err = VARIORUM_ERROR_UNSUPPORTED_PLATFORM;
     }
 
-    initNVML();
+    if (init_complete == 0)
+    {
+        initNVML();
+        init_complete = 1;
+    }
     return err;
 }