[XLA:Python] Fix build failure on non-linux platforms.

Don't call the transfer manager registration code if we haven't included the header. PiperOrigin-RevId: 720189214
openxla · Jan 27, 2025 · a1f85b7 · a1f85b7
1 parent b8a356a
commit a1f85b7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/xla/python/xla.cc b/xla/python/xla.cc
@@ -600,7 +600,9 @@ NB_MODULE(xla_extension, m) {
   BuildTracebackSubmodule(m);
   BuildMlirSubmodule(m);
   BuildCustomCallShardingPybindAPI(m);
+#if defined(__linux__)
   aux::RegisterTransferServerTypes(m);
+#endif  // defined(__linux__)
 
   // The following uses python bindings for PyClient defined above using
   // pybind11, and hence needs pybind11::module_ (not just nanobind::module_).