[dynamo, 3.14] Ensure typing.Union is correctly traced in Dynamo.

guilhermeleobas · guilhermeleobas · commit bb84cd85a9f3 · 2025-11-29T03:09:54.000Z
Fixes `test_vector_norm_decom_unbacked_checks_cpu` from `test/test_linalg.py` on Python 3.14 ghstack-source-id: 03776e8 Pull-Request: #169084
diff --git a/torch/_dynamo/utils.py b/torch/_dynamo/utils.py
@@ -1065,6 +1065,10 @@ def istype(obj: object, allowed_types: Any) -> bool:
     )
 
 
+if sys.version_info >= (3, 14):
+    _builtin_final_typing_classes += (typing.Union,)
+
+
 def is_typing(value: Any) -> bool:
     # _Final catches most of typing classes:
     #   - Any

Original file line number	Diff line number	Diff line change
`@@ -1065,6 +1065,10 @@ def istype(obj: object, allowed_types: Any) -> bool:`
`1065`	`1065`	`)`
`1066`	`1066`
`1067`	`1067`
	`1068`	`+if sys.version_info >= (3, 14):`
	`1069`	`+ _builtin_final_typing_classes += (typing.Union,)`
	`1070`	`+`
	`1071`	`+`
`1068`	`1072`	`def is_typing(value: Any) -> bool:`
`1069`	`1073`	`# _Final catches most of typing classes:`
`1070`	`1074`	`# - Any`