1"""Provides the repository macro to import TensorRT Open Source components.""" 2 3load("//third_party:repo.bzl", "tf_http_archive") 4 5def repo(name = "tensorrt_oss_archive"): 6 """Imports TensorRT Open Source Components.""" 7 TRT_OSS_COMMIT = "9ec6eb6db39188c9f3d25f49c8ee3a9721636b56" 8 TRT_OSS_SHA256 = "4fa2a712a5f2350b81df01d55c1dc17451e09efd4b2a53322b0433721009e1c7" 9 10 tf_http_archive( 11 name = name, 12 sha256 = TRT_OSS_SHA256, 13 strip_prefix = "TensorRT-{commit}".format(commit = TRT_OSS_COMMIT), 14 urls = [ 15 # TODO: Google Mirror "https://storage.googleapis.com/...." 16 "https://storage.googleapis.com/mirror.tensorflow.org/github.com/NVIDIA/TensorRT/archive/{commit}.tar.gz".format(commit = TRT_OSS_COMMIT), 17 "https://github.com/NVIDIA/TensorRT/archive/{commit}.tar.gz".format(commit = TRT_OSS_COMMIT), 18 ], 19 build_file = "//third_party/tensorrt/plugin:BUILD", 20 patch_file = ["//third_party/tensorrt/plugin:tensorrt_oss.patch"], 21 ) 22