Created
January 19, 2024 00:17
-
-
Save Davidnet/dcff167d52f1433398832f3297f9ca91 to your computer and use it in GitHub Desktop.
ha.txt
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
cmake_minimum_required(VERSION 3.17) | |
project(tritonclient LANGUAGES C CXX) | |
# Use C++17 standard as Triton's minimum required. | |
set(TRITON_MIN_CXX_STANDARD 17 CACHE STRING "The minimum C++ standard which features are requested to build this target.") | |
# | |
# Options | |
# | |
set(TRITON_VERSION "0.0.0" CACHE STRING "Version for the clients") | |
set(PERF_ANALYZER_VERSION ${TRITON_VERSION} CACHE STRING "Build Version for Perf Analyzer") | |
option(TRITON_ENABLE_CC_HTTP "Build C++ HTTP client libraries" OFF) | |
option(TRITON_ENABLE_CC_GRPC "Build C++ GRPC client libraries" OFF) | |
option(TRITON_ENABLE_PYTHON_HTTP "Enable Python HTTP client libraries" OFF) | |
option(TRITON_ENABLE_PYTHON_GRPC "Enable Python GRPC client libraries" OFF) | |
option(TRITON_ENABLE_JAVA_HTTP "Enable JAVA HTTP client libraries" OFF) | |
option(TRITON_ENABLE_PERF_ANALYZER "Enable Performance Analyzer" OFF) | |
option(TRITON_ENABLE_PERF_ANALYZER_C_API "Enable Performance Analyzer C API" OFF) | |
option(TRITON_ENABLE_PERF_ANALYZER_TFS "Enable TensorFlow Serving support for Performance Analyzer" OFF) | |
option(TRITON_ENABLE_PERF_ANALYZER_TS "Enable TorchServe support for Performance Analyzer" OFF) | |
option(TRITON_ENABLE_EXAMPLES "Include examples in build" OFF) | |
option(TRITON_ENABLE_TESTS "Include tests in build" OFF) | |
option(TRITON_ENABLE_GPU "Enable GPU support in libraries" OFF) | |
option(TRITON_ENABLE_ZLIB "Include ZLIB library in build" ON) | |
set(TRITON_COMMON_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/common repo") | |
set(TRITON_THIRD_PARTY_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/third_party repo") | |
set(TRITON_CORE_REPO_TAG "main" CACHE STRING "Tag for triton-inference-server/core repo") | |
if(NOT CMAKE_BUILD_TYPE) | |
set(CMAKE_BUILD_TYPE Release) | |
endif() | |
if(NOT TRITON_ENABLE_PYTHON_GRPC) | |
set(TRITON_COMMON_ENABLE_PROTOBUF_PYTHON OFF) | |
endif() | |
# | |
# Dependencies | |
# | |
include(FetchContent) | |
FetchContent_Declare( | |
repo-third-party | |
GIT_REPOSITORY https://github.com/triton-inference-server/third_party.git | |
GIT_TAG ${TRITON_THIRD_PARTY_REPO_TAG} | |
GIT_SHALLOW ON | |
) | |
set(TRITON_THIRD_PARTY_INSTALL_PREFIX ${CMAKE_CURRENT_BINARY_DIR}/third-party) | |
FetchContent_MakeAvailable(repo-third-party) | |
# Some libs are installed to ${TRITON_THIRD_PARTY_INSTALL_PREFIX}/{LIB}/lib64 instead | |
# of ${TRITON_THIRD_PARTY_INSTALL_PREFIX}/{LIB}/lib on Centos | |
set (LIB_DIR "lib") | |
# Need to use ExternalProject for our builds so that we can get the | |
# correct dependencies between our components and the ExternalProject | |
# dependencies (found in the third_party repo) | |
include(ExternalProject) | |
# if (CMAKE_INSTALL_PREFIX_INITIALIZED_TO_DEFAULT) | |
# set(TRITON_INSTALL_PREFIX ${CMAKE_CURRENT_BINARY_DIR}/cc-clients/install) | |
# else() | |
# set(TRITON_INSTALL_PREFIX ${CMAKE_INSTALL_PREFIX}) | |
# endif() | |
set(TRITON_INSTALL_PREFIX ${CMAKE_INSTALL_PREFIX}) | |
set(_CMAKE_ARGS_OPENSSL_ROOT_DIR "") | |
set(_CMAKE_ARGS_CMAKE_TOOLCHAIN_FILE "") | |
set(_CMAKE_ARGS_VCPKG_TARGET_TRIPLET "") | |
set(_FINDPACKAGE_PROTOBUF_CONFIG_DIR "${TRITON_THIRD_PARTY_INSTALL_PREFIX}/protobuf/${LIB_DIR}/cmake/protobuf") | |
set(_cc_client_depends re2) | |
set(_cc_client_depends ${_cc_client_depends} grpc protobuf) | |
ExternalProject_Add(cc-clients | |
PREFIX cc-clients | |
SOURCE_DIR "${CMAKE_CURRENT_SOURCE_DIR}/src/c++" | |
BINARY_DIR "${CMAKE_CURRENT_BINARY_DIR}/cc-clients" | |
CMAKE_CACHE_ARGS | |
${_CMAKE_ARGS_OPENSSL_ROOT_DIR} | |
${_CMAKE_ARGS_CMAKE_TOOLCHAIN_FILE} | |
${_CMAKE_ARGS_VCPKG_TARGET_TRIPLET} | |
-DCURL_DIR:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/curl/${LIB_DIR}/cmake/CURL | |
-DProtobuf_DIR:PATH=${_FINDPACKAGE_PROTOBUF_CONFIG_DIR} | |
-DgRPC_DIR:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/grpc/lib/cmake/grpc | |
-Dabsl_DIR:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/absl/${LIB_DIR}/cmake/absl | |
-Dre2_DIR:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/re2/${LIB_DIR}/cmake/re2 | |
-Dc-ares_DIR:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/c-ares/${LIB_DIR}/cmake/c-ares | |
-DGTEST_ROOT:PATH=${TRITON_THIRD_PARTY_INSTALL_PREFIX}/googletest | |
-DTRITON_COMMON_REPO_TAG:STRING=${TRITON_COMMON_REPO_TAG} | |
-DTRITON_CORE_REPO_TAG:STRING=${TRITON_CORE_REPO_TAG} | |
-DPERF_ANALYZER_VERSION:STRING=${PERF_ANALYZER_VERSION} | |
-DTRITON_ENABLE_CC_HTTP:BOOL=${TRITON_ENABLE_CC_HTTP} | |
-DTRITON_ENABLE_CC_GRPC:BOOL=${TRITON_ENABLE_CC_GRPC} | |
-DTRITON_ENABLE_PERF_ANALYZER:BOOL=${TRITON_ENABLE_PERF_ANALYZER} | |
-DTRITON_ENABLE_PERF_ANALYZER_C_API:BOOL=${TRITON_ENABLE_PERF_ANALYZER_C_API} | |
-DTRITON_ENABLE_PERF_ANALYZER_TFS:BOOL=${TRITON_ENABLE_PERF_ANALYZER_TFS} | |
-DTRITON_ENABLE_PERF_ANALYZER_TS:BOOL=${TRITON_ENABLE_PERF_ANALYZER_TS} | |
-DTRITON_ENABLE_EXAMPLES:BOOL=${TRITON_ENABLE_EXAMPLES} | |
-DTRITON_ENABLE_TESTS:BOOL=${TRITON_ENABLE_TESTS} | |
-DTRITON_ENABLE_GPU:BOOL=${TRITON_ENABLE_GPU} | |
-DTRITON_ENABLE_ZLIB:BOOL=${TRITON_ENABLE_ZLIB} | |
-DTRITON_MIN_CXX_STANDARD:STRING=${TRITON_MIN_CXX_STANDARD} | |
-DCMAKE_BUILD_TYPE:STRING=${CMAKE_BUILD_TYPE} | |
-DCMAKE_INSTALL_PREFIX:PATH=${TRITON_INSTALL_PREFIX} | |
DEPENDS ${_cc_client_depends} | |
) |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment