gpustack-runtime 0.1.39.post2__py3-none-any.whl → 0.1.40__py3-none-any.whl

gpustack_runtime/detector/pyacl/__init__.py

@@ -384,13 +384,20 @@ def convertStrBytes(func):
 
 
 def _LoadAclLibrary():
+    """
+    Load the library if it isn't loaded already.
+    """
     global aclLib
+
     if aclLib is None:
+        # lock to ensure only one caller loads the library
         libLoadLock.acquire()
+
         try:
+            # ensure the library still isn't loaded
             if aclLib is None:
                 if sys.platform.startswith("win"):
-                    # ACL not supported on Windows
+                    # Do not support Windows yet.
                     raise ACLError(ACL_ERROR_LIBRARY_NOT_FOUND)
                 # Linux path
                 locs = [
@@ -419,6 +426,7 @@ def _LoadAclLibrary():
                 if aclLib is None:
                     raise ACLError(ACL_ERROR_LIBRARY_NOT_FOUND)
         finally:
+            # lock is always freed
             libLoadLock.release()
 
 

gpustack_runtime/detector/pyamdgpu/__init__.py

@@ -212,10 +212,17 @@ class c_amdgpu_gpu_info(_PrintableStructure):
 
 
 def _LoadAMDGPULibrary():
+    """
+    Load the library if it isn't loaded already.
+    """
     global amdgpuLib
+
     if amdgpuLib is None:
+        # lock to ensure only one caller loads the library
         libLoadLock.acquire()
+
         try:
+            # ensure the library still isn't loaded
             if amdgpuLib is None:
                 if sys.platform.startswith("win"):
                     # Do not support Windows yet.
@@ -235,6 +242,7 @@ def _LoadAMDGPULibrary():
                 if amdgpuLib is None:
                     raise AMDGPUError(AMDGPU_ERROR_LIBRARY_NOT_FOUND)
         finally:
+            # lock is always freed
             libLoadLock.release()
 
 

gpustack_runtime/detector/pycuda/__init__.py

@@ -1,10 +1,10 @@
 from __future__ import annotations
 
-import contextlib
 import string
 import sys
 import threading
 from ctypes import *
+from functools import wraps
 from typing import ClassVar
 
 ## C Type mappings ##
@@ -390,10 +390,17 @@ def convertStrBytes(func):
 
 
 def _LoadCudaLibrary():
+    """
+    Load the library if it isn't loaded already.
+    """
     global cudaLib
+
     if cudaLib is None:
+        # lock to ensure only one caller loads the library
         libLoadLock.acquire()
+
         try:
+            # ensure the library still isn't loaded
             if cudaLib is None:
                 if sys.platform.startswith("win"):
                     # Do not support Windows yet.
@@ -411,6 +418,7 @@ def _LoadCudaLibrary():
                 if cudaLib is None:
                     raise CUDAError(CUDA_ERROR_LIBRARY_NOT_FOUND)
         finally:
+            # lock is always freed
             libLoadLock.release()
 
 

gpustack_runtime/detector/pydcmi/__init__.py

@@ -733,14 +733,20 @@ def convertStrBytes(func):
 
 
 def _LoadDcmiLibrary():
+    """
+    Load the library if it isn't loaded already.
+    """
     global dcmiLib
+
     if dcmiLib is None:
+        # lock to ensure only one caller loads the library
         libLoadLock.acquire()
+
         try:
+            # ensure the library still isn't loaded
             if dcmiLib is None:
                 if sys.platform.startswith("win"):
-                    # DCMI is typically used on Linux, but for completeness,
-                    # Windows support would require different path handling.
+                    # Do not support Windows yet.
                     raise DCMIError(DCMI_ERROR_LIBRARY_NOT_FOUND)
                 # Linux path
                 locs = [
@@ -757,6 +763,7 @@ def _LoadDcmiLibrary():
                 if dcmiLib is None:
                     raise DCMIError(DCMI_ERROR_LIBRARY_NOT_FOUND)
         finally:
+            # lock is always freed
             libLoadLock.release()