Ensure typing.Union is correctly traced in Dynamo. (#169084)

generatedunixname499836121 · meta-codesync[bot] · commit 9e56263eaec8 · 2025-11-29T21:36:20.000-08:00
Summary: Fixes `test_vector_norm_decom_unbacked_checks_cpu` from `test/test_linalg.py` on Python 3.14 X-link: pytorch/pytorch#169084 Approved by: https://github.com/rtimpe, https://github.com/williamwen42 Reviewed By: wdvr Differential Revision: D88036656 fbshipit-source-id: 34d374685fc18a03afd2bcd2509b4a5566b24940
diff --git a/userbenchmark/dynamo/dynamobench/_dynamo/utils.py b/userbenchmark/dynamo/dynamobench/_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`