Files
triton/v1.1.2/.doctrees/python-api/triton.language.doctree

254 lines
60 KiB
Plaintext
Raw Normal View History

2022-02-08 20:02:05 +00:00
<EFBFBD><05>m<EFBFBD><00>sphinx.addnodes<65><73>document<6E><74><EFBFBD>)<29><>}<7D>(<28> rawsource<63><65><00><>children<65>]<5D><>docutils.nodes<65><73>section<6F><6E><EFBFBD>)<29><>}<7D>(hhh]<5D>(h <09>title<6C><65><EFBFBD>)<29><>}<7D>(h<05>triton.language<67>h]<5D>h <09>Text<78><74><EFBFBD><EFBFBD>triton.language<67><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<13>parent<6E>hhh<03>source<63>N<EFBFBD>line<6E>Nuba<62>
2022-04-20 00:43:07 +00:00
attributes<EFBFBD>}<7D>(<28>ids<64>]<5D><>classes<65>]<5D><>names<65>]<5D><>dupnames<65>]<5D><>backrefs<66>]<5D>u<EFBFBD>tagname<6D>hhh hhh<1C>]/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst<73>hKubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>Programming Model<65>h]<5D>h<16>Programming Model<65><6C><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh1hh/hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhh,hhhh+hKubh<00>tabular_col_spec<65><63><EFBFBD>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>spec<65><63>\X{1}{2}\X{1}{2}<7D>uh*h=hh,hhh<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:17:<autosummary><3E>hNub<75>sphinx.ext.autosummary<72><79>autosummary_table<6C><65><EFBFBD>)<29><>}<7D>(h<05><>
2022-02-08 20:02:05 +00:00
program_id
Returns the id of the current program instance along the given axis.
num_programs
Returns the number of program instances launched along the given axis.<2E>h]<5D>h <09>table<6C><65><EFBFBD>)<29><>}<7D>(hhh]<5D>h <09>tgroup<75><70><EFBFBD>)<29><>}<7D>(hhh]<5D>(h <09>colspec<65><63><EFBFBD>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
uh*h\hhYubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hhYubh <09>tbody<64><79><EFBFBD>)<29><>}<7D>(hhh]<5D>(h <09>row<6F><77><EFBFBD>)<29><>}<7D>(hhh]<5D>(h <09>entry<72><79><EFBFBD>)<29><>}<7D>(hhh]<5D>h <09> paragraph<70><68><EFBFBD>)<29><>}<7D>(h<05>.:obj:`program_id <triton.language.program_id>`<60>h]<5D>h<00> pending_xref<65><66><EFBFBD>)<29><>}<7D>(hh<>h]<5D>h <09>literal<61><6C><EFBFBD>)<29><>}<7D>(hh<>h]<5D>h<16>
2022-04-20 00:43:07 +00:00
program_id<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhh<>ubah}<7D>(h ]<5D>h"]<5D>(<28>xref<65><66>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F><63>python-api/triton.language<67><65> refdomain<69>h<EFBFBD><68>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E><EFBFBD> py:module<6C><65>triton.language<67><65>py:class<73>N<EFBFBD> reftarget<65><74>triton.language.program_id<69>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:17:<autosummary><3E>hKhh<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hKhh~ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hhyubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>LReturns the id of the current program instance along the given :code:`axis`.<2E>h]<5D>(h<16>?Returns the id of the current program instance along the given <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>?Returns the id of the current program instance along the given <20>hh<>ubh<62>)<29><>}<7D>(h<05> :code:`axis`<60>h]<5D>h<16>axis<69><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>axis<69>hh<>ubah}<7D>(h ]<5D>h"]<5D><>code<64>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>.<2E>hh<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:17:<autosummary><3E>hKhh<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hhyubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhhtubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>2:obj:`num_programs <triton.language.num_programs>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hh<>h]<5D>h<EFBFBD>)<29><>}<7D>(hh<>h]<5D>h<16> num_programs<6D><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhh<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.num_programs<6D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:17:<autosummary><3E>hKhh<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjhKhh<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hh<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>NReturns the number of program instances launched along the given :code:`axis`.<2E>h]<5D>(h<16>AReturns the number of program instances launched along the given <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>AReturns the number of program instances launched along the given <20>hj*ubh<62>)<29><>}<7D>(h<05> :code:`axis`<60>h]<5D>h<16>axis<69><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>axis<69>hj3ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj*ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj*ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hhJhKhj'ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hh<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhhtubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhhYubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhhTubah}<7D>(h ]<5D>h"]<5D><> longtable<6C>ah$]<5D>h&]<5D>h(]<5D>uh*hRhhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><> xml:space<63><65>preserve<76>uh*hLhh,hhhhJhNubhK<68>autosummary_toc<6F><63><EFBFBD>)<29><>}<7D>(hhh]<5D>h<00>toctree<65><65><EFBFBD>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><> includefiles<65>]<5D>(<28>/python-api/generated/triton.language.program_id<69><64>1python-api/generated/triton.language.num_programs<6D>e<EFBFBD>entries<65>]<5D>(Nj<4E><00><>Nj<4E><00><>e<EFBFBD>maxdepth<74>J<EFBFBD><4A><EFBFBD><EFBFBD><EFBFBD>glob<6F>N<EFBFBD>caption<6F>N<EFBFBD>
2022-02-08 20:02:05 +00:00
rawentries<EFBFBD>]<5D>uh*jyhjvubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthh,hhhhJhNubeh}<7D>(h ]<5D><>programming-model<65>ah"]<5D>h$]<5D><>programming model<65>ah&]<5D>h(]<5D>uh*h
2022-04-20 00:43:07 +00:00
hh hhhh+hKubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05> Creation Ops<70>h]<5D>h<16> Creation Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>hj<>hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj<>hhhh+hKubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj<>hhh<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:28:<autosummary><3E>hNubhM)<29><>}<7D>(h<05><>
2022-02-08 20:02:05 +00:00
arange
Returns contiguous values within the open interval [start, end).
zeros
Returns a block filled with the scalar value 0 for the given shape and dtype.<2E>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj<>ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj<>ubhs)<29><>}<7D>(hhh]<5D>(hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>&:obj:`arange <triton.language.arange>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>arange<67><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.arange<67>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:28:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjhKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>PReturns contiguous values within the open interval [:code:`start`, :code:`end`).<2E>h]<5D>(h<16>4Returns contiguous values within the open interval [<5B><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>4Returns contiguous values within the open interval [<5B>hjubh<62>)<29><>}<7D>(h<05> :code:`start`<60>h]<5D>h<16>start<72><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>start<72>hjubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjubh<16>, <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>, <20>hjubh<62>)<29><>}<7D>(h<05> :code:`end`<60>h]<5D>h<16>end<6E><64><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>end<6E>hj1ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjubh<16>).<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>).<2E>hjubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:28:<autosummary><3E>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>$:obj:`zeros <triton.language.zeros>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj`h]<5D>h<EFBFBD>)<29><>}<7D>(hj`h]<5D>h<16>zeros<6F><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjeubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hjbubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>jo<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.zeros<6F>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:28:<autosummary><3E>hKhj^ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj[ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjXubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>]Returns a block filled with the scalar value 0 for the given :code:`shape` and :code:`dtype`.<2E>h]<5D>(h<16>=Returns a block filled with the scalar value 0 for the given <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>=Returns a block filled with the scalar value 0 for the given <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`shape`<60>h]<5D>h<16>shape<70><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>shape<70>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16> and <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> and <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`dtype`<60>h]<5D>h<16>dtype<70><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>dtype<70>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjXubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj<>ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhj<>hhhj<>hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D>(<28>+python-api/generated/triton.language.arange<67><65>*python-api/generated/triton.language.zeros<6F>ej<65>]<5D>(Nj<4E><00><>Nj<4E><00><>ej<65>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthj<>hhhj<>hNubeh}<7D>(h ]<5D><> creation-ops<70>ah"]<5D>h$]<5D><> creation ops<70>ah&]<5D>h(]<5D>uh*h
hh hhhh+hKubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>Shape Manipulation Ops<70>h]<5D>h<16>Shape Manipulation Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjhjhhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj hhhh+hKubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj hhh<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:41:<autosummary><3E>hNubhM)<29><>}<7D>(h<05><>
2022-02-08 20:02:05 +00:00
broadcast_to
Tries to broadcast the given block to a new shape.
reshape
Tries to reshape the given block to a new shape.
ravel
Returns a contiguous flattened view of x<>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj/ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj/ubhs)<29><>}<7D>(hhh]<5D>(hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>2:obj:`broadcast_to <triton.language.broadcast_to>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hjQh]<5D>h<EFBFBD>)<29><>}<7D>(hjQh]<5D>h<16> broadcast_to<74><6F><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjVubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hjSubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j`<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.broadcast_to<74>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:41:<autosummary><3E>hKhjOubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjrhKhjLubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjIubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>:Tries to broadcast the given block to a new :code:`shape`.<2E>h]<5D>(h<16>,Tries to broadcast the given block to a new <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>,Tries to broadcast the given block to a new <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`shape`<60>h]<5D>h<16>shape<70><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>shape<70>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:41:<autosummary><3E>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjIubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhjFubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>(:obj:`reshape <triton.language.reshape>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>reshape<70><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.reshape<70>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:41:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>0Tries to reshape the given block to a new shape.<2E>h]<5D>h<16>0Tries to reshape the given block to a new shape.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:41:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhjFubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>$:obj:`ravel <triton.language.ravel>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj h]<5D>h<EFBFBD>)<29><>}<7D>(hj h]<5D>h<16>ravel<65><6C><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.ravel<65>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:41:<autosummary><3E>hKhj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj.hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>0Returns a contiguous flattened view of :code:`x`<60>h]<5D>(h<16>'Returns a contiguous flattened view of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>'Returns a contiguous flattened view of <20>hj>ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>x<>hjGubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj'hKhj;ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhjFubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj/ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj,ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj(ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhj hhhj'hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D>(<28>1python-api/generated/triton.language.broadcast_to<74><6F>,python-api/generated/triton.language.reshape<70><65>*python-api/generated/triton.language.ravel<65>ej<65>]<5D>(Nj<4E><00><>Nj<4E><00><>Nj<4E><00><>ej<65>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthj hhhj'hNubeh}<7D>(h ]<5D><>shape-manipulation-ops<70>ah"]<5D>h$]<5D><>shape manipulation ops<70>ah&]<5D>h(]<5D>uh*h
hh hhhh+hKubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>Linear Algebra Ops<70>h]<5D>h<16>Linear Algebra Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>hj<>hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj<>hhhh+hK+ubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj<>hhh<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:50:<autosummary><3E>hNubhM)<29><>}<7D>(h<05>2
2022-02-08 20:02:05 +00:00
dot
Returns the matrix product of two blocks.<2E>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj<>ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj<>ubhs)<29><>}<7D>(hhh]<5D>hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05> :obj:`dot <triton.language.dot>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>dot<6F><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.dot<6F>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:50:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>)Returns the matrix product of two blocks.<2E>h]<5D>h<16>)Returns the matrix product of two blocks.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj<>ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhj<>hhhj<>hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D><>(python-api/generated/triton.language.dot<6F>aj<61>]<5D>NjY<00><>aj<61>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhjLubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthj<>hhhj<>hNubeh}<7D>(h ]<5D><>linear-algebra-ops<70>ah"]<5D>h$]<5D><>linear algebra ops<70>ah&]<5D>h(]<5D>uh*h
2022-02-08 20:02:05 +00:00
hh hhhh+hK+ubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>
Memory Ops<70>h]<5D>h<16>
2022-04-20 00:43:07 +00:00
Memory Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjphjnhhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhjkhhhh+hK4ubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hjkhhh<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:63:<autosummary><3E>hNubhM)<29><>}<7D>(hX<>
2022-02-08 20:02:05 +00:00
load
Return a block of data whose values are, elementwise, loaded from memory at location defined by pointer.
store
Stores value block of elements in memory, element-wise, at the memory locations specified by pointer.
atomic_cas
Performs an atomic compare-and-swap at the memory location specified by pointer.
atomic_xchg
Performs an atomic exchange at the memory location specified by pointer.<2E>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj<>ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj<>ubhs)<29><>}<7D>(hhh]<5D>(hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>":obj:`load <triton.language.load>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>load<61><64><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.load<61>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:63:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>pReturn a block of data whose values are, elementwise, loaded from memory at location defined by :code:`pointer`.<2E>h]<5D>(h<16>`Return a block of data whose values are, elementwise, loaded from memory at location defined by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>`Return a block of data whose values are, elementwise, loaded from memory at location defined by <20>hj<>ubh<62>)<29><>}<7D>(h<05>:code:`pointer`<60>h]<5D>h<16>pointer<65><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>pointer<65>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:63:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>$:obj:`store <triton.language.store>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hjh]<5D>h<EFBFBD>)<29><>}<7D>(hjh]<5D>h<16>store<72><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j&<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.store<72>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:63:<autosummary><3E>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj8hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>uStores :code:`value` block of elements in memory, element-wise, at the memory locations specified by :code:`pointer`.<2E>h]<5D>(h<16>Stores <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>Stores <20>hjHubh<62>)<29><>}<7D>(h<05> :code:`value`<60>h]<5D>h<16>value<75><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>value<75>hjQubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjHubh<16>Q block of elements in memory, element-wise, at the memory locations specified by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>Q block of elements in memory, element-wise, at the memory locations specified by <20>hjHubh<62>)<29><>}<7D>(h<05>:code:`pointer`<60>h]<5D>h<16>pointer<65><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>pointer<65>hjeubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjHubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hjHubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:63:<autosummary><3E>hKhjEubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>.:obj:`atomic_cas <triton.language.atomic_cas>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>
atomic_cas<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.atomic_cas<61>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:63:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>XPerforms an atomic compare-and-swap at the memory location specified by :code:`pointer`.<2E>h]<5D>(h<16>HPerforms an atomic compare-and-swap at the memory location specified by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>HPerforms an atomic compare-and-swap at the memory location specified by <20>hj<>ubh<62>)<29><>}<7D>(h<05>:code:`pointer`<60>h]<5D>h<16>pointer<65><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>pointer<65>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:63:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>0:obj:`atomic_xchg <triton.language.atomic_xchg>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16> atomic_xchg<68><67><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j
<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.atomic_xchg<68>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:63:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjhKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>PPerforms an atomic exchange at the memory location specified by :code:`pointer`.<2E>h]<5D>(h<16>@Performs an atomic exchange at the memory location specified by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>@Performs an atomic exchange at the memory location specified by <20>hj,ubh<62>)<29><>}<7D>(h<05>:code:`pointer`<60>h]<5D>h<16>pointer<65><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>pointer<65>hj5ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj,ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj,ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj)ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj<>ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhjkhhhj<>hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D>(<28>)python-api/generated/triton.language.load<61><64>*python-api/generated/triton.language.store<72><65>/python-api/generated/triton.language.atomic_cas<61><73>0python-api/generated/triton.language.atomic_xchg<68>ej<65>]<5D>(Nj<4E><00><>Nj<4E><00><>Nj<4E><00><>Nj<4E><00><>ej<65>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhjsubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthjkhhhj<>hNubeh}<7D>(h ]<5D><>
2022-02-08 20:02:05 +00:00
memory-ops<70>ah"]<5D>h$]<5D><>
memory ops<70>ah&]<5D>h(]<5D>uh*h
2022-04-20 00:43:07 +00:00
hh hhhh+hK4ubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05> Indexing Ops<70>h]<5D>h<16> Indexing Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>hj<>hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj<>hhhh+hKAubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj<>hhh<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:73:<autosummary><3E>hNubhM)<29><>}<7D>(h<05>R
2022-02-08 20:02:05 +00:00
where
Returns a block of elements from either x or y, depending on condition.<2E>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj<>ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj<>ubhs)<29><>}<7D>(hhh]<5D>hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>$:obj:`where <triton.language.where>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>where<72><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.where<72>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:73:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>_Returns a block of elements from either :code:`x` or :code:`y`, depending on :code:`condition`.<2E>h]<5D>(h<16>(Returns a block of elements from either <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>(Returns a block of elements from either <20>hj ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhjubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj ubh<16> or <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> or <20>hj ubh<62>)<29><>}<7D>(h<05> :code:`y`<60>h]<5D>h<16>y<><79><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>y<>hj)ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj ubh<16>, depending on <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>, depending on <20>hj ubh<62>)<29><>}<7D>(h<05>:code:`condition`<60>h]<5D>h<16> condition<6F><6E><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> condition<6F>hj=ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj
2022-02-08 20:02:05 +00:00
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj<>ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhj<>hhhj<>hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D><>*python-api/generated/triton.language.where<72>aj<61>]<5D>Nj<4E><00><>aj<61>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhj{ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthj<>hhhj<>hNubeh}<7D>(h ]<5D><> indexing-ops<70>ah"]<5D>h$]<5D><> indexing ops<70>ah&]<5D>h(]<5D>uh*h
2022-04-20 00:43:07 +00:00
hh hhhh+hKAubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>Math Ops<70>h]<5D>h<16>Math Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>hj<>hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj<>hhhh+hKKubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj<>hhh<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hNubhM)<29><>}<7D>(hXT
2022-02-08 20:02:05 +00:00
exp
Computes the element-wise exponential of x
log
Computes the element-wise natural logarithm of x
cos
Computes the element-wise cosine of x
sin
Computes the element-wise sine of x
sqrt
Computes the element-wise square root of x
sigmoid
Computes the element-wise sigmoid of x
softmax
Computes the element-wise softmax of x<>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj<>ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj<>ubhs)<29><>}<7D>(hhh]<5D>(hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05> :obj:`exp <triton.language.exp>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>exp<78><70><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.exp<78>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>2Computes the element-wise exponential of :code:`x`<60>h]<5D>(h<16>)Computes the element-wise exponential of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>)Computes the element-wise exponential of <20>hj ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhj ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05> :obj:`log <triton.language.log>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hjA h]<5D>h<EFBFBD>)<29><>}<7D>(hjA h]<5D>h<16>log<6F><67><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjF ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hjC ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>jP <00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.log<6F>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj? ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjb hKhj< ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj9 ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>8Computes the element-wise natural logarithm of :code:`x`<60>h]<5D>(h<16>/Computes the element-wise natural logarithm of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>/Computes the element-wise natural logarithm of <20>hjr ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhj{ ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjr ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhjo ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj9 ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05> :obj:`cos <triton.language.cos>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<> h]<5D>h<EFBFBD>)<29><>}<7D>(hj<> h]<5D>h<16>cos<6F><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<> ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD> <00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.cos<6F>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> hKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>-Computes the element-wise cosine of :code:`x`<60>h]<5D>(h<16>$Computes the element-wise cosine of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>$Computes the element-wise cosine of <20>hj<> ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhj<> ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05> :obj:`sin <triton.language.sin>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj
2022-02-08 20:02:05 +00:00
h]<5D>h<EFBFBD>)<29><>}<7D>(hj
h]<5D>h<16>sin<69><6E><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj
ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j
2022-04-20 00:43:07 +00:00
<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.sin<69>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj
2022-02-08 20:02:05 +00:00
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj(
hKhj
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>+Computes the element-wise sine of :code:`x`<60>h]<5D>(h<16>"Computes the element-wise sine of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>"Computes the element-wise sine of <20>hj8
ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhjA
ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj8
2022-04-20 00:43:07 +00:00
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj5
2022-02-08 20:02:05 +00:00
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>":obj:`sqrt <triton.language.sqrt>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hjj
h]<5D>h<EFBFBD>)<29><>}<7D>(hjj
h]<5D>h<16>sqrt<72><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjo
ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hjl
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>jy
2022-04-20 00:43:07 +00:00
<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.sqrt<72>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhjh
2022-02-08 20:02:05 +00:00
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>
hKhje
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjb
ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>2Computes the element-wise square root of :code:`x`<60>h]<5D>(h<16>)Computes the element-wise square root of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>)Computes the element-wise square root of <20>hj<>
ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhj<>
ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>
2022-04-20 00:43:07 +00:00
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<>
2022-02-08 20:02:05 +00:00
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjb
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>(:obj:`sigmoid <triton.language.sigmoid>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>
h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>
h]<5D>h<16>sigmoid<69><64><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>
ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD>
2022-04-20 00:43:07 +00:00
<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.sigmoid<69>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<>
2022-02-08 20:02:05 +00:00
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>
hKhj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>
ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>.Computes the element-wise sigmoid of :code:`x`<60>h]<5D>(h<16>%Computes the element-wise sigmoid of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>%Computes the element-wise sigmoid of <20>hj<>
ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhj ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>
2022-04-20 00:43:07 +00:00
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<>
2022-02-08 20:02:05 +00:00
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>
2022-04-20 00:43:07 +00:00
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>(:obj:`softmax <triton.language.softmax>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj0 h]<5D>h<EFBFBD>)<29><>}<7D>(hj0 h]<5D>h<16>softmax<61><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj5 ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj2 ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j? <00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.softmax<61>uh*h<>h<1C>n/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj. ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjQ hKhj+ ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj( ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>.Computes the element-wise softmax of :code:`x`<60>h]<5D>(h<16>%Computes the element-wise softmax of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>%Computes the element-wise softmax of <20>hja ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhjj ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hja ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj^ ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj( ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj<>ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhj<>hhhj<>hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D>(<28>(python-api/generated/triton.language.exp<78><70>(python-api/generated/triton.language.log<6F><67>(python-api/generated/triton.language.cos<6F><73>(python-api/generated/triton.language.sin<69><6E>)python-api/generated/triton.language.sqrt<72><74>,python-api/generated/triton.language.sigmoid<69><64>,python-api/generated/triton.language.softmax<61>ej<65>]<5D>(Nj<4E> <00><>Nj<4E> <00><>Nj<4E> <00><>Nj<4E> <00><>Nj<4E> <00><>Nj<4E> <00><>Nj<4E> <00><>ej<65>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthj<>hhhj<>hNubeh}<7D>(h ]<5D><>math-ops<70>ah"]<5D>h$]<5D><>math ops<70>ah&]<5D>h(]<5D>uh*h
hh hhhh+hKKubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05> Reduction Ops<70>h]<5D>h<16> Reduction Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<> hj<> hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj<> hhhh+hK[ubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj<> hhh<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:100:<autosummary><3E>hNubhM)<29><>}<7D>(h<05><>
2022-02-08 20:02:05 +00:00
max
Returns the maximum of all elements in the input block along the provided axis
min
Returns the minimum of all elements in the input block along the provided axis
sum
Returns the sum of all elements in the input block along the provided axis<69>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj<> ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj<> ubhs)<29><>}<7D>(hhh]<5D>(hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05> :obj:`max <triton.language.max>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj h]<5D>h<EFBFBD>)<29><>}<7D>(hj h]<5D>h<16>max<61><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j! <00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.max<61>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:100:<autosummary><3E>hKhj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj3 hKhj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj
ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>^Returns the maximum of all elements in the :code:`input` block along the provided :code:`axis`<60>h]<5D>(h<16>+Returns the maximum of all elements in the <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>+Returns the maximum of all elements in the <20>hjC ubh<62>)<29><>}<7D>(h<05> :code:`input`<60>h]<5D>h<16>input<75><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>input<75>hjL ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjC ubh<16> block along the provided <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> block along the provided <20>hjC ubh<62>)<29><>}<7D>(h<05> :code:`axis`<60>h]<5D>h<16>axis<69><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>axis<69>hj` ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjC ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:100:<autosummary><3E>hKhj@ ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05> :obj:`min <triton.language.min>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<> h]<5D>h<EFBFBD>)<29><>}<7D>(hj<> h]<5D>h<16>min<69><6E><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<> ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD> <00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.min<69>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:100:<autosummary><3E>hKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> hKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>^Returns the minimum of all elements in the :code:`input` block along the provided :code:`axis`<60>h]<5D>(h<16>+Returns the minimum of all elements in the <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>+Returns the minimum of all elements in the <20>hj<> ubh<62>)<29><>}<7D>(h<05> :code:`input`<60>h]<5D>h<16>input<75><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>input<75>hj<> ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> ubh<16> block along the provided <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> block along the provided <20>hj<> ubh<62>)<29><>}<7D>(h<05> :code:`axis`<60>h]<5D>h<16>axis<69><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>axis<69>hj<> ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:100:<autosummary><3E>hKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05> :obj:`sum <triton.language.sum>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj h]<5D>h<EFBFBD>)<29><>}<7D>(hj h]<5D>h<16>sum<75><6D><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j <00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.sum<75>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:100:<autosummary><3E>hKhj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj# hKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>ZReturns the sum of all elements in the :code:`input` block along the provided :code:`axis`<60>h]<5D>(h<16>'Returns the sum of all elements in the <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>'Returns the sum of all elements in the <20>hj3 ubh<62>)<29><>}<7D>(h<05> :code:`input`<60>h]<5D>h<16>input<75><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>input<75>hj< ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj3 ubh<16> block along the provided <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> block along the provided <20>hj3 ubh<62>)<29><>}<7D>(h<05> :code:`axis`<60>h]<5D>h<16>axis<69><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>axis<69>hjP ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj3 ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> hKhj0 ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj<> ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhj<> hhhj<> hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D>(<28>(python-api/generated/triton.language.max<61><78>(python-api/generated/triton.language.min<69><6E>(python-api/generated/triton.language.sum<75>ej<65>]<5D>(Nj<4E> <00><>Nj<4E> <00><>Nj<4E> <00><>ej<65>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthj<> hhhj<> hNubeh}<7D>(h ]<5D><> reduction-ops<70>ah"]<5D>h$]<5D><> reduction ops<70>ah&]<5D>h(]<5D>uh*h
2022-02-08 20:02:05 +00:00
hh hhhh+hK[ubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>
Atomic Ops<70>h]<5D>h<16>
2022-04-20 00:43:07 +00:00
Atomic Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<> hj<> hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj<> hhhh+hKfubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj<> hhh<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:113:<autosummary><3E>hNubhM)<29><>}<7D>(hXS
2022-02-08 20:02:05 +00:00
atomic_cas
Performs an atomic compare-and-swap at the memory location specified by pointer.
atomic_add
Performs an atomic add at the memory location specified by pointer.
atomic_max
Performs an atomic max at the memory location specified by pointer.
atomic_min
Performs an atomic min at the memory location specified by pointer.<2E>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
uh*h\hj<> ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj<> ubhs)<29><>}<7D>(hhh]<5D>(hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>.:obj:`atomic_cas <triton.language.atomic_cas>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<> h]<5D>h<EFBFBD>)<29><>}<7D>(hj<> h]<5D>h<16>
2022-04-20 00:43:07 +00:00
atomic_cas<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<> ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.atomic_cas<61>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:113:<autosummary><3E>hKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjhKhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>XPerforms an atomic compare-and-swap at the memory location specified by :code:`pointer`.<2E>h]<5D>(h<16>HPerforms an atomic compare-and-swap at the memory location specified by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>HPerforms an atomic compare-and-swap at the memory location specified by <20>hj"ubh<62>)<29><>}<7D>(h<05>:code:`pointer`<60>h]<5D>h<16>pointer<65><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>pointer<65>hj+ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj"ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj"ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:113:<autosummary><3E>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<> ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>.:obj:`atomic_add <triton.language.atomic_add>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hjYh]<5D>h<EFBFBD>)<29><>}<7D>(hjYh]<5D>h<16>
atomic_add<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj^ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj[ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>jh<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.atomic_add<64>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:113:<autosummary><3E>hKhjWubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjzhKhjTubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjQubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>KPerforms an atomic add at the memory location specified by :code:`pointer`.<2E>h]<5D>(h<16>;Performs an atomic add at the memory location specified by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>;Performs an atomic add at the memory location specified by <20>hj<>ubh<62>)<29><>}<7D>(h<05>:code:`pointer`<60>h]<5D>h<16>pointer<65><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>pointer<65>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:113:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjQubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<> ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>.:obj:`atomic_max <triton.language.atomic_max>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>
atomic_max<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.atomic_max<61>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:113:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>KPerforms an atomic max at the memory location specified by :code:`pointer`.<2E>h]<5D>(h<16>;Performs an atomic max at the memory location specified by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>;Performs an atomic max at the memory location specified by <20>hj<>ubh<62>)<29><>}<7D>(h<05>:code:`pointer`<60>h]<5D>h<16>pointer<65><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>pointer<65>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:113:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<> ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>.:obj:`atomic_min <triton.language.atomic_min>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj)h]<5D>h<EFBFBD>)<29><>}<7D>(hj)h]<5D>h<16>
atomic_min<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj.ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj+ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j8<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.atomic_min<69>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:113:<autosummary><3E>hKhj'ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjJhKhj$ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj!ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>KPerforms an atomic min at the memory location specified by :code:`pointer`.<2E>h]<5D>(h<16>;Performs an atomic min at the memory location specified by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>;Performs an atomic min at the memory location specified by <20>hjZubh<62>)<29><>}<7D>(h<05>:code:`pointer`<60>h]<5D>h<16>pointer<65><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>pointer<65>hjcubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjZubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hjZubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<> hKhjWubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj!ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj<> ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj<> ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj<> ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhj<> hhhj<> hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D>(<28>/python-api/generated/triton.language.atomic_cas<61><73>/python-api/generated/triton.language.atomic_add<64><64>/python-api/generated/triton.language.atomic_max<61><78>/python-api/generated/triton.language.atomic_min<69>ej<65>]<5D>(Nj<4E><00><>Nj<4E><00><>Nj<4E><00><>Nj<4E><00><>ej<65>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthj<> hhhj<> hNubeh}<7D>(h ]<5D><>
2022-02-08 20:02:05 +00:00
atomic-ops<70>ah"]<5D>h$]<5D><>
atomic ops<70>ah&]<5D>h(]<5D>uh*h
2022-04-20 00:43:07 +00:00
hh hhhh+hKfubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>Comparison ops<70>h]<5D>h<16>Comparison ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>hj<>hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj<>hhhh+hKsubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj<>hhh<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:123:<autosummary><3E>hNubhM)<29><>}<7D>(h<05>r
2022-02-08 20:02:05 +00:00
minimum
Computes the element-wise minimum of x and y.
maximum
Computes the element-wise maximum of x and y.<2E>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
uh*h\hj<>ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj<>ubhs)<29><>}<7D>(hhh]<5D>(hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>(:obj:`minimum <triton.language.minimum>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj
h]<5D>h<EFBFBD>)<29><>}<7D>(hj
2022-04-20 00:43:07 +00:00
h]<5D>h<16>minimum<75><6D><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.minimum<75>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:123:<autosummary><3E>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj+hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>=Computes the element-wise minimum of :code:`x` and :code:`y`.<2E>h]<5D>(h<16>%Computes the element-wise minimum of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>%Computes the element-wise minimum of <20>hj;ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhjDubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj;ubh<16> and <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> and <20>hj;ubh<62>)<29><>}<7D>(h<05> :code:`y`<60>h]<5D>h<16>y<><79><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj1hjWubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj;ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj;ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:123:<autosummary><3E>hKhj8ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>(:obj:`maximum <triton.language.maximum>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>maximum<75><6D><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.maximum<75>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:123:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj|ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>=Computes the element-wise maximum of :code:`x` and :code:`y`.<2E>h]<5D>(h<16>%Computes the element-wise maximum of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>%Computes the element-wise maximum of <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`x`<60>h]<5D>h<16>x<><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjOhj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16> and <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> and <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`y`<60>h]<5D>h<16>y<><79><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj1hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj|ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj<>ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhj<>hhhj<>hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D>(<28>,python-api/generated/triton.language.minimum<75><6D>,python-api/generated/triton.language.maximum<75>ej<65>]<5D>(Nj<00><>Nj<00><>ej<65>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthj<>hhhj<>hNubh <09>target<65><74><EFBFBD>)<29><>}<7D>(h<05>.. _Random Number Generation:<3A>h]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refid<69><64>random-number-generation<6F>uh*j'hK|hj<>hhhh+ubeh}<7D>(h ]<5D><>comparison-ops<70>ah"]<5D>h$]<5D><>comparison ops<70>ah&]<5D>h(]<5D>uh*h
hh hhhh+hKsubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>Random Number Generation<6F>h]<5D>h<16>Random Number Generation<6F><6E><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjBhj@hhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhj=hhhh+hKubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hj=hhh<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:137:<autosummary><3E>hNubhM)<29><>}<7D>(hX|
2022-02-08 20:02:05 +00:00
randint4x
Given a seed scalar and an offset block, returns four blocks of random int32.
randint
Given a seed scalar and an offset block, returns a single block of random int32.
rand
Given a seed scalar and an offset block, returns a block of random float32 in U(0, 1)
randn
Given a seed scalar and an offset block, returns a block of random float32 in \mathcal{N}(0, 1)<29>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj_ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj_ubhs)<29><>}<7D>(hhh]<5D>(hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>,:obj:`randint4x <triton.language.randint4x>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16> randint4x<34><78><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.randint4x<34>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:137:<autosummary><3E>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj|ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjyubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>eGiven a :code:`seed` scalar and an :code:`offset` block, returns four blocks of random :code:`int32`.<2E>h]<5D>(h<16>Given a <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>Given a <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`seed`<60>h]<5D>h<16>seed<65><64><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>seed<65>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16> scalar and an <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> scalar and an <20>hj<>ubh<62>)<29><>}<7D>(h<05>:code:`offset`<60>h]<5D>h<16>offset<65><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>offset<65>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>& block, returns four blocks of random <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>& block, returns four blocks of random <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`int32`<60>h]<5D>h<16>int32<33><32><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>int32<33>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:137:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjyubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhjvubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>(:obj:`randint <triton.language.randint>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hjh]<5D>h<EFBFBD>)<29><>}<7D>(hjh]<5D>h<16>randint<6E><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j <00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.randint<6E>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:137:<autosummary><3E>hKhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj2hKhj ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>hGiven a :code:`seed` scalar and an :code:`offset` block, returns a single block of random :code:`int32`.<2E>h]<5D>(h<16>Given a <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>Given a <20>hjBubh<62>)<29><>}<7D>(h<05> :code:`seed`<60>h]<5D>h<16>seed<65><64><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>seed<65>hjKubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjBubh<16> scalar and an <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> scalar and an <20>hjBubh<62>)<29><>}<7D>(h<05>:code:`offset`<60>h]<5D>h<16>offset<65><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>offset<65>hj_ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjBubh<16>) block, returns a single block of random <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>) block, returns a single block of random <20>hjBubh<62>)<29><>}<7D>(h<05> :code:`int32`<60>h]<5D>h<16>int32<33><32><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>int32<33>hjsubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hjBubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hjBubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:137:<autosummary><3E>hKhj?ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhjvubhx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>":obj:`rand <triton.language.rand>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>h]<5D>h<16>rand<6E><64><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>j<EFBFBD><00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.rand<6E>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:137:<autosummary><3E>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>ubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>uGiven a :code:`seed` scalar and an :code:`offset` block, returns a block of random :code:`float32` in :math:`U(0, 1)`<60>h]<5D>(h<16>Given a <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>Given a <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`seed`<60>h]<5D>h<16>seed<65><64><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>seed<65>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16> scalar and an <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> scalar and an <20>hj<>ubh<62>)<29><>}<7D>(h<05>:code:`offset`<60>h]<5D>h<16>offset<65><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>offse
2022-02-08 20:02:05 +00:00
hh hhhh+hK<4B>
2022-04-20 00:43:07 +00:00
referenced<EFBFBD>K<01>expect_referenced_by_name<6D>}<7D>jj)s<>expect_referenced_by_id<69>}<7D>j4j)subh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>Compiler Hint Ops<70>h]<5D>h<16>Compiler Hint Ops<70><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj!hjhhhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hhjhhhh+hK<>ubh>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hHhIuh*h=hjhhh<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:145:<autosummary><3E>hNubhM)<29><>}<7D>(h<05>\
2022-02-08 20:02:05 +00:00
multiple_of
Let the compiler knows that the values in input are all multiples of value.<2E>h]<5D>hS)<29><>}<7D>(hhh]<5D>hX)<29><>}<7D>(hhh]<5D>(h])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>K
2022-04-20 00:43:07 +00:00
uh*h\hj>ubh])<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>colwidth<74>KZuh*h\hj>ubhs)<29><>}<7D>(hhh]<5D>hx)<29><>}<7D>(hhh]<5D>(h})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>0:obj:`multiple_of <triton.language.multiple_of>`<60>h]<5D>h<EFBFBD>)<29><>}<7D>(hj`h]<5D>h<EFBFBD>)<29><>}<7D>(hj`h]<5D>h<16> multiple_of<6F><66><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjeubah}<7D>(h ]<5D>h"]<5D>(h<><68>py<70><79>py-obj<62>eh$]<5D>h&]<5D>h(]<5D>uh*h<>hjbubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refdoc<6F>h<EFBFBD><68> refdomain<69>jo<00>reftype<70><65>obj<62><6A> refexplicit<69><74><EFBFBD>refwarn<72><6E>h<EFBFBD>h<EFBFBD>h<EFBFBD>Nh<4E><68>triton.language.multiple_of<6F>uh*h<>h<1C>o/tmp/tmplzzj8p5m/2d6df9b518a8152f777eb79b6b0a84becb706353/docs/python-api/triton.language.rst:145:<autosummary><3E>hKhj^ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>hKhj[ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjXubh})<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>[Let the compiler knows that the values in :code:`input` are all multiples of :code:`value`.<2E>h]<5D>(h<16>*Let the compiler knows that the values in <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>*Let the compiler knows that the values in <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`input`<60>h]<5D>h<16>input<75><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>input<75>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16> are all multiples of <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> are all multiples of <20>hj<>ubh<62>)<29><>}<7D>(h<05> :code:`value`<60>h]<5D>h<16>value<75><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>value<75>hj<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubh<16>.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hj<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj6hKhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hjXubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hwhjUubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*hrhj>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>cols<6C>Kuh*hWhj;ubah}<7D>(h ]<5D>h"]<5D>jhah$]<5D>h&]<5D>h(]<5D>uh*hRhj7ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*hLhjhhhj6hNubju)<29><>}<7D>(hhh]<5D>jz)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>j<EFBFBD>]<5D><>0python-api/generated/triton.language.multiple_of<6F>aj<61>]<5D>Nj<4E><00><>aj<61>J<><4A><EFBFBD><EFBFBD>j<EFBFBD>Nj<4E>Nj<4E>]<5D>uh*jyhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>jrjsuh*jthjhhhj6hNubeh}<7D>(h ]<5D><>compiler-hint-ops<70>ah"]<5D>h$]<5D><>compiler hint ops<70>ah&]<5D>h(]<5D>uh*h
2022-02-08 20:02:05 +00:00
hh hhhh+hK<>ubeh}<7D>(h ]<5D><>triton-language<67>ah"]<5D>h$]<5D><>triton.language<67>ah&]<5D>h(]<5D>uh*h
hhhhhh+hKubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>source<63>h+uh*h<01>current_source<63>N<EFBFBD> current_line<6E>N<EFBFBD>settings<67><73>docutils.frontend<6E><64>Values<65><73><EFBFBD>)<29><>}<7D>(hN<> generator<6F>N<EFBFBD> datestamp<6D>N<EFBFBD> source_link<6E>N<EFBFBD>
source_url<EFBFBD>N<EFBFBD> toc_backlinks<6B>h|<7C>footnote_backlinks<6B>K<01> sectnum_xform<72>K<01>strip_comments<74>N<EFBFBD>strip_elements_with_classes<65>N<EFBFBD> strip_classes<65>N<EFBFBD> report_level<65>K<02>
halt_level<EFBFBD>K<05>exit_status_level<65>K<05>debug<75>N<EFBFBD>warning_stream<61>N<EFBFBD> traceback<63><6B><EFBFBD>input_encoding<6E><67> utf-8-sig<69><67>input_encoding_error_handler<65><72>strict<63><74>output_encoding<6E><67>utf-8<><38>output_encoding_error_handler<65>j5<00>error_encoding<6E><67>utf-8<><38>error_encoding_error_handler<65><72>backslashreplace<63><65> language_code<64><65>en<65><6E>record_dependencies<65>N<EFBFBD>config<69>N<EFBFBD> id_prefix<69>h<06>auto_id_prefix<69><78>id<69><64> dump_settings<67>N<EFBFBD>dump_internals<6C>N<EFBFBD>dump_transforms<6D>N<EFBFBD>dump_pseudo_xml<6D>N<EFBFBD>expose_internals<6C>N<EFBFBD>strict_visitor<6F>N<EFBFBD>_disable_config<69>N<EFBFBD>_source<63>h+<2B> _destination<6F>N<EFBFBD> _config_files<65>]<5D><>pep_references<65>N<EFBFBD> pep_base_url<72><6C> https://www.python.org/dev/peps/<2F><>pep_file_url_template<74><65>pep-%04d<34><64>rfc_references<65>N<EFBFBD> rfc_base_url<72><6C>https://tools.ietf.org/html/<2F><> tab_width<74>K<08>trim_footnote_reference_space<63><65><EFBFBD>file_insertion_enabled<65><64><EFBFBD> raw_enabled<65>K<01>syntax_highlight<68><74>long<6E><67> smart_quotes<65><73><EFBFBD>smartquotes_locales<65>]<5D><>character_level_inline_markup<75><70><EFBFBD>doctitle_xform<72><6D><EFBFBD> docinfo_xform<72>K<01>sectsubtitle_xform<72><6D><EFBFBD>embed_stylesheet<65><74><EFBFBD>cloak_email_addresses<65><73><EFBFBD>env<6E>Nub<75>reporter<65>N<EFBFBD>indirect_targets<74>]<5D><>substitution_defs<66>}<7D><>substitution_names<65>}<7D><>refnames<65>}<7D><>refids<64>}<7D>j4]<5D>j)as<61>nameids<64>}<7D>(jj j<>j<>j
jj<>j<>jhjej<>j<>j<>j<>j<> j<> j<> j<> j<>j<>j:j7jj4jju<> nametypes<65>}<7D>(jNj<4E>Nj
Nj<4E>NjhNj<4E>Nj<4E>Nj<4E> Nj<4E> Nj<4E>Nj:Nj<00>jNuh }<7D>(j h j<>h,jj<>j<>j jej<>j<>jkj<>j<>j<> j<>j<> j<> j<>j<> j7j<>j4j=jj=jju<> footnote_refs<66>}<7D><> citation_refs<66>}<7D><> autofootnotes<65>]<5D><>autofootnote_refs<66>]<5D><>symbol_footnotes<65>]<5D><>symbol_footnote_refs<66>]<5D><> footnotes<65>]<5D><> citations<6E>]<5D><>autofootnote_start<72>K<01>symbol_footnote_start<72>K<00>
id_counter<EFBFBD><EFBFBD> collections<6E><73>Counter<65><72><EFBFBD>}<7D>jCKs<><73>R<EFBFBD><52>parse_messages<65>]<5D>h <09>system_message<67><65><EFBFBD>)<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>;Duplicate implicit target name: "random number generation".<2E>h]<5D>h<16>?Duplicate implicit target name: “random number generation”.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>ja<>level<65>K<01>type<70><65>INFO<46><4F>source<63>h+<2B>line<6E>Kuh*j<>hj=hhhh+hKuba<62>transform_messages<65>]<5D>j<EFBFBD>)<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(hhh]<5D>h<16>>Hyperlink target "random-number-generation" is not referenced.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>level<65>K<01>type<70>j<EFBFBD><00>source<63>h+<2B>line<6E>K|uh*j<>uba<62> transformer<65>N<EFBFBD>
decoration<EFBFBD>Nhhub.