mirror of
https://github.com/NixOS/nixpkgs.git
synced 2026-10-30 08:00:59 +00:00
b665afa036
Diff: https://github.com/NVIDIA/cccl/compare/python-1.2.1...python-1.3.0 Changelog: https://github.com/NVIDIA/cccl/releases/tag/python-1.3.0
26 lines
832 B
Diff
26 lines
832 B
Diff
diff --git a/python/cuda_cccl/cuda/compute/_bindings.py b/python/cuda_cccl/cuda/compute/_bindings.py
|
|
index 4c0012b1bf..5c0ee776ea 100644
|
|
--- a/python/cuda_cccl/cuda/compute/_bindings.py
|
|
+++ b/python/cuda_cccl/cuda/compute/_bindings.py
|
|
@@ -27,17 +27,15 @@ from __future__ import annotations
|
|
import importlib
|
|
import os
|
|
|
|
-from cuda.pathfinder import ( # type: ignore[import-not-found]
|
|
- load_nvidia_dynamic_lib,
|
|
-)
|
|
+from ctypes import CDLL
|
|
|
|
from cuda.cccl._cuda_version_utils import detect_cuda_version, get_recommended_extra
|
|
|
|
|
|
def _load_cuda_libraries():
|
|
# Load appropriate libraries for the detected CUDA version.
|
|
- for libname in ("nvrtc", "nvJitLink"):
|
|
- load_nvidia_dynamic_lib(libname)
|
|
+ CDLL("@libnvrtc@/lib/libnvrtc.so")
|
|
+ CDLL("@libnvjitlink@/lib/libnvJitLink.so")
|
|
|
|
|
|
_load_cuda_libraries()
|