From 919d6b57eb4bd353ef809e31f22f5f509b6e431a Mon Sep 17 00:00:00 2001
From: "Di Xu (SWE)" <dixu@meta.com>
Date: Tue, 23 Dec 2025 19:31:29 -0800
Subject: [PATCH] Fix dynamic shape missing Symbol errors

Summary: More context in https://fb.workplace.com/groups/pytorch.edge.users/permalink/1941980700005547/

Differential Revision: D89754133
---
 exir/program/_program.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/exir/program/_program.py b/exir/program/_program.py
index c5100394aa3..4ce09c2bf2c 100644
--- a/exir/program/_program.py
+++ b/exir/program/_program.py
@@ -149,9 +149,8 @@ def get_shape_env(gm):
     if shape_env is None:
         return {}
     range_constraints = {
-        k: v
+        shape_env.replacements.get(k, k): v
         for k, v in shape_env.var_to_range.items()
-        if k not in shape_env.replacements
     }
     # Only when we have an unbacked symint, and it's used as constructor inputs,
     # runtime_var_to_range will make a difference compated to var_to_range.