fix previous pytorch version (#65)

Signed-off-by: youkaichao <[email protected]>
thuml · Nov 4, 2024 · 2ba8467 · 2ba8467
1 parent 6ca9d8c
commit 2ba8467
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/depyf/explain/utils.py b/depyf/explain/utils.py
@@ -99,6 +99,7 @@ def __init__(self, original_code, module, cache):
 
         try:
             klass = getattr(torch._dynamo.guards, "GuardManagerWrapper", None) or \
+                getattr(torch._dynamo.guards, "GuardManager", None) or \
                 getattr(torch._C._dynamo.guards, "GuardManager", None)
             assert klass is not None
             cpp_guard = isinstance(guard_manager, klass)