|
1 | 1 | # Platform-specific build configurations. |
2 | 2 |
|
3 | | -load("//google/protobuf:protobuf.bzl", "cc_proto_library") |
4 | | -load("//google/protobuf:protobuf.bzl", "py_proto_library") |
| 3 | +load("@protobuf//:protobuf.bzl", "cc_proto_library") |
| 4 | +load("@protobuf//:protobuf.bzl", "py_proto_library") |
5 | 5 |
|
6 | 6 | # configure may change the following line to True |
7 | 7 | WITH_GCP_SUPPORT = False |
@@ -31,49 +31,63 @@ def tf_proto_library_cc(name, srcs = [], has_services = None, |
31 | 31 | cc_api_version = 2, go_api_version = 2, |
32 | 32 | java_api_version = 2, |
33 | 33 | py_api_version = 2): |
34 | | - native.filegroup(name=name + "_proto_srcs", |
35 | | - srcs=srcs + tf_deps(deps, "_proto_srcs"), |
36 | | - testonly=testonly,) |
| 34 | + native.filegroup( |
| 35 | + name = name + "_proto_srcs", |
| 36 | + srcs = srcs + tf_deps(deps, "_proto_srcs"), |
| 37 | + testonly = testonly, |
| 38 | + ) |
37 | 39 |
|
38 | 40 | use_grpc_plugin = None |
39 | 41 | if cc_grpc_version: |
40 | 42 | use_grpc_plugin = True |
41 | | - cc_proto_library(name=name + "_cc", |
42 | | - srcs=srcs + tf_deps(deps, "_proto_srcs"), |
43 | | - deps=deps + ["//google/protobuf:cc_wkt_protos"], |
44 | | - cc_libs = cc_libs + ["//google/protobuf:protobuf"], |
45 | | - use_grpc_plugin = use_grpc_plugin, |
46 | | - testonly=testonly, |
47 | | - visibility=visibility,) |
| 43 | + cc_proto_library( |
| 44 | + name = name + "_cc", |
| 45 | + srcs = srcs + tf_deps(deps, "_proto_srcs"), |
| 46 | + deps = deps + ["@protobuf//:cc_wkt_protos"], |
| 47 | + cc_libs = cc_libs + ["@protobuf//:protobuf"], |
| 48 | + protoc = "@protobuf//:protoc", |
| 49 | + default_runtime = "@protobuf//:protobuf", |
| 50 | + use_grpc_plugin = use_grpc_plugin, |
| 51 | + testonly = testonly, |
| 52 | + visibility = visibility, |
| 53 | + ) |
48 | 54 |
|
49 | 55 | def tf_proto_library_py(name, srcs=[], deps=[], visibility=[], testonly=0, |
50 | 56 | srcs_version="PY2AND3"): |
51 | | - py_proto_library(name = name + "_py", |
52 | | - srcs = srcs, |
53 | | - srcs_version = srcs_version, |
54 | | - deps = deps, |
55 | | - visibility = visibility, |
56 | | - testonly = testonly) |
| 57 | + py_proto_library( |
| 58 | + name = name + "_py", |
| 59 | + srcs = srcs, |
| 60 | + srcs_version = srcs_version, |
| 61 | + deps = deps, |
| 62 | + protoc = "@protobuf//:protoc", |
| 63 | + default_runtime = "@protobuf//:protobuf_python", |
| 64 | + visibility = visibility, |
| 65 | + testonly = testonly, |
| 66 | + ) |
57 | 67 |
|
58 | 68 | def tf_proto_library(name, srcs = [], has_services = None, |
59 | 69 | deps = [], visibility = [], testonly = 0, |
60 | 70 | cc_libs = [], |
61 | 71 | cc_api_version = 2, go_api_version = 2, |
62 | 72 | java_api_version = 2, |
63 | 73 | py_api_version = 2): |
64 | | - tf_proto_library_cc(name=name, |
65 | | - srcs=srcs + tf_deps(deps, "_proto_srcs"), |
66 | | - deps=deps, |
67 | | - cc_libs=cc_libs, |
68 | | - testonly=testonly, |
69 | | - visibility=visibility,) |
70 | | - |
71 | | - tf_proto_library_py(name=name, |
72 | | - srcs=srcs + tf_deps(deps, "_proto_srcs"), |
73 | | - srcs_version="PY2AND3", |
74 | | - deps=deps + ["//google/protobuf:protobuf_python"], |
75 | | - testonly=testonly, |
76 | | - visibility=visibility,) |
| 74 | + tf_proto_library_cc( |
| 75 | + name = name, |
| 76 | + srcs = srcs + tf_deps(deps, "_proto_srcs"), |
| 77 | + deps = deps, |
| 78 | + cc_libs = cc_libs, |
| 79 | + testonly = testonly, |
| 80 | + visibility = visibility, |
| 81 | + ) |
| 82 | + |
| 83 | + tf_proto_library_py( |
| 84 | + name = name, |
| 85 | + srcs = srcs + tf_deps(deps, "_proto_srcs"), |
| 86 | + srcs_version = "PY2AND3", |
| 87 | + deps = deps + ["@protobuf//:protobuf_python"], |
| 88 | + testonly = testonly, |
| 89 | + visibility = visibility, |
| 90 | + ) |
77 | 91 |
|
78 | 92 | def tf_additional_lib_srcs(): |
79 | 93 | return [ |
|
0 commit comments