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: