Skip to content
Snippets Groups Projects
Commit 28599770 authored by Carl Philipp Klemm's avatar Carl Philipp Klemm
Browse files

add inital torch support

parent 8bb2ebd7
No related branches found
No related tags found
No related merge requests found
...@@ -7,6 +7,7 @@ link_directories(${CMAKE_CURRENT_BINARY_DIR}) ...@@ -7,6 +7,7 @@ link_directories(${CMAKE_CURRENT_BINARY_DIR})
set (CMAKE_CXX_STANDARD 20) set (CMAKE_CXX_STANDARD 20)
find_package(Eigen3 REQUIRED) find_package(Eigen3 REQUIRED)
find_package(Torch)
function(dump_variables) function(dump_variables)
get_cmake_property(_variableNames VARIABLES) get_cmake_property(_variableNames VARIABLES)
...@@ -22,10 +23,21 @@ endif(CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT) ...@@ -22,10 +23,21 @@ endif(CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT)
add_library(${PROJECT_NAME} SHARED drt.cpp) add_library(${PROJECT_NAME} SHARED drt.cpp)
target_link_libraries(${PROJECT_NAME} ${EIGEN3_LIBRARIES} eisgenerator) 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") set_target_properties(${PROJECT_NAME} PROPERTIES COMPILE_FLAGS "-Wall -O2 -march=native -g" LINK_FLAGS "-flto")
install(TARGETS ${PROJECT_NAME} DESTINATION lib) 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}) link_directories(${CMAKE_CURRENT_BINARY_DIR})
add_executable(${PROJECT_NAME}_test main.cpp) add_executable(${PROJECT_NAME}_test main.cpp)
add_dependencies(${PROJECT_NAME}_test ${PROJECT_NAME}) add_dependencies(${PROJECT_NAME}_test ${PROJECT_NAME})
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment