Skip to content

Commit 08919d9

Browse files
committed
Load nvrtc using path_finder
1 parent dc63296 commit 08919d9

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

cuda_core/cuda/core/__init__.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,4 +2,7 @@
22
#
33
# SPDX-License-Identifier: Apache-2.0
44

5+
from cuda.bindings.path_finder import _load_nvidia_dynamic_library
56
from cuda.core._version import __version__
7+
8+
""" _load_nvidia_dynamic_library("nvrtc") """

cuda_core/pyproject.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,8 +46,8 @@ dependencies = [
4646
]
4747

4848
[project.optional-dependencies]
49-
cu11 = ["cuda-bindings==11.8.*"]
50-
cu12 = ["cuda-bindings==12.*"]
49+
cu11 = ["cuda-bindings[all]==11.8.*"]
50+
cu12 = ["cuda-bindings[all]==12.*"]
5151
test = ["cython>=3.0", "setuptools", "pytest>=6.2.4"]
5252
test-cu11 = ["cuda-core[test]", "cupy-cuda11x", "nvidia-cuda-runtime-cu11"] # runtime headers needed by CuPy
5353
test-cu12 = ["cuda-core[test]", "cupy-cuda12x", "nvidia-cuda-runtime-cu12"] # runtime headers needed by CuPy

0 commit comments

Comments
 (0)