Files
triton/master/.doctrees/python-api/triton.language.doctree

254 lines
60 KiB
Plaintext
Raw Normal View History

2022-04-08 00:44:05 +00:00
<EFBFBD><05><><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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:28:<autosummary><3E>hNubhM)<29><>}<7D>(h<05><>
2022-02-09 03:17:47 +00:00
arange
Returns contiguous values within the open interval [start, end).
zeros
2022-04-08 00:44:05 +00:00
Returns a tensor 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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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 tensor filled with the scalar value 0 for the given :code:`shape` and :code:`dtype`.<2E>h]<5D>(h<16>>Returns a tensor filled with the scalar value 0 for the given <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>>Returns a tensor 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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:41:<autosummary><3E>hNubhM)<29><>}<7D>(h<05><>
2022-02-09 03:17:47 +00:00
broadcast_to
2022-04-08 00:44:05 +00:00
Tries to broadcast the given tensor to a new shape.
2022-02-09 03:17:47 +00:00
reshape
2022-04-08 00:44:05 +00:00
Tries to reshape the given tensor to a new shape.
2022-02-09 03:17:47 +00:00
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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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 tensor to a new :code:`shape`.<2E>h]<5D>(h<16>-Tries to broadcast the given tensor to a new <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>-Tries to broadcast the given tensor 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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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>1Tries to reshape the given tensor to a new shape.<2E>h]<5D>h<16>1Tries to reshape the given tensor 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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:50:<autosummary><3E>hNubhM)<29><>}<7D>(h<05>2
2022-02-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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-09 03:17:47 +00:00
hh hhhh+hK+ubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>
Memory Ops<70>h]<5D>h<16>
2022-04-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:63:<autosummary><3E>hNubhM)<29><>}<7D>(hX<>
2022-02-09 03:17:47 +00:00
load
2022-04-08 00:44:05 +00:00
Return a tensor of data whose values are, elementwise, loaded from memory at location defined by pointer.
2022-02-09 03:17:47 +00:00
store
2022-04-08 00:44:05 +00:00
Stores value tensor of elements in memory, element-wise, at the memory locations specified by pointer.
2022-02-09 03:17:47 +00:00
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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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>qReturn a tensor of data whose values are, elementwise, loaded from memory at location defined by :code:`pointer`.<2E>h]<5D>(h<16>aReturn a tensor of data whose values are, elementwise, loaded from memory at location defined by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>aReturn a tensor 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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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>vStores :code:`value` tensor 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>R tensor of elements in memory, element-wise, at the memory locations specified by <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>R tensor 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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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-09 03:17:47 +00:00
memory-ops<70>ah"]<5D>h$]<5D><>
memory ops<70>ah&]<5D>h(]<5D>uh*h
2022-04-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:73:<autosummary><3E>hNubhM)<29><>}<7D>(h<05>S
2022-02-09 03:17:47 +00:00
where
2022-04-08 00:44:05 +00:00
Returns a tensor 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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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 tensor of elements from either :code:`x` or :code:`y`, depending on :code:`condition`.<2E>h]<5D>(h<16>)Returns a tensor of elements from either <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>)Returns a tensor 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-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:89:<autosummary><3E>hNubhM)<29><>}<7D>(hXT
2022-02-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj
2022-02-09 03:17:47 +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-13 00:45:31 +00:00
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj5
2022-02-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhjh
2022-02-09 03:17:47 +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-13 00:45:31 +00:00
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<>
2022-02-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<>
2022-02-09 03:17:47 +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-13 00:45:31 +00:00
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>n/tmp/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:89:<autosummary><3E>hKhj<>
2022-02-09 03:17:47 +00:00
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h|hj<>
2022-04-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:100:<autosummary><3E>hNubhM)<29><>}<7D>(hX
2022-02-09 03:17:47 +00:00
max
2022-04-08 00:44:05 +00:00
Returns the maximum of all elements in the input tensor along the provided axis
2022-02-09 03:17:47 +00:00
min
2022-04-08 00:44:05 +00:00
Returns the minimum of all elements in the input tensor along the provided axis
2022-02-09 03:17:47 +00:00
sum
2022-04-08 00:44:05 +00:00
Returns the sum of all elements in the input tensor 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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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` tensor 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> tensor along the provided <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> tensor 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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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` tensor 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> tensor along the provided <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> tensor 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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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 sum of all elements in the :code:`input` tensor 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> tensor along the provided <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> tensor 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-09 03:17:47 +00:00
hh hhhh+hK[ubh )<29><>}<7D>(hhh]<5D>(h)<29><>}<7D>(h<05>
Atomic Ops<70>h]<5D>h<16>
2022-04-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:113:<autosummary><3E>hNubhM)<29><>}<7D>(hXS
2022-02-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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-09 03:17:47 +00:00
atomic-ops<70>ah"]<5D>h$]<5D><>
atomic ops<70>ah&]<5D>h(]<5D>uh*h
2022-04-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:123:<autosummary><3E>hNubhM)<29><>}<7D>(h<05>r
2022-02-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:137:<autosummary><3E>hNubhM)<29><>}<7D>(hX|
2022-02-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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-09 03:17:47 +00:00
hh hhhh+hK<4B>
2022-04-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/docs/python-api/triton.language.rst:145:<autosummary><3E>hNubhM)<29><>}<7D>(h<05>\
2022-02-09 03:17:47 +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-13 00:45:31 +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/tmpd9zaamfp/76bfac9f15ebc3e3cadcacefce71918b12928407/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-09 03:17:47 +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.