Techt3o's picture
45523e4c6d179b4a6cb59cbd723522dfca93224ef225a4e21b493fe9ea37a2ab
43605c2 verified
raw
history blame
518 Bytes
# optional libraries
find_package(Pangolin REQUIRED)
set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS}")
pybind11_add_module(dpviewerx viewer.cpp viewer_cuda.cu)
include_directories(${Pangolin_INCLUDE_DIRS})
# EXAMPLE_VERSION_INFO is defined by setup.py and passed into the C++ code as a
# define (VERSION_INFO) here.
target_compile_definitions(dpviewerx PRIVATE VERSION_INFO=${EXAMPLE_VERSION_INFO})
target_link_libraries(dpviewerx PRIVATE ${TORCH_LIBRARIES} ${TORCH_PYTHON_LIBRARY} ${Pangolin_LIBRARIES})