fix a bug in checking wrong object's type (#66)

thuml · Nov 24, 2024 · 2acfd19 · 2acfd19
1 parent 2acb742
commit 2acfd19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/depyf/explain/utils.py b/depyf/explain/utils.py
@@ -119,7 +119,7 @@ def __init__(self, original_code, module, cache):
             def visit(root, ans):
                 nonlocal tensor_aliasing_guard_seen
                 for leaf_guard in root.get_leaf_guards():
-                    if isinstance(guard_codes, torch._C._dynamo.guards.NO_TENSOR_ALIASING):
+                    if isinstance(leaf_guard, torch._C._dynamo.guards.NO_TENSOR_ALIASING):
                         if not tensor_aliasing_guard_seen:
                             tensor_aliasing_guard_seen = True
                         else: