[DOCS] Various improvements (#224)

- Added docstr for autotune, Config, heuristics - Added docstr for atomics - Hiding internal _builder argument used for built-in language primitives - Re-factor docstr to use common templates between similar functions.
2021-08-18 11:15:53 -07:00
parent 226fde6ea1
commit f26a48a3b4
6 changed files with 275 additions and 222 deletions
--- a/docs/python-api/triton.language.rst
+++ b/docs/python-api/triton.language.rst
@@ -98,6 +98,18 @@ Reduction Ops
    min
    sum

+Atomic Ops
+---------------
+
+.. autosummary::
+    :toctree: generated
+    :nosignatures:
+
+    atomic_cas
+    atomic_add
+    atomic_max
+    atomic_min
+

 Comparison ops
 ---------------
--- a/docs/python-api/triton.rst
+++ b/docs/python-api/triton.rst
@@ -7,4 +7,7 @@ triton
    :toctree: generated
    :nosignatures:

-    jit
+    jit
+    autotune
+    heuristics
+    Config