diff options
-rw-r--r-- | third_party/aws.BUILD | 75 | ||||
-rw-r--r-- | third_party/mpi/.gitignore | 3 | ||||
-rw-r--r-- | third_party/mpi_collectives/BUILD | 29 |
3 files changed, 107 insertions, 0 deletions
diff --git a/third_party/aws.BUILD b/third_party/aws.BUILD new file mode 100644 index 0000000000..858a55ee07 --- /dev/null +++ b/third_party/aws.BUILD @@ -0,0 +1,75 @@ +# Description: +# AWS C++ SDK + +package(default_visibility = ["//visibility:public"]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +load("@%ws%//third_party:common.bzl", "template_rule") + +cc_library( + name = "aws", + srcs = select({ + "@%ws%//tensorflow:linux_x86_64": glob([ + "aws-cpp-sdk-core/source/platform/linux-shared/*.cpp", + ]), + "@%ws%//tensorflow:darwin": glob([ + "aws-cpp-sdk-core/source/platform/linux-shared/*.cpp", + ]), + }) + glob([ + "aws-cpp-sdk-core/source/*.cpp", + "aws-cpp-sdk-core/source/auth/**/*.cpp", + "aws-cpp-sdk-core/source/config/**/*.cpp", + "aws-cpp-sdk-core/source/client/**/*.cpp", + "aws-cpp-sdk-core/source/external/**/*.cpp", + "aws-cpp-sdk-core/source/internal/**/*.cpp", + "aws-cpp-sdk-core/source/http/*.cpp", + "aws-cpp-sdk-core/source/http/curl/**/*.cpp", + "aws-cpp-sdk-core/source/http/standard/**/*.cpp", + "aws-cpp-sdk-core/source/utils/*.cpp", + "aws-cpp-sdk-core/source/utils/base64/**/*.cpp", + "aws-cpp-sdk-core/source/utils/json/**/*.cpp", + "aws-cpp-sdk-core/source/utils/logging/**/*.cpp", + "aws-cpp-sdk-core/source/utils/memory/**/*.cpp", + "aws-cpp-sdk-core/source/utils/stream/**/*.cpp", + "aws-cpp-sdk-core/source/utils/threading/**/*.cpp", + "aws-cpp-sdk-core/source/utils/xml/**/*.cpp", + "aws-cpp-sdk-core/source/utils/crypto/*.cpp", + "aws-cpp-sdk-core/source/utils/crypto/factory/**/*.cpp", + "aws-cpp-sdk-s3/source/**/*.cpp", + ]), + hdrs = [ + "aws-cpp-sdk-core/include/aws/core/SDKConfig.h", + ], + defines = select({ + "@%ws%//tensorflow:linux_x86_64": [ + "PLATFORM_LINUX", + "ENABLE_CURL_CLIENT", + "ENABLE_NO_ENCRYPTION", + ], + "@%ws%//tensorflow:darwin": [ + "PLATFORM_APPLE", + "ENABLE_CURL_CLIENT", + "ENABLE_NO_ENCRYPTION", + ], + "//conditions:default": [], + }), + includes = [ + "aws-cpp-sdk-core/include/", + "aws-cpp-sdk-s3/include/", + ], + deps = [ + "@curl//:curl", + ], +) + +template_rule( + name = "SDKConfig_h", + src = "aws-cpp-sdk-core/include/aws/core/SDKConfig.h.in", + out = "aws-cpp-sdk-core/include/aws/core/SDKConfig.h", + substitutions = { + "cmakedefine": "define", + }, +) diff --git a/third_party/mpi/.gitignore b/third_party/mpi/.gitignore new file mode 100644 index 0000000000..ab011617a6 --- /dev/null +++ b/third_party/mpi/.gitignore @@ -0,0 +1,3 @@ +*.h +*.dylib +*.so diff --git a/third_party/mpi_collectives/BUILD b/third_party/mpi_collectives/BUILD new file mode 100644 index 0000000000..d5f9edb0b9 --- /dev/null +++ b/third_party/mpi_collectives/BUILD @@ -0,0 +1,29 @@ +package(default_visibility = ["//visibility:public"]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE.txt"]) + +filegroup( + name = "all_files", + srcs = glob( + ["**/*"], + exclude = [ + "**/METADATA", + "**/OWNERS", + ], + ), + visibility = ["//tensorflow:__subpackages__"], +) + +cc_library( + name = "mpi", + srcs = select({ + "//tensorflow:darwin": ["libmpi.dylib"], + "//conditions:default": ["libmpi.so"], + }), + hdrs = [ + "mpi.h", + "mpi_portable_platform", + ], +) |