Skip to content

Commit 3024f18

Browse files
committed
Support PTDS for Runtime
1 parent 9e08cef commit 3024f18

File tree

7 files changed

+8273
-889
lines changed

7 files changed

+8273
-889
lines changed

cuda_bindings/cuda/bindings/_bindings/cyruntime.pxd.in

+877-887
Large diffs are not rendered by default.

cuda_bindings/cuda/bindings/_bindings/cyruntime.pxi.in

+1,496
Large diffs are not rendered by default.

cuda_bindings/cuda/bindings/_bindings/cyruntime.pyx.in

+2,653
Large diffs are not rendered by default.

cuda_bindings/cuda/bindings/_bindings/cyruntime_ptds.pxd.in

+1,475
Large diffs are not rendered by default.

cuda_bindings/cuda/bindings/_bindings/cyruntime_ptds.pyx.in

+1,767
Large diffs are not rendered by default.

cuda_bindings/cuda/bindings/cydriver.pxd.in

+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
# Copyright 2021-2024 NVIDIA Corporation. All rights reserved.
1+
# Copyright 2021-2025 NVIDIA Corporation. All rights reserved.
22
#
33
# Please refer to the NVIDIA end user license agreement (EULA) associated
44
# with this source code for terms and conditions that govern your use of

cuda_bindings/setup.py

+4-1
Original file line numberDiff line numberDiff line change
@@ -345,8 +345,11 @@ def do_cythonize(extensions):
345345
["cuda/bindings/_internal/utils.pyx"],
346346
]
347347

348+
libraries = ["cudart_static"]
349+
if sys.platform == "linux":
350+
libraries += ["rt"]
348351
for sources in sources_list:
349-
extensions += prep_extensions(sources, ["cudart_static"])
352+
extensions += prep_extensions(sources, libraries)
350353

351354
# ---------------------------------------------------------------------
352355
# Custom cmdclass extensions

0 commit comments

Comments
 (0)