fix

Author: rniczh

frontend/catalyst/autograph/ag_primitives.py

@@ -609,7 +609,7 @@ def qnode_call_wrapper():
         if hasattr(fn, "__wrapped__") and hasattr(fn, "__module__"):
             # Find the decorator function
             if decorator := next(
-                (f for f in _known_wrapper_functions if hasattr(f, "__module__")), None
+                (f for f in _known_wrapper_functions if f.__module__ == fn.__module__), None
             ):
                 original_fn = fn.__wrapped__