From 3bee0a7ac4f0ec7513beea8d9543160d46d5f349 Mon Sep 17 00:00:00 2001
From: Jiewen Tan <jwtan@google.com>
Date: Fri, 2 Feb 2024 10:25:09 -0800
Subject: [PATCH] Unify TORCH_LIBRARY definitions (#6455)

Summary:
This pull request tries to unify all TORCH_LIBRARY definitions across torch_xla into one xla library.

Test Plan:
CI
---
 torch_xla/core/xla_model.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/torch_xla/core/xla_model.py b/torch_xla/core/xla_model.py
index 1e4243480c5..91c2bf22b3b 100755
--- 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