Don't assume compare_arg is fx.Node (pytorch#126771)

Signed-off-by: Edward Z. Yang <[email protected]> Pull Request resolved: pytorch#126771 Approved by: https://github.com/Skylion007
weiyusheng · May 21, 2024 · 97eef61 · 97eef61
1 parent fc594ed
commit 97eef61
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch/_export/passes/add_runtime_assertions_for_constraints_pass.py b/torch/_export/passes/add_runtime_assertions_for_constraints_pass.py
@@ -177,6 +177,7 @@ def _get_existing_inline_assertions(
 
             compare_arg = node.args[0]
             if not (
+                isinstance(compare_arg, torch.fx.Node) and
                 compare_arg.op == "call_function" and
                 compare_arg.target in (operator.le, operator.ge) and
                 len(compare_arg.args) == 2