From 665f79fbdab32716ec6cc341e7aa3ef6e93e9d6a Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Sun, 19 Jan 2025 07:33:33 -0800 Subject: [PATCH] [XLA:Python] Remove unused GOOGLE_CUDA/TENSORFLOW_USE_ROCM macro guarding a header inclusion. As far as I can tell, this does nothing at all. PiperOrigin-RevId: 717243277 --- xla/python/BUILD | 3 --- xla/python/py_client.cc | 4 ---- 2 files changed, 7 deletions(-) diff --git a/xla/python/BUILD b/xla/python/BUILD index e848931e3f42b..8da625a4a4ace 100644 --- a/xla/python/BUILD +++ b/xla/python/BUILD @@ -333,9 +333,6 @@ cc_library( "-fexceptions", "-fno-strict-aliasing", ], - defines = if_cuda(["GOOGLE_CUDA=1"]) + if_rocm([ - "TENSORFLOW_USE_ROCM=1", - ]), features = ["-use_header_modules"], deps = [ ":aggregate_profile", diff --git a/xla/python/py_client.cc b/xla/python/py_client.cc index da4eca022ed73..72c3d9cb95929 100644 --- a/xla/python/py_client.cc +++ b/xla/python/py_client.cc @@ -94,10 +94,6 @@ limitations under the License. #include "tsl/platform/status.h" #include "tsl/platform/statusor.h" -#if GOOGLE_CUDA || TENSORFLOW_USE_ROCM || TENSORFLOW_USE_SYCL -#include "xla/python/py_client_gpu.h" -#endif // GOOGLE_CUDA || TENSORFLOW_USE_ROCM || TENSORFLOW_USE_SYCL - namespace xla { namespace nb = nanobind;