use Torch variables for libraries and include paths

christian-rauch · Dec 27, 2023 · 1022b5b · 1022b5b
1 parent 6ce8a72
commit 1022b5b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -11,7 +11,8 @@ find_package(Torch REQUIRED)
 
 add_library(${PROJECT_NAME} SHARED src/super_point_inference.cpp)
 target_link_libraries(${PROJECT_NAME} PUBLIC ${OpenCV_LIBRARIES})
-target_link_libraries(${PROJECT_NAME} PRIVATE torch "stdc++fs")
+target_link_libraries(${PROJECT_NAME} PRIVATE ${TORCH_LIBRARIES} "stdc++fs")
+target_include_directories(${PROJECT_NAME} PRIVATE ${TORCH_INCLUDE_DIRS})
 
 target_link_options(${PROJECT_NAME} PRIVATE ${TORCH_CXX_FLAGS})