diff --git a/docs/conf.py b/docs/conf.py index 09d355dcd..4d62c5650 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -34,14 +34,17 @@ def process_sig(app, what, name, obj, options, signature, return_annotation): def setup(app): """Customize function args retrieving to get args under decorator.""" import sphinx - import triton + import os app.connect("autodoc-process-signature", process_sig) + os.system("pip install -e ../python") + def forward_jit_fn(func): old = func def wrapped(obj, **kwargs): + import triton if isinstance(obj, triton.code_gen.JITFunction): obj = obj.fn return old(obj) @@ -52,6 +55,7 @@ def setup(app): old_documenter = sphinx.ext.autosummary.get_documenter def documenter(app, obj, parent): + import triton if isinstance(obj, triton.code_gen.JITFunction): obj = obj.fn return old_documenter(app, obj, parent) @@ -70,7 +74,7 @@ extensions = ['sphinx.ext.autodoc', 'sphinx.ext.intersphinx', 'sphinx.ext.autosu autosummary_generate = True # versioning config -smv_tag_whitelist = r'^(v1.1.1)$' +smv_tag_whitelist = r'^(v1.1.2)$' smv_branch_whitelist = r'^master$' smv_remote_whitelist = None smv_released_pattern = r'^tags/.*$'