From 92cd06d9d36e5f269dab5b3911afb27026e162ec Mon Sep 17 00:00:00 2001 From: Thomas Jannaud Date: Wed, 7 May 2025 11:48:36 -0700 Subject: [PATCH] constant fold None (#10755) Summary: Pull Request resolved: https://github.com/pytorch/executorch/pull/10755 Constant folding should fold 'None' and consider it a constant Reviewed By: dulinriley Differential Revision: D74280168 --- exir/passes/constant_prop_pass.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/exir/passes/constant_prop_pass.py b/exir/passes/constant_prop_pass.py index 0049e597f8d..cf864153ba8 100644 --- 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: