1
- # Copyright 2020-2023 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
1
+ # Copyright 2020-2022 , NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2
2
#
3
3
# Redistribution and use in source and binary forms, with or without
4
4
# modification, are permitted provided that the following conditions
@@ -38,7 +38,6 @@ option(TRITON_ENABLE_TRACING "Include tracing support in server" OFF)
38
38
option (TRITON_ENABLE_NVTX "Include NVTX support in server" OFF )
39
39
option (TRITON_ENABLE_GPU "Enable GPU support in server" ON )
40
40
option (TRITON_ENABLE_MALI_GPU "Enable Arm Mali GPU support in server" OFF )
41
- option (TRITON_IGPU_BUILD "Enable options for iGPU compilation in sever" OFF )
42
41
set (TRITON_MIN_COMPUTE_CAPABILITY "6.0" CACHE STRING
43
42
"The minimum CUDA compute capability supported by Triton" )
44
43
set (TRITON_EXTRA_LIB_PATHS "" CACHE PATH "Extra library paths for Triton Server build" )
@@ -55,7 +54,6 @@ option(TRITON_ENABLE_VERTEX_AI "Include Vertex AI API in server" OFF)
55
54
# Metrics
56
55
option (TRITON_ENABLE_METRICS "Include metrics support in server" ON )
57
56
option (TRITON_ENABLE_METRICS_GPU "Include GPU metrics support in server" ON )
58
- option (TRITON_ENABLE_METRICS_CPU "Include CPU metrics support in server" ON )
59
57
60
58
# Cloud storage
61
59
option (TRITON_ENABLE_GCS "Include GCS Filesystem support in server" OFF )
@@ -87,10 +85,6 @@ if(TRITON_ENABLE_TRACING AND NOT TRITON_ENABLE_STATS)
87
85
message (FATAL_ERROR "TRITON_ENABLE_TRACING=ON requires TRITON_ENABLE_STATS=ON" )
88
86
endif ()
89
87
90
- if (TRITON_ENABLE_METRICS_CPU AND NOT TRITON_ENABLE_METRICS)
91
- message (FATAL_ERROR "TRITON_ENABLE_METRICS_CPU=ON requires TRITON_ENABLE_METRICS=ON" )
92
- endif ()
93
-
94
88
if (TRITON_ENABLE_METRICS_GPU AND NOT TRITON_ENABLE_METRICS)
95
89
message (FATAL_ERROR "TRITON_ENABLE_METRICS_GPU=ON requires TRITON_ENABLE_METRICS=ON" )
96
90
endif ()
@@ -119,19 +113,6 @@ FetchContent_Declare(
119
113
GIT_TAG ${TRITON_THIRD_PARTY_REPO_TAG}
120
114
)
121
115
122
- # Some libs are installed to ${TRITON_THIRD_PARTY_INSTALL_PREFIX}/{LIB}/lib64 instead
123
- # of ${TRITON_THIRD_PARTY_INSTALL_PREFIX}/{LIB}/lib on Centos
124
- set (LIB_DIR "lib" )
125
- # /etc/os-release does not exist on Windows
126
- if (EXISTS "/etc/os-release" )
127
- file (STRINGS /etc/os-release DISTRO REGEX "^NAME=" )
128
- string (REGEX REPLACE "NAME=\" (.*)\" " "\\ 1" DISTRO "${DISTRO} " )
129
- message (STATUS "Distro Name: ${DISTRO} " )
130
- if (DISTRO MATCHES "CentOS.*" )
131
- set (LIB_DIR "lib64" )
132
- endif ()
133
- endif ()
134
-
135
116
set (TRITON_CORE_HEADERS_ONLY OFF )
136
117
137
118
FetchContent_MakeAvailable(repo-third-party repo-core)
@@ -171,16 +152,7 @@ endif()
171
152
if (WIN32 )
172
153
set (_FINDPACKAGE_PROTOBUF_CONFIG_DIR "${TRITON_THIRD_PARTY_INSTALL_PREFIX} /protobuf/cmake" )
173
154
else ()
174
- set (_FINDPACKAGE_PROTOBUF_CONFIG_DIR "${TRITON_THIRD_PARTY_INSTALL_PREFIX} /protobuf/${LIB_DIR} /cmake/protobuf" )
175
- endif ()
176
-
177
- # Triton with Opentelemetry is not supported on Windows
178
- # FIXME: add location for Windows, when support is added
179
- # JIRA DLIS-4786
180
- if (WIN32 )
181
- set (_FINDPACKAGE_OPENTELEMETRY_CONFIG_DIR "" )
182
- else ()
183
- set (_FINDPACKAGE_OPENTELEMETRY_CONFIG_DIR "${TRITON_THIRD_PARTY_INSTALL_PREFIX} /opentelemetry-cpp/${LIB_DIR} /cmake/opentelemetry-cpp" )
155
+ set (_FINDPACKAGE_PROTOBUF_CONFIG_DIR "${TRITON_THIRD_PARTY_INSTALL_PREFIX} /protobuf/lib/cmake/protobuf" )
184
156
endif ()
185
157
186
158
if (CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT)
@@ -196,15 +168,15 @@ endif() # TRITON_ENABLE_GCS
196
168
if (${TRITON_ENABLE_S3} )
197
169
set (TRITON_DEPENDS ${TRITON_DEPENDS} aws-sdk-cpp)
198
170
endif () # TRITON_ENABLE_S3
171
+ if (${TRITON_ENABLE_AZURE_STORAGE} )
172
+ set (TRITON_DEPENDS ${TRITON_DEPENDS} azure-storage-cpplite)
173
+ endif () # TRITON_ENABLE_AZURE_STORAGE
199
174
if (${TRITON_ENABLE_HTTP} OR ${TRITON_ENABLE_METRICS} OR ${TRITON_ENABLE_SAGEMAKER} OR ${TRITON_ENABLE_VERTEX_AI} )
200
175
set (TRITON_DEPENDS ${TRITON_DEPENDS} libevent libevhtp)
201
176
endif () # TRITON_ENABLE_HTTP || TRITON_ENABLE_METRICS || TRITON_ENABLE_SAGEMAKER || TRITON_ENABLE_VERTEX_AI
202
177
if (${TRITON_ENABLE_GRPC} )
203
178
set (TRITON_DEPENDS ${TRITON_DEPENDS} grpc)
204
179
endif () # TRITON_ENABLE_GRPC
205
- if (NOT WIN32 AND ${TRITON_ENABLE_TRACING} )
206
- set (TRITON_DEPENDS ${TRITON_DEPENDS} opentelemetry-cpp)
207
- endif () # TRITON_ENABLE_TRACING
208
180
209
181
ExternalProject_Add(triton-server
210
182
PREFIX triton-server
@@ -217,23 +189,21 @@ ExternalProject_Add(triton-server
217
189
${_CMAKE_ARGS_VCPKG_TARGET_TRIPLET}
218
190
-DGTEST_ROOT:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /googletest
219
191
-DgRPC_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /grpc/lib/cmake/grpc
220
- -Dc-ares_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /c-ares/${LIB_DIR} /cmake/c-ares
221
- -Dabsl_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /absl/${LIB_DIR} /cmake/absl
222
- -DCURL_DIR:STRING =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /curl/${LIB_DIR} /cmake/CURL
223
- -Dnlohmann_json_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /nlohmann_json/${LIB_DIR} /cmake/nlohmann_json
192
+ -Dc-ares_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /c-ares/lib/cmake/c-ares
193
+ -Dabsl_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /absl/lib/cmake/absl
194
+ -Dnlohmann_json_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /nlohmann_json/lib/cmake/nlohmann_json
224
195
-DLibevent_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /libevent/lib/cmake/libevent
225
196
-Dlibevhtp_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /libevhtp/lib/cmake/libevhtp
226
- -Dstorage_client_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /google-cloud-cpp/${LIB_DIR} /cmake/storage_client
227
- -Dgoogle_cloud_cpp_common_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /google-cloud-cpp/${LIB_DIR} /cmake/google_cloud_cpp_common
228
- -DCrc32c_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /crc32c/${LIB_DIR} /cmake/Crc32c
229
- -DAWSSDK_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/${LIB_DIR} /cmake/AWSSDK
230
- -Daws-cpp-sdk-core_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/${LIB_DIR} /cmake/aws-cpp-sdk-core
231
- -Daws-cpp-sdk-s3_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/${LIB_DIR} /cmake/aws-cpp-sdk-s3
232
- -Daws-c-event-stream_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/${LIB_DIR} /aws-c-event-stream/cmake
233
- -Daws-c-common_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/${LIB_DIR} /aws-c-common/cmake
234
- -Daws-checksums_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/${LIB_DIR} /aws-checksums/cmake
235
- -Dopentelemetry-cpp_DIR:PATH =${_FINDPACKAGE_OPENTELEMETRY_CONFIG_DIR}
236
- -DTRITON_IGPU_BUILD:BOOL =${TRITON_IGPU_BUILD}
197
+ -Dstorage_client_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /google-cloud-cpp/lib/cmake/storage_client
198
+ -Dazure-storage-cpplite_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /azure-storage-cpplite
199
+ -Dgoogle_cloud_cpp_common_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /google-cloud-cpp/lib/cmake/google_cloud_cpp_common
200
+ -DCrc32c_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /crc32c/lib/cmake/Crc32c
201
+ -DAWSSDK_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/lib/cmake/AWSSDK
202
+ -Daws-cpp-sdk-core_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/lib/cmake/aws-cpp-sdk-core
203
+ -Daws-cpp-sdk-s3_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/lib/cmake/aws-cpp-sdk-s3
204
+ -Daws-c-event-stream_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/lib/aws-c-event-stream/cmake
205
+ -Daws-c-common_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/lib/aws-c-common/cmake
206
+ -Daws-checksums_DIR:PATH =${TRITON_THIRD_PARTY_INSTALL_PREFIX} /aws-sdk-cpp/lib/aws-checksums/cmake
237
207
-DTRITON_THIRD_PARTY_REPO_TAG:STRING =${TRITON_THIRD_PARTY_REPO_TAG}
238
208
-DTRITON_COMMON_REPO_TAG:STRING =${TRITON_COMMON_REPO_TAG}
239
209
-DTRITON_CORE_REPO_TAG:STRING =${TRITON_CORE_REPO_TAG}
@@ -253,7 +223,6 @@ ExternalProject_Add(triton-server
253
223
-DTRITON_MIN_COMPUTE_CAPABILITY:STRING =${TRITON_MIN_COMPUTE_CAPABILITY}
254
224
-DTRITON_ENABLE_METRICS:BOOL =${TRITON_ENABLE_METRICS}
255
225
-DTRITON_ENABLE_METRICS_GPU:BOOL =${TRITON_ENABLE_METRICS_GPU}
256
- -DTRITON_ENABLE_METRICS_CPU:BOOL =${TRITON_ENABLE_METRICS_CPU}
257
226
-DTRITON_ENABLE_GCS:BOOL =${TRITON_ENABLE_GCS}
258
227
-DTRITON_ENABLE_AZURE_STORAGE:BOOL =${TRITON_ENABLE_AZURE_STORAGE}
259
228
-DTRITON_ENABLE_S3:BOOL =${TRITON_ENABLE_S3}
0 commit comments