From 719f90b2cdebc9e718f308785433dc67a3894b56 Mon Sep 17 00:00:00 2001
From: "codeflash-ai[bot]"
 <148906541+codeflash-ai[bot]@users.noreply.github.com>
Date: Thu, 4 Dec 2025 15:48:39 +0000
Subject: [PATCH] Optimize in_stateless_scope

The optimization replaces `getattr(GLOBAL_STATE_TRACKER, name, None)` with `GLOBAL_STATE_TRACKER.__dict__.get(name, None)` in the `get_global_attribute` function, providing a **72% speedup**.

**Key optimization:**
- **Direct dictionary lookup** instead of Python's reflection mechanism (`getattr`)
- Bypasses the overhead of Python's attribute resolution protocol and default value handling in C code
- Uses the faster `dict.get()` method which is optimized at the C level

**Why this is faster:**
`getattr()` involves multiple layers of Python's attribute resolution machinery, including descriptor protocol checks and special method lookups. In contrast, `threading.local()` objects store their per-thread data in a simple `__dict__`, so direct dictionary access via `.get()` is much more efficient.

**Impact on workloads:**
The function references show `in_stateless_scope()` is called frequently in Keras variable operations - during variable initialization, value access, and assignment operations. Since these are core operations that can occur thousands of times during model training/inference, this micro-optimization has significant cumulative impact.

**Test case performance:**
The annotated tests show consistent 50-88% speedups across all scenarios, with the optimization being particularly effective for:
- Repeated attribute lookups (74.8% faster in the 1000-iteration test)
- Variable state checking in hot paths
- Both when attributes exist and when they're missing (consistent performance gains)

This optimization is safe because `threading.local().__dict__` is the documented way to access thread-local storage and maintains identical behavior while being substantially faster.
---
 keras/src/backend/common/global_state.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/keras/src/backend/common/global_state.py b/keras/src/backend/common/global_state.py
index 8ecf11b95056..50654b53629f 100644
--- a/keras/src/backend/common/global_state.py
+++ b/keras/src/backend/common/global_state.py
@@ -13,7 +13,7 @@ def set_global_attribute(name, value):
 
 
 def get_global_attribute(name, default=None, set_to_default=False):
-    attr = getattr(GLOBAL_STATE_TRACKER, name, None)
+    attr = GLOBAL_STATE_TRACKER.__dict__.get(name, None)
     if attr is None and default is not None:
         attr = default
         if set_to_default: