Google-internal change, should be no-op externally.

PiperOrigin-RevId: 718144810
Intel-tensorflow · Jan 22, 2025 · c1245bb · c1245bb
1 parent 7b27f6c
commit c1245bb
Show file tree

Hide file tree

Showing 5 changed files with 8 additions and 2 deletions.
diff --git a/xla/python/ifrt/BUILD b/xla/python/ifrt/BUILD
@@ -642,6 +642,7 @@ cc_library(
     name = "plugin_program",
     srcs = ["plugin_program.cc"],
     hdrs = ["plugin_program.h"],
+    compatible_with = get_compatible_with_portable(),
     deps = [
         ":ifrt",
         "@llvm-project//llvm:Support",
@@ -651,6 +652,7 @@ cc_library(
 cc_library(
     name = "plugin_program_serdes",
     srcs = ["plugin_program_serdes.cc"],
+    compatible_with = get_compatible_with_portable(),
     deps = [
         ":plugin_program",
         ":serdes",

diff --git a/xla/python/ifrt_proxy/common/BUILD b/xla/python/ifrt_proxy/common/BUILD
@@ -12,7 +12,7 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-load("//xla/python/ifrt_proxy/common:ifrt_proxy.bzl", "default_ifrt_proxy_visibility", "ifrt_proxy_cc_test")
+load("//xla/python/ifrt_proxy/common:ifrt_proxy.bzl", "cc_library", "default_ifrt_proxy_visibility", "ifrt_proxy_cc_test")
 load("//xla/tsl:tsl.bzl", "if_google")
 load("//xla/tsl/platform:build_config.bzl", "tf_proto_library")
 # copybara:uncomment load("@bazel_skylib//:bzl_library.bzl", "bzl_library")

diff --git a/xla/python/ifrt_proxy/common/ifrt_proxy.bzl b/xla/python/ifrt_proxy/common/ifrt_proxy.bzl
@@ -6,3 +6,6 @@ def ifrt_proxy_cc_test(**kwargs):
     xla_cc_test(**kwargs)
 
 default_ifrt_proxy_visibility = ["//xla/python/ifrt_proxy:__subpackages__"]
+
+def cc_library(**attrs):
+    native.cc_library(**attrs)
diff --git a/xla/python/ifrt_proxy/server/BUILD b/xla/python/ifrt_proxy/server/BUILD
@@ -11,7 +11,7 @@
 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 # See the License for the specific language governing permissions and
 # limitations under the License.
-load("//xla/python/ifrt_proxy/common:ifrt_proxy.bzl", "default_ifrt_proxy_visibility", "ifrt_proxy_cc_test")
+load("//xla/python/ifrt_proxy/common:ifrt_proxy.bzl", "cc_library", "default_ifrt_proxy_visibility", "ifrt_proxy_cc_test")
 
 package(
     # copybara:uncomment default_applicable_licenses = ["//tensorflow:license"],

diff --git a/xla/python/pjrt_ifrt/BUILD b/xla/python/pjrt_ifrt/BUILD
@@ -81,6 +81,7 @@ tf_proto_library(
 cc_library(
     name = "xla_sharding_serdes",
     srcs = ["xla_sharding_serdes.cc"],
+    compatible_with = get_compatible_with_portable(),
     deps = [
         ":xla_ifrt",
         ":xla_sharding_proto_cc",