code: def setup(app):
      """Customize function args retrieving to get args under decorator."""
      import os
  
      import sphinx
  
      app.connect("autodoc-process-signature", process_sig)
      os.system("pip install -e ../python")
      setup_generated_mlir_docs()
  
      def forward_jit_fn(func):
          old = func
  
          def wrapped(obj, **kwargs):
              import triton
  
              if isinstance(obj, triton.runtime.JITFunction):
                  obj = obj.fn
              return old(obj)
  
          return wrapped
  
      old_documenter = sphinx.ext.autosummary.get_documenter
  
      def documenter(app, obj, parent):
          import triton
  
          if isinstance(obj, triton.runtime.JITFunction):
              obj = obj.fn
          return old_documenter(app, obj, parent)
  
      sphinx.ext.autosummary.get_documenter = documenter
      sphinx.util.inspect.unwrap_all = forward_jit_fn(
          sphinx.util.inspect.unwrap_all
      )
      sphinx.util.inspect.signature = forward_jit_fn(
          sphinx.util.inspect.signature
      )
      sphinx.util.inspect.object_description = forward_jit_fn(
          sphinx.util.inspect.object_description
      )
  
file_path: triton/docs/conf.py
imports: import os
  import shutil
  import sys
  import sysconfig
  from pathlib import Path
  import sphinx_rtd_theme
  from sphinx_gallery.sorting import FileNameSortKey
label_desc: Use image node_id 4 for calling a global function with example usage: setup(app) and returns: wrapped, old_documenter, old