From e5dba5bbd5a29901d73d73b9fa693ee361594054 Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Fri, 24 Jan 2025 12:39:19 -0800 Subject: [PATCH] [XLA:Python] Remove JAX CUDA rpaths setting. This is now no longer used by JAX, and never matter on xla_extension.so because we no longer build it in a CUDA-specific configuration, instead using CUDA plugins. PiperOrigin-RevId: 719399140 --- xla/python/BUILD | 27 --------------------------- 1 file changed, 27 deletions(-) diff --git a/xla/python/BUILD b/xla/python/BUILD index e03b2b727309d..de1a057bcc00f 100644 --- a/xla/python/BUILD +++ b/xla/python/BUILD @@ -1,4 +1,3 @@ -load("@bazel_skylib//rules:common_settings.bzl", "bool_flag") load("@local_config_cuda//cuda:build_defs.bzl", "if_cuda") load("@local_config_rocm//rocm:build_defs.bzl", "if_rocm") load("//xla:pytype.default.bzl", "pytype_strict_library") @@ -1170,20 +1169,6 @@ tf_proto_library( srcs = ["py_host_callback.proto"], ) -# If this flag is enabled, it sets RPATH on the xla_extension to values that are suitable for -# finding NVIDIA's CUDA libraries when they are installed as pip packages. -bool_flag( - name = "jax_cuda_pip_rpaths", - build_setting_default = False, -) - -config_setting( - name = "use_jax_cuda_pip_rpaths", - flag_values = { - ":jax_cuda_pip_rpaths": "True", - }, -) - cc_library( name = "logging", srcs = ["logging.cc"], @@ -1251,18 +1236,6 @@ tsl_pybind_extension( "-fno-strict-aliasing", ], features = ["-use_header_modules"], - linkopts = select({ - ":use_jax_cuda_pip_rpaths": [ - "-Wl,-rpath,$$ORIGIN/../nvidia/cuda_cupti/lib", - "-Wl,-rpath,$$ORIGIN/../nvidia/cuda_runtime/lib", - "-Wl,-rpath,$$ORIGIN/../nvidia/cublas/lib", - "-Wl,-rpath,$$ORIGIN/../nvidia/cufft/lib", - "-Wl,-rpath,$$ORIGIN/../nvidia/cudnn/lib", - "-Wl,-rpath,$$ORIGIN/../nvidia/cusolver/lib", - "-Wl,-rpath,$$ORIGIN/../nvidia/nccl/lib", - ], - "//conditions:default": [], - }), pytype_deps = [ "//third_party/py/numpy", ],