early-access version 2853
This commit is contained in:
53
externals/vcpkg/ports/tensorpipe/portfile.cmake
vendored
Executable file
53
externals/vcpkg/ports/tensorpipe/portfile.cmake
vendored
Executable file
@@ -0,0 +1,53 @@
|
||||
vcpkg_check_linkage(ONLY_STATIC_LIBRARY)
|
||||
|
||||
vcpkg_download_distfile(INSTALL_PACKAGE_CONFIG_PATCH
|
||||
URLS "https://patch-diff.githubusercontent.com/raw/pytorch/tensorpipe/pull/435.diff"
|
||||
FILENAME tensorpipe-pr-435.patch
|
||||
SHA512 7bcf604a967da36b8af936f8b8ab87b442834024b0b2cb886811c15e80893be842fbee2667bbbc39886814ec9b2f4ed0c2527de51fdb7dc293b25cce515f5e4b
|
||||
)
|
||||
|
||||
vcpkg_from_github(
|
||||
OUT_SOURCE_PATH SOURCE_PATH
|
||||
REPO pytorch/tensorpipe
|
||||
REF 52791a2fd214b2a9dc5759d36725909c1daa7f2e
|
||||
SHA512 1e5faf17a7236c5506c08cb28be16069b11bb929bbca64ed9745ce4277d46739186ab7d6597da7437d90ed2d166d4c37ef2f3bceabe8083ef3adbb0e8e5f227e
|
||||
PATCHES
|
||||
"${INSTALL_PACKAGE_CONFIG_PATCH}"
|
||||
use-vcpkg.patch
|
||||
support-test.patch
|
||||
)
|
||||
|
||||
vcpkg_check_features(OUT_FEATURE_OPTIONS FEATURE_OPTIONS
|
||||
FEATURES
|
||||
cuda TP_USE_CUDA
|
||||
cuda TP_ENABLE_CUDA_IPC
|
||||
pybind11 TP_BUILD_PYTHON
|
||||
test TP_BUILD_TESTING
|
||||
)
|
||||
|
||||
if("pybind11" IN_LIST FEATURES)
|
||||
vcpkg_find_acquire_program(PYTHON3)
|
||||
list(APPEND FEATURE_OPTIONS -DPYTHON_EXECUTABLE=${PYTHON3})
|
||||
endif()
|
||||
|
||||
vcpkg_cmake_configure(
|
||||
SOURCE_PATH "${SOURCE_PATH}"
|
||||
OPTIONS
|
||||
${FEATURE_OPTIONS}
|
||||
-DTP_ENABLE_SHM=${VCPKG_TARGET_IS_LINUX}
|
||||
-DTP_ENABLE_IBV=OFF
|
||||
-DTP_ENABLE_CMA=OFF
|
||||
-DTP_BUILD_LIBUV=OFF # will use libuv package
|
||||
-DTP_ENABLE_CUDA_GDR=OFF
|
||||
MAYBE_UNUSED_VARIABLES
|
||||
TP_ENABLE_CUDA_GDR
|
||||
TP_ENABLE_CUDA_IPC
|
||||
)
|
||||
vcpkg_cmake_install()
|
||||
vcpkg_copy_pdbs()
|
||||
vcpkg_cmake_config_fixup(CONFIG_PATH "share/cmake/Tensorpipe")
|
||||
|
||||
file(INSTALL "${SOURCE_PATH}/LICENSE.txt" DESTINATION "${CURRENT_PACKAGES_DIR}/share/${PORT}" RENAME copyright)
|
||||
file(REMOVE_RECURSE "${CURRENT_PACKAGES_DIR}/debug/include"
|
||||
"${CURRENT_PACKAGES_DIR}/debug/share"
|
||||
)
|
||||
39
externals/vcpkg/ports/tensorpipe/support-test.patch
vendored
Executable file
39
externals/vcpkg/ports/tensorpipe/support-test.patch
vendored
Executable file
@@ -0,0 +1,39 @@
|
||||
diff --git a/tensorpipe/test/CMakeLists.txt b/tensorpipe/test/CMakeLists.txt
|
||||
index 53e8643..c49ada6 100644
|
||||
--- a/tensorpipe/test/CMakeLists.txt
|
||||
+++ b/tensorpipe/test/CMakeLists.txt
|
||||
@@ -93,25 +93,24 @@ if(TP_USE_CUDA)
|
||||
channel/cuda_ipc/cuda_ipc_test.cc
|
||||
)
|
||||
endif()
|
||||
-
|
||||
- list(APPEND TP_TEST_SRCS
|
||||
- channel/cuda_gdr/cuda_gdr_test.cc
|
||||
- )
|
||||
-
|
||||
+ if(TP_ENABLE_CUDA_GDR)
|
||||
+ list(APPEND TP_TEST_SRCS
|
||||
+ channel/cuda_gdr/cuda_gdr_test.cc
|
||||
+ )
|
||||
+ endif()
|
||||
cuda_add_library(tensorpipe_cuda_kernel channel/kernel.cu)
|
||||
list(APPEND TP_TEST_LINK_LIBRARIES tensorpipe_cuda_kernel)
|
||||
|
||||
list(APPEND TP_TEST_LINK_LIBRARIES tensorpipe_cuda)
|
||||
endif()
|
||||
|
||||
-add_subdirectory(${PROJECT_SOURCE_DIR}/third_party/googletest
|
||||
- ${PROJECT_BINARY_DIR}/third_party/googletest EXCLUDE_FROM_ALL)
|
||||
+find_package(GTest CONFIG REQUIRED)
|
||||
|
||||
list(APPEND TP_TEST_LINK_LIBRARIES
|
||||
tensorpipe
|
||||
- uv::uv
|
||||
- gmock
|
||||
- gtest_main)
|
||||
+ unofficial::libuv::libuv
|
||||
+ GTest::gmock
|
||||
+ GTest::gtest_main)
|
||||
|
||||
add_executable(tensorpipe_test ${TP_TEST_SRCS})
|
||||
|
||||
48
externals/vcpkg/ports/tensorpipe/use-vcpkg.patch
vendored
Executable file
48
externals/vcpkg/ports/tensorpipe/use-vcpkg.patch
vendored
Executable file
@@ -0,0 +1,48 @@
|
||||
diff --git a/tensorpipe/CMakeLists.txt b/tensorpipe/CMakeLists.txt
|
||||
index efcffc2..e167fae 100644
|
||||
--- a/tensorpipe/CMakeLists.txt
|
||||
+++ b/tensorpipe/CMakeLists.txt
|
||||
@@ -119,8 +119,8 @@ list(APPEND TP_PUBLIC_HDRS
|
||||
transport/uv/utility.h)
|
||||
|
||||
# Add uv package
|
||||
-find_package(uv REQUIRED)
|
||||
-list(APPEND TP_LINK_LIBRARIES uv::uv)
|
||||
+find_package(unofficial-libuv CONFIG REQUIRED)
|
||||
+list(APPEND TP_LINK_LIBRARIES unofficial::libuv::libuv)
|
||||
|
||||
### shm
|
||||
|
||||
@@ -184,7 +184,8 @@ configure_file(config.h.in config.h)
|
||||
# We should keep libnop headers private as they should not be exposed to downstream users,
|
||||
# but they're currently transitively included by tensorpipe/transport/connection.h (which
|
||||
# is still unclear whether it should be a public or private header).
|
||||
-list(APPEND TP_INCLUDE_DIRS $<BUILD_INTERFACE:${PROJECT_SOURCE_DIR}/third_party/libnop/include>)
|
||||
+find_path(LIBNOP_INCLUDE_DIRS "nop/serializer.h" REQUIRED)
|
||||
+list(APPEND TP_INCLUDE_DIRS $<BUILD_INTERFACE:${LIBNOP_INCLUDE_DIRS}>)
|
||||
|
||||
|
||||
## Target
|
||||
diff --git a/tensorpipe/python/CMakeLists.txt b/tensorpipe/python/CMakeLists.txt
|
||||
index b722381..688886a 100644
|
||||
--- a/tensorpipe/python/CMakeLists.txt
|
||||
+++ b/tensorpipe/python/CMakeLists.txt
|
||||
@@ -4,16 +4,11 @@
|
||||
# This source code is licensed under the BSD-style license found in the
|
||||
# LICENSE file in the root directory of this source tree.
|
||||
|
||||
-if(NOT (COMMAND pybind11_add_module))
|
||||
- add_subdirectory(
|
||||
- ${PROJECT_SOURCE_DIR}/third_party/pybind11
|
||||
- ${PROJECT_BINARY_DIR}/third_party/pybind11
|
||||
- EXCLUDE_FROM_ALL)
|
||||
-endif()
|
||||
+find_package(pybind11 CONFIG REQUIRED)
|
||||
|
||||
set(PYBIND11_CPP_STANDARD -std=c++14)
|
||||
pybind11_add_module(pytensorpipe tensorpipe.cc)
|
||||
-target_link_libraries(pytensorpipe PRIVATE tensorpipe)
|
||||
+target_link_libraries(pytensorpipe PRIVATE tensorpipe pybind11::module)
|
||||
|
||||
install(TARGETS pytensorpipe
|
||||
EXPORT TensorpipeTargets
|
||||
40
externals/vcpkg/ports/tensorpipe/vcpkg.json
vendored
Executable file
40
externals/vcpkg/ports/tensorpipe/vcpkg.json
vendored
Executable file
@@ -0,0 +1,40 @@
|
||||
{
|
||||
"name": "tensorpipe",
|
||||
"version-date": "2022-03-16",
|
||||
"description": "A tensor-aware point-to-point communication primitive for machine learning",
|
||||
"homepage": "https://github.com/pytorch/tensorpipe",
|
||||
"license": "BSD-3-Clause",
|
||||
"supports": "linux | osx",
|
||||
"dependencies": [
|
||||
"libnop",
|
||||
"libuv",
|
||||
{
|
||||
"name": "vcpkg-cmake",
|
||||
"host": true
|
||||
},
|
||||
{
|
||||
"name": "vcpkg-cmake-config",
|
||||
"host": true
|
||||
}
|
||||
],
|
||||
"features": {
|
||||
"cuda": {
|
||||
"description": "Enable support for CUDA tensors, CUDA IPC channel",
|
||||
"dependencies": [
|
||||
"cuda"
|
||||
]
|
||||
},
|
||||
"pybind11": {
|
||||
"description": "Build Python bindings",
|
||||
"dependencies": [
|
||||
"pybind11"
|
||||
]
|
||||
},
|
||||
"test": {
|
||||
"description": "Build with Google.Test",
|
||||
"dependencies": [
|
||||
"gtest"
|
||||
]
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user