-
Notifications
You must be signed in to change notification settings - Fork 20
/
CMakeLists.txt
78 lines (66 loc) · 2.75 KB
/
CMakeLists.txt
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
cmake_minimum_required(VERSION 3.13)
get_cmake_property(vars CACHE_VARIABLES)
foreach(var ${vars})
if (var MATCHES ".*_DIR$" OR var MATCHES ".*_ROOT$")
#message(STATUS "${var} = [${${var}}]")
list(APPEND CL_ARGS "-D${var}=${${var}}")
endif()
endforeach()
project(ava-spec-onnxruntime)
include(ExternalProject)
###### Compile onnxruntime/onnx_dump.c ######
find_package(PkgConfig REQUIRED)
pkg_search_module(GLIB2 REQUIRED glib-2.0)
ExternalProject_Add_Step(cava onnx_dump-nwcc
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/samples/onnxruntime/onnx_dump.c
COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/nwcc ${CMAKE_CURRENT_SOURCE_DIR}/samples/onnxruntime/onnx_dump.c -I /usr/local/cuda-10.1/include -I ${CMAKE_CURRENT_SOURCE_DIR}/headers ${GLIB2_CFLAGS}
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
DEPENDEES download
)
ExternalProject_Add_Step(cava onnx_dump-link
COMMAND mkdir -p generated &&
ln -f -s ${CMAKE_CURRENT_SOURCE_DIR}/headers ${CMAKE_CURRENT_BINARY_DIR}/headers &&
ln -f -s ${CMAKE_CURRENT_BINARY_DIR}/onnx_dump_nw/libguestlib.so
${CMAKE_CURRENT_BINARY_DIR}/onnx_dump_nw/libcuda.so.1
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/..
DEPENDEES onnx_dump-nwcc
)
ExternalProject_Add(onnx_dump_nw
PREFIX onnx_dump_nw
SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/onnx_dump_nw
BINARY_DIR ${CMAKE_CURRENT_BINARY_DIR}/onnx_dump_nw
DOWNLOAD_COMMAND ""
INSTALL_COMMAND ""
CMAKE_ARGS ${CL_ARGS}
-DCMAKE_C_COMPILER:STRING=${CMAKE_C_COMPILER}
-DCMAKE_CXX_COMPILER:STRING=${CMAKE_CXX_COMPILER}
BUILD_ALWAYS ON
DEPENDS cava
)
###### Compile onnxruntime/onnx_opt.c ######
ExternalProject_Add_Step(cava onnx_opt-nwcc
DEPENDS ${CMAKE_CURRENT_SOURCE_DIR}/samples/onnxruntime/onnx_opt.c
COMMAND ${CMAKE_CURRENT_SOURCE_DIR}/nwcc ${CMAKE_CURRENT_SOURCE_DIR}/samples/onnxruntime/onnx_opt.c -I /usr/local/cuda-10.1/include -I ${CMAKE_CURRENT_SOURCE_DIR}/headers ${GLIB2_CFLAGS}
WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}
DEPENDEES download
)
ExternalProject_Add_Step(cava onnx_opt-link
COMMAND mkdir -p generated &&
ln -f -s ${CMAKE_CURRENT_SOURCE_DIR}/headers ${CMAKE_CURRENT_BINARY_DIR}/headers &&
ln -f -s ${CMAKE_CURRENT_BINARY_DIR}/onnx_opt_nw/libguestlib.so
${CMAKE_CURRENT_BINARY_DIR}/onnx_opt_nw/libcuda.so.1
WORKING_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/..
DEPENDEES onnx_opt-nwcc
)
ExternalProject_Add(onnx_opt_nw
PREFIX onnx_opt_nw
SOURCE_DIR ${CMAKE_CURRENT_SOURCE_DIR}/onnx_opt_nw
BINARY_DIR ${CMAKE_CURRENT_BINARY_DIR}/onnx_opt_nw
DOWNLOAD_COMMAND ""
INSTALL_COMMAND ""
CMAKE_ARGS ${CL_ARGS}
-DCMAKE_C_COMPILER:STRING=${CMAKE_C_COMPILER}
-DCMAKE_CXX_COMPILER:STRING=${CMAKE_CXX_COMPILER}
BUILD_ALWAYS ON
DEPENDS cava
)