: constant fold None (pytorch#10755)

ThomasJannaud · facebook-github-bot · commit d1a7f97395ab · 2025-05-07T11:46:01.000-07:00
Summary:

Constant folding should fold 'None' and consider it a constant

Reviewed By: dulinriley

Differential Revision: D74280168
diff --git a/exir/passes/constant_prop_pass.py b/exir/passes/constant_prop_pass.py
@@ -61,6 +61,8 @@ def is_const(
         )
     elif isinstance(arg, _PRIMITIVE_TYPES):
         return True
+    elif arg is None:
+        return True
     elif not isinstance(arg, torch.fx.Node):
         return False
     elif arg in const_node_to_tensor:

Original file line number	Diff line number	Diff line change
`@@ -61,6 +61,8 @@ def is_const(`
`61`	`61`	`)`
`62`	`62`	`elif isinstance(arg, _PRIMITIVE_TYPES):`
`63`	`63`	`return True`
	`64`	`+ elif arg is None:`
	`65`	`+ return True`
`64`	`66`	`elif not isinstance(arg, torch.fx.Node):`
`65`	`67`	`return False`
`66`	`68`	`elif arg in const_node_to_tensor:`