Skip to content

Commit

Permalink
[XLA:Python] Remove unused GOOGLE_CUDA/TENSORFLOW_USE_ROCM macro guar…
Browse files Browse the repository at this point in the history
…ding a header inclusion.

As far as I can tell, this does nothing at all.

PiperOrigin-RevId: 717243277
  • Loading branch information
hawkinsp authored and Google-ML-Automation committed Jan 19, 2025
1 parent b21aaff commit 665f79f
Show file tree
Hide file tree
Showing 2 changed files with 0 additions and 7 deletions.
3 changes: 0 additions & 3 deletions xla/python/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down
4 changes: 0 additions & 4 deletions xla/python/py_client.cc
Original file line number Diff line number Diff line change
Expand Up @@ -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;
Expand Down

0 comments on commit 665f79f

Please sign in to comment.