[DOCS] Some more fixes (#455)

This commit is contained in:
Philippe Tillet
2022-02-08 16:53:56 -08:00
committed by GitHub
parent 2fdf0a4fe8
commit 4941bc7001

View File

@@ -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/.*$'