Use capture autograph if capture is enabled (#1960)

Author: albi3ro

doc/releases/changelog-dev.md

@@ -57,6 +57,9 @@
 
 <h3>Internal changes ⚙️</h3>
 
+* When capture is enabled, `qjit(autograph=True)` will use capture autograph instead of catalyst autograph.
+  [(#1960)](https://github.com/PennyLaneAI/catalyst/pull/1960)
+
 * QJitDevice helper `extract_backend_info` removed its redundant `capabilities` argument.
   [(#1956)](https://github.com/PennyLaneAI/catalyst/pull/1956)
 

frontend/catalyst/jit.py

@@ -692,6 +692,12 @@ def jit_compile(self, args, **kwargs):
     def pre_compilation(self):
         """Perform pre-processing tasks on the Python function, such as AST transformations."""
         if self.compile_options.autograph:
+            if qml.capture.enabled():
+                if self.compile_options.autograph_include:
+                    raise NotImplementedError(
+                        "capture autograph does not yet support autograph_include."
+                    )
+                return qml.capture.run_autograph(self.original_function)
             return run_autograph(self.original_function, *self.compile_options.autograph_include)
 
         return self.original_function