Skip to content

Commit c336faf

Browse files
authored
relax python versions + fix build linkage error (#24)
1. relax python version to >3.9 2. link `nvshmem::nvshmem_bootstrap_uid` , otherwise we have runtime link error. Signed-off-by: youkaichao <[email protected]>
1 parent 521848e commit c336faf

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

csrc/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -69,6 +69,7 @@ target_link_libraries(pplx_kernels PUBLIC
6969
CUDA::cuda_driver
7070
CUDA::cudart
7171
nvshmem::nvshmem
72+
nvshmem::nvshmem_bootstrap_uid
7273
)
7374
set_target_properties(pplx_kernels PROPERTIES
7475
LIBRARY_OUTPUT_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/../src/pplx_kernels

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ name = "pplx-kernels"
33
version = "0.0.1"
44
description = "Perplexity CUDA Kernels"
55
readme = "README.md"
6-
requires-python = ">=3.12"
6+
requires-python = ">=3.9"
77

88
[build-system]
99
requires = ["setuptools>=61.0", "wheel", "torch"]

0 commit comments

Comments
 (0)