From f610f704209ad9213087caa29b1385cea4a8d80b Mon Sep 17 00:00:00 2001
From: Rui <179625410+rpsilva-aws@users.noreply.github.com>
Date: Wed, 13 Nov 2024 17:26:52 -0800
Subject: [PATCH] Expose lazy all number special scalars (#8361)

---
 torch_xla/csrc/init_python_bindings.cpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/torch_xla/csrc/init_python_bindings.cpp b/torch_xla/csrc/init_python_bindings.cpp
index 2ac991bc814..7e36a20c0e5 100644
--- a/torch_xla/csrc/init_python_bindings.cpp
+++ b/torch_xla/csrc/init_python_bindings.cpp
@@ -2504,6 +2504,8 @@ void InitXlaModuleBindings(py::module m) {
           FLAGS_torch_lazy_all_numbers_special_scalars =
               all_numbers_special_scalars;
         });
+  m.def("_get_xla_all_numbers_special_scalars",
+        []() { return FLAGS_torch_lazy_all_numbers_special_scalars; });
   m.def("_set_xla_handle_special_scalars", [](bool handle_special_scalars) {
     FLAGS_torch_lazy_handle_special_scalars = handle_special_scalars;
   });