CMakeLists.txt 1.35 KB
Newer Older
traveller59's avatar
traveller59 committed
1
2
3
4
5
6
7
8
9
10
11
if (SPCONV_BuildCUDA)
    add_library(spconv_nms STATIC nms.cu)
    set_target_properties(spconv_nms PROPERTIES VERSION ${PROJECT_VERSION})
    set_target_properties(spconv_nms PROPERTIES SOVERSION 1)
    target_include_directories(spconv_nms PRIVATE ${ALL_INCLUDE})
    set_property(TARGET spconv_nms PROPERTY CXX_STANDARD 14)
    set_property(TARGET spconv_nms PROPERTY CUDA_STANDARD 14)
    set_property(TARGET spconv_nms PROPERTY POSITION_INDEPENDENT_CODE ON)
    target_link_libraries(spconv_nms ${CUDA_CUDART})
    install (TARGETS spconv_nms DESTINATION lib)
endif()
traveller59's avatar
traveller59 committed
12
13
14
15
16
17
18
19
20
21

add_library(spconv_utils SHARED all.cc)
set_target_properties(spconv_utils PROPERTIES VERSION ${PROJECT_VERSION})
set_target_properties(spconv_utils PROPERTIES SOVERSION 1)
target_include_directories(spconv_utils PRIVATE ${ALL_INCLUDE}
                    ${PROJECT_SOURCE_DIR}/third_party/pybind11/include)
set_property(TARGET spconv_utils PROPERTY CXX_STANDARD 14)
set_property(TARGET spconv_utils PROPERTY CUDA_STANDARD 14)
set_target_properties(spconv_utils PROPERTIES PREFIX "${PYTHON_MODULE_PREFIX}"
                                         SUFFIX "${PYTHON_MODULE_EXTENSION}")
traveller59's avatar
traveller59 committed
22
23
24
25
26
if (SPCONV_BuildCUDA)
    target_link_libraries(spconv_utils ${CUDA_CUDART} pybind11::module spconv_nms)
else()
    target_link_libraries(spconv_utils pybind11::module)
endif()
traveller59's avatar
traveller59 committed
27
install (TARGETS spconv_utils DESTINATION lib)