From 4941bc7001f84485837212e8e6d8eb5d128ca4f1 Mon Sep 17 00:00:00 2001 From: Philippe Tillet Date: Tue, 8 Feb 2022 16:53:56 -0800 Subject: [PATCH] [DOCS] Some more fixes (#455) --- docs/conf.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) 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/.*$'