NVIDIA/nvbench
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvbench.git synced 2026-03-14 20:27:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use pybind11==3.0.1, do not use pybind11_add_module

Browse Source
This commit is contained in:
Oleksandr Pavlyk
2025-12-05 14:39:11 -06:00
parent f651636501
commit de471e1d42
1 changed files with 5 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
python/CMakeLists.txt
View File

@@ -23,15 +23,17 @@ CPMAddPackage(
FIND_PACKAGE_ARGS CONFIG REQUIRED
)
CPMAddPackage("gh:pybind/pybind11@3.0.0")
CPMAddPackage("gh:pybind/pybind11@3.0.1")
pybind11_add_module(_nvbench MODULE src/py_nvbench.cpp)
add_library(_nvbench MODULE src/py_nvbench.cpp)
target_include_directories(_nvbench PRIVATE ${Python_INCLUDE_DIRS})
target_link_libraries(_nvbench PUBLIC nvbench::nvbench)
target_link_libraries(_nvbench PRIVATE CUDA::cudart_static)
target_link_libraries(_nvbench PRIVATE CUDA::cudart_static pybind11::headers)
set_target_properties(_nvbench PROPERTIES INSTALL_RPATH "$ORIGIN")
set_target_properties(_nvbench PROPERTIES INTERPROCEDURAL_OPTIMIZATION ON)
set_target_properties(_nvbench PROPERTIES POSITION_INDEPENDENT_CODE ON)
set_target_properties(_nvbench PROPERTIES PREFIX "" SUFFIX "${PYTHON_MODULE_EXTENSION}")
install(TARGETS _nvbench DESTINATION cuda/bench)