Handle case when native_schema is None

metascroy · facebook-github-bot · commit 72b4e0839dfc · 2025-03-18T14:51:52.000-07:00
Summary: https://fb.workplace.com/groups/pytorch.edge.users/posts/1714672769403009/?comment_id=1721960698674216 Differential Revision: D71413713
diff --git a/exir/program/_program.py b/exir/program/_program.py
@@ -995,13 +995,13 @@ def keep(op):
 
         schema = op._schema
         native_schema = _pybind_schema_to_native_schema(schema)
-        if native_schema.is_mutable:
+        if (native_schema is not None) and native_schema.is_mutable:
             logging.warn(
                 f"Op {op} was requested for preservation by partitioner.  This request is ignored because it is mutable."
             )
             return False
 
-        if native_schema.aliased_return_names() != [None]:
+        if (native_schema is not None) and native_schema.aliased_return_names() != [None]:
             logging.warn(
                 f"Op {op} was requested for preservation by partitioner.  This request is ignored because it aliases output."
             )

Original file line number	Diff line number	Diff line change
`@@ -995,13 +995,13 @@ def keep(op):`
`995`	`995`
`996`	`996`	`schema = op._schema`
`997`	`997`	`native_schema = _pybind_schema_to_native_schema(schema)`
`998`		`- if native_schema.is_mutable:`
	`998`	`+ if (native_schema is not None) and native_schema.is_mutable:`
`999`	`999`	`logging.warn(`
`1000`	`1000`	`f"Op {op} was requested for preservation by partitioner. This request is ignored because it is mutable."`
`1001`	`1001`	`)`
`1002`	`1002`	`return False`
`1003`	`1003`
`1004`		`- if native_schema.aliased_return_names() != [None]:`
	`1004`	`+ if (native_schema is not None) and native_schema.aliased_return_names() != [None]:`
`1005`	`1005`	`logging.warn(`
`1006`	`1006`	`f"Op {op} was requested for preservation by partitioner. This request is ignored because it aliases output."`
`1007`	`1007`	`)`