--- tensorflow/python/BUILD.orig 2026-03-04 21:33:50 UTC +++ tensorflow/python/BUILD @@ -1479,6 +1479,21 @@ pybind_extension( "//tensorflow/distribute/experimental/rpc/kernels:rpc_ops", "//tensorflow/dtensor/cc:tensor_layout", "@xla//xla/backends/profiler/cpu:python_tracer", + # Shardy dialect deps: ensure these archives are included in libtensorflow_cc.so.2 + # via the pywrap split-library mechanism (they would otherwise be filtered out as + # starlark_only, since pywrap_quantization is starlark_only=True). + "@shardy//shardy/common:file_utils", + "@shardy//shardy/common:logging", + "@shardy//shardy/dialect/sdy/ir:axis_list_ref", + "@shardy//shardy/dialect/sdy/ir:dialect", + "@shardy//shardy/dialect/sdy/transforms/common:op_properties", + "@shardy//shardy/dialect/sdy/transforms/common:sharding_walker", + "@shardy//shardy/dialect/sdy/transforms/import:passes", + "@shardy//shardy/dialect/sdy/transforms/propagation:op_sharding_rule_builder", + "@shardy//shardy/dialect/sdy/transforms/propagation:op_sharding_rule_registry", + "@shardy//shardy/dialect/sdy/transforms/propagation:sharding_projection", + "@shardy//shardy/dialect/sdy/transforms/propagation:utils", + "@shardy//shardy/dialect/sdy/transforms/propagation/debugging:source_sharding", ], ) @@ -1577,6 +1592,11 @@ pywrap_library( "-ldl", "-lm", ], + "//tensorflow:freebsd": [ + "-lpthread", + "-ldl", + "-lm", + ], "//conditions:default": [ "-z defs", "-lpthread", @@ -1593,6 +1613,11 @@ pywrap_library( "-DEFAULTLIB:ntdll.lib", ], "//tensorflow:macos": [ + "-lpthread", + "-ldl", + "-lm", + ], + "//tensorflow:freebsd": [ "-lpthread", "-ldl", "-lm",