Book a Demo!
CoCalc Logo Icon
StoreFeaturesDocsShareSupportNewsAboutPoliciesSign UpSign In
pytorch
GitHub Repository: pytorch/tutorials
Path: blob/main/advanced_source/custom_classes/CMakeLists.txt
1751 views
1
cmake_minimum_required(VERSION 3.1 FATAL_ERROR)
2
project(infer)
3
4
find_package(Torch REQUIRED)
5
6
add_subdirectory(custom_class_project)
7
8
# Define our library target
9
add_executable(infer infer.cpp)
10
set(CMAKE_CXX_STANDARD 14)
11
# Link against LibTorch
12
target_link_libraries(infer "${TORCH_LIBRARIES}")
13
# This is where we link in our libcustom_class code, making our
14
# custom class available in our binary.
15
target_link_libraries(infer -Wl,--no-as-needed custom_class)
16
17