Unify TORCH_LIBRARY definitions (pytorch#6455)

Summary: This pull request tries to unify all TORCH_LIBRARY definitions across torch_xla into one xla library. Test Plan: CI
amithrm · Mar 1, 2024 · 3bee0a7 · 3bee0a7
1 parent 5c9c4a9
commit 3bee0a7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/torch_xla/core/xla_model.py b/torch_xla/core/xla_model.py
@@ -40,6 +40,8 @@
 # Default bucket size for all-reduce
 _ALLREDUCE_BUCKET_CAP_MB = 50
 
+XLA_LIB = Library("xla", "DEF")
+
 
 def _init_world_size_ordinal():
   global _WORLD_SIZE, _ORDINAL