Something went wrong. Try again.
Nix Packages collection & NixOS github.com/NixOS/nixpkgs
Something went wrong. Try again.
6.8 kB · 166 lines
at master
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167diff --git a/cuda_bindings/cuda/bindings/_internal/cufile_linux.pyx b/cuda_bindings/cuda/bindings/_internal/cufile_linux.pyxindex 78ea77b283..0b875a6cd3 100644--- a/cuda_bindings/cuda/bindings/_internal/cufile_linux.pyx+++ b/cuda_bindings/cuda/bindings/_internal/cufile_linux.pyx@@ -9,7 +9,7 @@ import threading from .utils import FunctionNotFoundError, NotSupportedError -from cuda.pathfinder import load_nvidia_dynamic_lib+from ctypes import CDLL import cython @@ -95,7 +95,7 @@ cdef void* __cuFileSetParameterString = NULL cdef void* load_library() except* with gil:- cdef uintptr_t handle = load_nvidia_dynamic_lib("cufile")._handle_uint+ cdef uintptr_t handle = CDLL("@libcufile@/lib/libcufile.so")._handle return <void*>handle diff --git a/cuda_bindings/cuda/bindings/_internal/nvfatbin_linux.pyx b/cuda_bindings/cuda/bindings/_internal/nvfatbin_linux.pyxindex f5a9bbd218..8271f7aa20 100644--- a/cuda_bindings/cuda/bindings/_internal/nvfatbin_linux.pyx+++ b/cuda_bindings/cuda/bindings/_internal/nvfatbin_linux.pyx@@ -9,7 +9,7 @@ from libc.stdint cimport intptr_t, uintptr_t import threading from .utils import FunctionNotFoundError, NotSupportedError -from cuda.pathfinder import load_nvidia_dynamic_lib+from ctypes import CDLL ###############################################################################@@ -73,7 +73,7 @@ cdef void* __nvFatbinAddTileIR = NULL cdef void* load_library() except* with gil:- cdef uintptr_t handle = load_nvidia_dynamic_lib("nvfatbin")._handle_uint+ cdef uintptr_t handle = CDLL("@libnvfatbin@/lib/libnvfatbin.so")._handle return <void*>handle diff --git a/cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx b/cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyxindex d676aac372..ed3c000566 100644--- a/cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx+++ b/cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx@@ -9,7 +9,7 @@ from libc.stdint cimport intptr_t, uintptr_t import threading from .utils import FunctionNotFoundError, NotSupportedError -from cuda.pathfinder import load_nvidia_dynamic_lib+from ctypes import CDLL ###############################################################################@@ -76,7 +76,7 @@ cdef void* __nvJitLinkVersion = NULL cdef void* load_library() except* with gil:- cdef uintptr_t handle = load_nvidia_dynamic_lib("nvJitLink")._handle_uint+ cdef uintptr_t handle = CDLL("@libnvjitlink@/lib/libnvJitLink.so")._handle return <void*>handle diff --git a/cuda_bindings/cuda/bindings/_internal/nvml_linux.pyx b/cuda_bindings/cuda/bindings/_internal/nvml_linux.pyxindex 51f6e8205b..1dd64a6d5e 100644--- a/cuda_bindings/cuda/bindings/_internal/nvml_linux.pyx+++ b/cuda_bindings/cuda/bindings/_internal/nvml_linux.pyx@@ -10,7 +10,7 @@ import threading from .utils import FunctionNotFoundError, NotSupportedError -from cuda.pathfinder import load_nvidia_dynamic_lib+from ctypes import CDLL ###############################################################################@@ -414,7 +414,13 @@ cdef void* __nvmlGpuInstanceSetVgpuSchedulerState_v2 = NULL cdef void* load_library() except* with gil:- cdef uintptr_t handle = load_nvidia_dynamic_lib("nvml")._handle_uint+ # Search like the dynamic linker would for a library with the driver link in its RUNPATH: LD_LIBRARY_PATH+ # first (e.g. a host driver exposed by nixglhost or a container runtime), then the configured path.+ cdef uintptr_t handle+ try:+ handle = CDLL("libnvidia-ml.so.1")._handle+ except OSError:+ handle = CDLL("@libnvml@/lib/libnvidia-ml.so")._handle return <void*>handle diff --git a/cuda_bindings/cuda/bindings/_internal/nvrtc_linux.pyx b/cuda_bindings/cuda/bindings/_internal/nvrtc_linux.pyxindex d3eb62be13..6d5d093c1b 100644--- a/cuda_bindings/cuda/bindings/_internal/nvrtc_linux.pyx+++ b/cuda_bindings/cuda/bindings/_internal/nvrtc_linux.pyx@@ -9,7 +9,7 @@ from libc.stdint cimport intptr_t, uintptr_t import threading from .utils import FunctionNotFoundError, NotSupportedError -from cuda.pathfinder import load_nvidia_dynamic_lib+from ctypes import CDLL ############################################################################### # Extern@@ -82,7 +82,7 @@ cdef void* __nvrtcGetPCHHeapSizeRequired = NULL cdef void* __nvrtcSetFlowCallback = NULL cdef void* load_library() except* with gil:- cdef uintptr_t handle = load_nvidia_dynamic_lib("nvrtc")._handle_uint+ cdef uintptr_t handle = CDLL("@libnvrtc@/lib/libnvrtc.so")._handle return <void*>handle cdef int _init_nvrtc() except -1 nogil:diff --git a/cuda_bindings/cuda/bindings/_internal/nvvm_linux.pyx b/cuda_bindings/cuda/bindings/_internal/nvvm_linux.pyxindex 03ade4f939..0385da3087 100644--- a/cuda_bindings/cuda/bindings/_internal/nvvm_linux.pyx+++ b/cuda_bindings/cuda/bindings/_internal/nvvm_linux.pyx@@ -9,7 +9,7 @@ from libc.stdint cimport intptr_t, uintptr_t import threading from .utils import FunctionNotFoundError, NotSupportedError -from cuda.pathfinder import load_nvidia_dynamic_lib+from ctypes import CDLL ###############################################################################@@ -76,7 +76,7 @@ cdef void* __nvvmLLVMVersion = NULL cdef void* load_library() except* with gil:- cdef uintptr_t handle = load_nvidia_dynamic_lib("nvvm")._handle_uint+ cdef uintptr_t handle = CDLL("@libnvvm@/lib/libnvvm.so")._handle return <void*>handle diff --git a/cuda_bindings/cuda/bindings/cyruntime.pyx.in b/cuda_bindings/cuda/bindings/cyruntime.pyx.inindex 5cd65fbd96..fbbcffbbb7 100644--- a/cuda_bindings/cuda/bindings/cyruntime.pyx.in+++ b/cuda_bindings/cuda/bindings/cyruntime.pyx.in@@ -1874,7 +1874,7 @@ cdef cudaError_t cudaGraphicsVDPAURegisterOutputSurface(cudaGraphicsResource** r {{if True}} from libc.stdint cimport uintptr_t-from cuda.pathfinder import load_nvidia_dynamic_lib+from ctypes import CDLL {{if 'Windows' == platform.system()}} cimport cuda.bindings._lib.windll as windll {{else}}@@ -1884,11 +1884,11 @@ cimport cuda.bindings._lib.dlfcn as dlfcn cdef cudaError_t getLocalRuntimeVersion(int* runtimeVersion) except ?cudaErrorCallRequiresNewerDriver nogil: # Load with gil:- loaded_dl = load_nvidia_dynamic_lib("cudart")+ loaded_dl = CDLL("@libcudart@/lib/libcudart.so") {{if 'Windows' == platform.system()}}- handle = <uintptr_t>loaded_dl._handle_uint+ handle = <uintptr_t>loaded_dl._handle {{else}}- handle = <void *><uintptr_t>loaded_dl._handle_uint+ handle = <void *><uintptr_t>loaded_dl._handle {{endif}} {{if 'Windows' == platform.system()}}