From 34f5e2f6e0edc8c93f8c905aa6eac66d9bb92d29 Mon Sep 17 00:00:00 2001 From: Sandeep Dasgupta Date: Wed, 22 May 2024 18:24:17 -0700 Subject: [PATCH] Fix audit wheel issue PiperOrigin-RevId: 636362500 --- tensorflow/tools/pip_package/build_pip_package.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/tensorflow/tools/pip_package/build_pip_package.py b/tensorflow/tools/pip_package/build_pip_package.py index 10593503395532..99764afab49005 100644 --- a/tensorflow/tools/pip_package/build_pip_package.py +++ b/tensorflow/tools/pip_package/build_pip_package.py @@ -230,6 +230,10 @@ def patch_so(srcs_dir: str) -> None: "tensorflow/compiler/mlir/quantization/tensorflow/python/" "pywrap_quantize_model.so" ): "$ORIGIN/../../../../../python", + ( + "tensorflow/compiler/mlir/tensorflow_to_stablehlo/python/" + "pywrap_tensorflow_to_stablehlo.so" + ): "$ORIGIN/../../../../python", } for file, path in to_patch.items(): rpath = subprocess.check_output(