diff --git a/WORKSPACE b/WORKSPACE index 02f5e925a..bd30addd8 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -31,10 +31,20 @@ http_archive( ) http_archive( name = "bazel_skylib", - sha256 = "2c62d8cd4ab1e65c08647eb4afe38f51591f43f7f0885e7769832fa137633dcb", - strip_prefix = "bazel-skylib-0.7.0", - urls = ["https://github.com/bazelbuild/bazel-skylib/archive/0.7.0.tar.gz"], + sha256 = "2ef429f5d7ce7111263289644d233707dba35e39696377ebab8b0bc701f7818e", + urls = ["https://github.com/bazelbuild/bazel-skylib/releases/download/0.8.0/bazel-skylib.0.8.0.tar.gz"], ) + +http_archive( + name = "com_github_nelhage_rules_boost", + sha256 = "f8c9653c1c49489c04f9f87ab1ee93d7b59bb26a39d9e30e9687fca3c6197c3f", + strip_prefix = "rules_boost-9f9fb8b2f0213989247c9d5c0e814a8451d18d7f", + urls = ["https://github.com/nelhage/rules_boost/archive/9f9fb8b2f0213989247c9d5c0e814a8451d18d7f.tar.gz"], +) + +load("@com_github_nelhage_rules_boost//:boost/boost.bzl", "boost_deps") +boost_deps() + # END: Upstream TensorFlow dependencies # Please add all new TensorFlow Serving dependencies in workspace.bzl. diff --git a/tensorflow_serving/workspace.bzl b/tensorflow_serving/workspace.bzl index c24c53d61..c97858a21 100644 --- a/tensorflow_serving/workspace.bzl +++ b/tensorflow_serving/workspace.bzl @@ -56,3 +56,59 @@ def tf_serving_workspace(): "https://github.com/abseil/abseil-cpp/archive/36d37ab992038f52276ca66b9da80c1cf0f57dc2.tar.gz", ], ) + + http_archive( + name = "aws_c_common", + build_file = str(Label("//third_party/aws_util:aws_c_common.BUILD")), + sha256 = "e9462a141b5db30006704f537d19b92357a59be38d590272e6118976b0356ccd", + strip_prefix = "aws-c-common-0.7.4", + urls = [ + "https://storage.googleapis.com/mirror.tensorflow.org/github.com/awslabs/aws-c-common/archive/refs/tags/v0.7.4.tar.gz", + "https://github.com/awslabs/aws-c-common/archive/refs/tags/v0.7.4.tar.gz", + ], + ) + + http_archive( + name = "aws_c_io", + build_file = str(Label("//third_party/aws_util:aws_c_io.BUILD")), + sha256 = "b60270d23b6e2f4a5d80e64ca6538ba114cd6044b53752964c940f87e59bf0d9", + strip_prefix = "aws-c-io-0.11.2", + urls = [ + "https://storage.googleapis.com/mirror.tensorflow.org/github.com/awslabs/aws-c-io/archive/refs/tags/v0.11.2.tar.gz", + "https://github.com/awslabs/aws-c-io/archive/refs/tags/v0.11.2.tar.gz", + ], + ) + + http_archive( + name = "aws_c_event_stream", + build_file = str(Label("//third_party/aws_util:aws_c_event_stream.BUILD")), + sha256 = "bae0c762b6a4b779a0db0f4730512da6cb500e76681ffdcb9f7286d8e26e547a", + strip_prefix = "aws-c-event-stream-0.2.6", + urls = [ + "https://storage.googleapis.com/mirror.tensorflow.org/github.com/awslabs/aws-c-event-stream/archive/refs/tags/v0.2.6.tar.gz", + "https://github.com/awslabs/aws-c-event-stream/archive/refs/tags/v0.2.6.tar.gz", + ], + ) + + http_archive( + name = "aws_checksums", + build_file = str(Label("//third_party/aws_util:aws_checksums.BUILD")), + sha256 = "394723034b81cc7cd528401775bc7aca2b12c7471c92350c80a0e2fb9d2909fe", + strip_prefix = "aws-checksums-0.1.12", + urls = [ + "https://storage.googleapis.com/mirror.tensorflow.org/github.com/awslabs/aws-checksums/archive/refs/tags/v0.1.12.tar.gz", + "https://github.com/awslabs/aws-checksums/archive/refs/tags/v0.1.12.tar.gz", + ], + ) + + http_archive( + name = "aws_c_cal", + build_file = str(Label("//third_party/aws_util:aws_c_cal.BUILD")), + sha256 = "40297da04443d4ee2988d1c5fb0dc4a156d0e4cfaf80e6a1df1867452566d540", + strip_prefix = "aws-c-cal-0.5.17", + urls = [ + "https://storage.googleapis.com/mirror.tensorflow.org/github.com/awslabs/aws-c-cal/archive/refs/tags/v0.5.17.tar.gz", + "https://github.com/awslabs/aws-c-cal/archive/refs/tags/v0.5.17.tar.gz", + ], + ) + diff --git a/third_party/aws_util/BUILD b/third_party/aws_util/BUILD new file mode 100644 index 000000000..2f5d02bec --- /dev/null +++ b/third_party/aws_util/BUILD @@ -0,0 +1 @@ +# Dummy BUILD file to make this directory a package. diff --git a/third_party/aws_util/aws_c_cal.BUILD b/third_party/aws_util/aws_c_cal.BUILD new file mode 100644 index 000000000..ddb9ea6a7 --- /dev/null +++ b/third_party/aws_util/aws_c_cal.BUILD @@ -0,0 +1,23 @@ +# Description: +# AWS C CAL + +package(default_visibility = ["//visibility:public"]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +cc_library( + name = "aws_c_cal", + srcs = glob([ + "include/aws/cal/**/*.h", + "source/*.c", + "source/unix/*.c", + ]), + includes = [ + "include/", + ], + deps = [ + "@aws_c_common", + ], +) diff --git a/third_party/aws_util/aws_c_common.BUILD b/third_party/aws_util/aws_c_common.BUILD new file mode 100644 index 000000000..dc28f3a63 --- /dev/null +++ b/third_party/aws_util/aws_c_common.BUILD @@ -0,0 +1,40 @@ +# Description: +# AWS C COMMON + +package(default_visibility = ["//visibility:public"]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +load("@org_tensorflow//third_party:common.bzl", "template_rule") + +cc_library( + name = "aws_c_common", + srcs = glob([ + "include/aws/common/**/*.h", + "include/aws/common/**/*.inl", + "source/*.c", + "source/posix/*.c", + "source/arch/intel/cpuid.c", + "source/arch/intel/asm/cpuid.c", + ]), + hdrs = [ + "include/aws/common/config.h", + ], + includes = [ + "include/", + ], + defines = [ + "AWS_AFFINITY_METHOD=AWS_AFFINITY_METHOD_PTHREAD_ATTR", + ], +) + +template_rule( + name = "COMMONConfig_h", + src = "include/aws/common/config.h.in", + out = "include/aws/common/config.h", + substitutions = { + "cmakedefine": "define", + }, +) diff --git a/third_party/aws_util/aws_c_event_stream.BUILD b/third_party/aws_util/aws_c_event_stream.BUILD new file mode 100644 index 000000000..49fb10619 --- /dev/null +++ b/third_party/aws_util/aws_c_event_stream.BUILD @@ -0,0 +1,24 @@ +# Description: +# AWS C EVENT STREAM + +package(default_visibility = ["//visibility:public"]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +cc_library( + name = "aws_c_event_stream", + srcs = glob([ + "include/aws/event-stream/*.h", + "source/event_stream.c", + ]), + includes = [ + "include/", + ], + deps = [ + "@aws_c_common", + "@aws_checksums", + "@aws_c_io", + ], +) diff --git a/third_party/aws_util/aws_c_io.BUILD b/third_party/aws_util/aws_c_io.BUILD new file mode 100644 index 000000000..88f82ad80 --- /dev/null +++ b/third_party/aws_util/aws_c_io.BUILD @@ -0,0 +1,31 @@ +# Description: +# AWS C IO + +package(default_visibility = ["//visibility:public"]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +cc_library( + name = "aws_c_io", + srcs = glob([ + "include/aws/io/**/*.h", + "source/*.c", + "source/pkcs11/v2.40/*.h", + "source/pkcs11_private.h", + "source/posix/*.c", + "source/linux/*.c", + ]), + includes = [ + "include/", + "source/", + ], + deps = [ + "@aws_c_common", + "@aws_c_cal" + ], + defines = [ + "BYO_CRYPTO", + ], +) diff --git a/third_party/aws_util/aws_checksums.BUILD b/third_party/aws_util/aws_checksums.BUILD new file mode 100644 index 000000000..b332a3c5a --- /dev/null +++ b/third_party/aws_util/aws_checksums.BUILD @@ -0,0 +1,23 @@ +# Description: +# AWS CHECKSUMS + +package(default_visibility = ["//visibility:public"]) + +licenses(["notice"]) # Apache 2.0 + +exports_files(["LICENSE"]) + +cc_library( + name = "aws_checksums", + srcs = glob([ + "include/aws/checksums/**/*.h", + "source/*.c", + "source/intel/asm/*.c", + ]), + includes = [ + "include/", + ], + deps = [ + "@aws_c_common", + ], +)