diff --git a/install.sh b/install.sh index c700733a..8f075744 100755 --- a/install.sh +++ b/install.sh @@ -17,5 +17,11 @@ print_finish() { echo "| deepflame (linked with libcantera) compiled successfully! Enjoy!! |" echo " = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = = =" } - +if [ $USE_LIBTORCH = true ]; then + cd "$DF_SRC/dfChemistryModel/DNNInferencer" + mkdir build + cd build + cmake .. + make install +fi ./Allwmake -j && print_finish diff --git a/src/dfChemistryModel/DNNInferencer/CMakeLists.txt b/src/dfChemistryModel/DNNInferencer/CMakeLists.txt index d81d3c5c..9e9f5a13 100644 --- a/src/dfChemistryModel/DNNInferencer/CMakeLists.txt +++ b/src/dfChemistryModel/DNNInferencer/CMakeLists.txt @@ -1,6 +1,6 @@ cmake_minimum_required(VERSION 3.0) -set(CMAKE_PREFIX_PATH "~/libtorch") +set(CMAKE_PREFIX_PATH $ENV{LIBTORCH_ROOT}) project(DNNInferencer) find_package(Torch REQUIRED) set(CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} ${TORCH_CXX_FLAGS}") @@ -8,3 +8,8 @@ add_library(DNNInferencer SHARED DNNInferencer.cpp) target_compile_features(DNNInferencer PUBLIC cxx_range_for) target_link_libraries(DNNInferencer ${TORCH_LIBRARIES}) set_property(TARGET DNNInferencer PROPERTY CXX_STANDARD 14) + +set(CMAKE_INSTALL_PREFIX $ENV{DF_ROOT}) +install (TARGETS ${PROJECT_NAME} + LIBRARY DESTINATION lib + )