github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-02-21 03:00:39 -05:00
Code Issues Packages Projects Releases Wiki Activity

[DOCS] Add missing ops and corresponding comments (#1699)

Browse Source
This commit is contained in:
Keren Zhou
2023-05-21 15:18:48 -04:00
committed by GitHub
parent a2433f3135
commit 74dbb2fc0a
2 changed files with 69 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
docs/python-api/triton.language.rst
View File

@@ -23,6 +23,8 @@ Creation Ops
:nosignatures:
arange
cat
full
zeros
@@ -33,11 +35,13 @@ Shape Manipulation Ops
:toctree: generated
:nosignatures:
broadcast
broadcast_to
expand_dims
reshape
ravel
reshape
trans
view
Linear Algebra Ops
@@ -83,11 +87,13 @@ Math Ops
abs
exp
log
fdiv
cos
sin
sqrt
sigmoid
softmax
umulhi
Reduction Ops
@@ -151,4 +157,27 @@ Compiler Hint Ops
:toctree: generated
:nosignatures:
debug_barrier
max_contiguous
multiple_of
Debug Ops
-----------------
.. autosummary::
:toctree: generated
:nosignatures:
static_print
static_assert
device_print
device_assert
Iterators
-----------------
.. autosummary::
:toctree: generated
:nosignatures:
static_range