From 28599770ce8a63edf08a7b533813c71d60888e8e Mon Sep 17 00:00:00 2001
From: Carl Philipp Klemm <philipp@uvos.xyz>
Date: Mon, 15 May 2023 16:47:17 +0200
Subject: [PATCH] add inital torch support
---
CMakeLists.txt | 14 +++++++++++++-
1 file changed, 13 insertions(+), 1 deletion(-)
diff --git a/CMakeLists.txt b/CMakeLists.txt
index eb4f9d1..2d916f0 100644
--- a/CMakeLists.txt
+++ b/CMakeLists.txt
@@ -7,6 +7,7 @@ link_directories(${CMAKE_CURRENT_BINARY_DIR})
set (CMAKE_CXX_STANDARD 20)
find_package(Eigen3 REQUIRED)
+find_package(Torch)
function(dump_variables)
get_cmake_property(_variableNames VARIABLES)
@@ -22,10 +23,21 @@ endif(CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT)
add_library(${PROJECT_NAME} SHARED drt.cpp)
target_link_libraries(${PROJECT_NAME} ${EIGEN3_LIBRARIES} eisgenerator)
-target_include_directories(${PROJECT_NAME} PUBLIC ${TORCH_INCLUDE_DIRS} ${EIGEN3_INCLUDE_DIRS} ./LBFG)
+target_include_directories(${PROJECT_NAME} PUBLIC ${EIGEN3_INCLUDE_DIRS} ./LBFG)
set_target_properties(${PROJECT_NAME} PROPERTIES COMPILE_FLAGS "-Wall -O2 -march=native -g" LINK_FLAGS "-flto")
install(TARGETS ${PROJECT_NAME} DESTINATION lib)
+if(DEFINED TORCH_LIBRARIES)
+ message("torch support enabled")
+ add_library(${PROJECT_NAME}_torch SHARED torchdrt.cpp)
+ target_link_libraries(${PROJECT_NAME}_torch ${TORCH_LIBRARIES} ${EIGEN3_LIBRARIES} eisgenerator)
+ target_include_directories(${PROJECT_NAME}_torch PUBLIC ${TORCH_INCLUDE_DIRS} ${EIGEN3_INCLUDE_DIRS} ./LBFG)
+ set_target_properties(${PROJECT_NAME}_torch PROPERTIES COMPILE_FLAGS "-Wall -O2 -march=native -g" LINK_FLAGS "-flto")
+ install(TARGETS ${PROJECT_NAME}_torch DESTINATION lib)
+else()
+ message("torch support disabled")
+endif(DEFINED TORCH_LIBRARIES)
+
link_directories(${CMAKE_CURRENT_BINARY_DIR})
add_executable(${PROJECT_NAME}_test main.cpp)
add_dependencies(${PROJECT_NAME}_test ${PROJECT_NAME})
--
GitLab