Skip to content

Commit b21b993

Browse files
committed
Make working CMake file for torchbridge
1 parent 81fa1f2 commit b21b993

49 files changed

Lines changed: 31856 additions & 0 deletions

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

CMakeLists.txt

Lines changed: 79 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,79 @@
1+
cmake_minimum_required(VERSION 3.31 FATAL_ERROR)
2+
# project(MyProject LANGUAGES CXX)
3+
4+
include(CMakePrintHelpers)
5+
6+
set(CMAKE_VERBOSE_MAKEFILE ON)
7+
8+
if(UNIX AND NOT APPLE)
9+
set(LINUX TRUE)
10+
endif()
11+
12+
set(PROJECT_ROOT_DIR "${CMAKE_CURRENT_SOURCE_DIR}")
13+
set(PROJECT_BINARY_DIR "${CMAKE_BINARY_DIR}")
14+
15+
16+
set(BRIDGE_DIR "${PROJECT_ROOT_DIR}/bridge")
17+
18+
set(LIBTORCH_DIR "${BRIDGE_DIR}/libtorch")
19+
20+
find_package(chpl REQUIRED HINTS ${PROJECT_ROOT_DIR}/cmake/chapel)
21+
list(APPEND CMAKE_MODULE_PATH "${PROJECT_ROOT_DIR}/cmake")
22+
list(APPEND CMAKE_MODULE_PATH "${PROJECT_ROOT_DIR}/cmake/chapel")
23+
24+
project(MyProject LANGUAGES CXX C CHPL)
25+
message(STATUS "Using chpl: ${CMAKE_CHPL_COMPILER}")
26+
27+
set(CMAKE_C_COMPILER "/usr/bin/clang")
28+
set(CMAKE_CXX_COMPILER "/usr/bin/clang++")
29+
set(CMAKE_CXX_STANDARD 17)
30+
31+
32+
add_library(bridge OBJECT ${BRIDGE_DIR}/lib/bridge.cpp)
33+
34+
target_include_directories(
35+
bridge
36+
PUBLIC
37+
${BRIDGE_DIR}/include
38+
${LIBTORCH_DIR}/include
39+
${LIBTORCH_DIR}/include/torch/csrc/api/include
40+
)
41+
42+
set(BRIDGE_OBJECT_FILES $<TARGET_OBJECTS:bridge>)
43+
44+
45+
# set_target_properties(bridge PROPERTIES
46+
# OUTPUT_NAME "bridge"
47+
# )
48+
# install(TARGETS bridge DESTINATION ".")
49+
50+
51+
52+
# file(COPY ${bridge}
53+
# DESTINATION ${CMAKE_INSTALL_PREFIX}/lib/bridge.o
54+
# )
55+
56+
57+
58+
# get_target_property(bridge_o_path bridge LOCATION)
59+
# set(bridge_o_path "${CMAKE_CURRENT_BINARY_DIR}/bridge.o")
60+
# set(bridge_o_path $<TARGET_OBJECTS:bridge>)
61+
62+
# message(STATUS "Library path: ${bridge_o_path}")
63+
64+
65+
add_executable(BridgeTest
66+
${BRIDGE_DIR}/lib/Bridge.chpl
67+
${BRIDGE_DIR}/include/Bridge.h
68+
)
69+
target_link_options(BridgeTest
70+
PRIVATE
71+
${BRIDGE_DIR}/include/Bridge.h
72+
${BRIDGE_OBJECT_FILES}
73+
-L ${LIBTORCH_DIR}/lib
74+
"-ltorch"
75+
"-ltorch_cpu"
76+
"-lc10"
77+
--ldflags "-Wl,-rpath,${LIBTORCH_DIR}/lib"
78+
)
79+
# install(TARGETS BridgeTest DESTINATION ".")

0 commit comments

Comments
 (0)