triton/master/.doctrees/python-api/generated/triton.jit.doctree

<EFBFBD><05><><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.jit<69>h]<5D>h	<09>Text<78><74><EFBFBD><EFBFBD>
triton.jit<69><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<13>parent<6E>hh
h<03>source<63>N<EFBFBD>line<6E>Nuba<62>
attributes<EFBFBD>}<7D>(<28>ids<64>]<5D><>classes<65>]<5D><>names<65>]<5D><>dupnames<65>]<5D><>backrefs<66>]<5D>u<EFBFBD>tagname<6D>hhhh
hh<1C>b/tmp/tmpnr15x0cb/7394d732adcec5a67034926a28080af2a7853217/docs/python-api/generated/triton.jit.rst<73>hKubh<00>index<65><78><EFBFBD>)<29><>}<7D>(hhh]<5D>h}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>entries<65>]<5D>(<28>single<6C><65>jit() (in module triton)<29><>
triton.jit<69>hNt<4E>auh*h,hhh
hhNhNubh<00>desc<73><63><EFBFBD>)<29><>}<7D>(hhh]<5D>(h<00>desc_signature<72><65><EFBFBD>)<29><>}<7D>(h<05>jit(*args, **kwargs)<29>h]<5D>(h<00>desc_addname<6D><65><EFBFBD>)<29><>}<7D>(h<05>triton.<2E>h]<5D>h<16>triton.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhhJh
hhNhNubah}<7D>(h ]<5D>h"]<5D>(<28>sig-prename<6D><65>
descclassname<6D>eh$]<5D>h&]<5D>h(]<5D><>	xml:space<63><65>preserve<76>uh*hHhhDh
hh<1C>t/tmp/tmpnr15x0cb/7394d732adcec5a67034926a28080af2a7853217/python/triton/code_gen.py:docstring of triton.code_gen.jit<69>hNubh<00>	desc_name<6D><65><EFBFBD>)<29><>}<7D>(h<05>jit<69>h]<5D>h<16>jit<69><74><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhh_h
hhNhNubah}<7D>(h ]<5D>h"]<5D>(<28>sig-name<6D><65>descname<6D>eh$]<5D>h&]<5D>h(]<5D>hZh[uh*h]hhDh
hhh\hNubh<00>desc_parameterlist<73><74><EFBFBD>)<29><>}<7D>(h<05>*args, **kwargs<67>h]<5D>(h<00>desc_parameter<65><72><EFBFBD>)<29><>}<7D>(h<05>*args<67>h]<5D>(h<00>desc_sig_operator<6F><72><EFBFBD>)<29><>}<7D>(h<05>
*<2A>h]<5D>h<16>
*<2A><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhh}ubah}<7D>(h ]<5D>h"]<5D><>
o<>ah$]<5D>h&]<5D>h(]<5D>uh*h{hhwubh<00>
desc_sig_name<6D><65><EFBFBD>)<29><>}<7D>(h<05>args<67>h]<5D>h<16>args<67><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhh<>ubah}<7D>(h ]<5D>h"]<5D><>
n<>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hhwubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hZh[uh*huhhqubhv)<29><>}<7D>(h<05>**kwargs<67>h]<5D>(h|)<29><>}<7D>(h<05>**<2A>h]<5D>h<16>**<2A><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhh<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h{hh<>ubh<62>)<29><>}<7D>(h<05>kwargs<67>h]<5D>h<16>kwargs<67><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhh<>ubah}<7D>(h ]<5D>h"]<5D>h<EFBFBD>ah$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hZh[uh*huhhqubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>hZh[uh*hohhDh
hhh\hNubeh}<7D>(h ]<5D>h;ah"]<5D>(<28>sig<69><67>
sig-object<63>eh$]<5D>h&]<5D>h(]<5D><>module<6C><65>triton<6F><6E>class<73>h<06>fullname<6D>hauh*hBhh\hK
hh?h
hubh<00>desc_content<6E><74><EFBFBD>)<29><>}<7D>(hhh]<5D>(h	<09>	paragraph<70><68><EFBFBD>)<29><>}<7D>(h<05>ADecorator for JIT-compiling a function using the Triton compiler.<2E>h]<5D>h<16>ADecorator for JIT-compiling a function using the Triton compiler.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hh<>hh<>h
hhNhNubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>h<1C>t/tmp/tmpnr15x0cb/7394d732adcec5a67034926a28080af2a7853217/python/triton/code_gen.py:docstring of triton.code_gen.jit<69>hK
hh<>h
hubh	<09>
field_list<EFBFBD><EFBFBD><EFBFBD>)<29><>}<7D>(hhh]<5D>(h	<09>field<6C><64><EFBFBD>)<29><>}<7D>(hhh]<5D>(h	<09>
field_name<EFBFBD><EFBFBD><EFBFBD>)<29><>}<7D>(h<05>note<74>h]<5D>h<16>Note<74><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhh<>ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hh<>hKubh	<09>
field_body<EFBFBD><EFBFBD><EFBFBD>)<29><>}<7D>(h<05><>When a jit'd function is called, :code:`torch.tensor` arguments are implicitly converted to pointers using the :code:`.data_ptr()` method.
<EFBFBD>h]<5D>h<EFBFBD>)<29><>}<7D>(h<05><>When a jit'd function is called, :code:`torch.tensor` arguments are implicitly converted to pointers using the :code:`.data_ptr()` method.<2E>h]<5D>(h<16>#When a jit’d function is called, <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>!When a jit'd function is called, <20>hj
ubh	<09>literal<61><6C><EFBFBD>)<29><>}<7D>(h<05>:code:`torch.tensor`<60>h]<5D>h<16>torch.tensor<6F><72><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>torch.tensor<6F>hj
ubah}<7D>(h ]<5D>h"]<5D><>code<64>ah$]<5D>h&]<5D>h(]<5D>uh*j
hj
ubh<16>: arguments are implicitly converted to pointers using the <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>: arguments are implicitly converted to pointers using the <20>hj
ubj
)<29><>}<7D>(h<05>:code:`.data_ptr()`<60>h]<5D>h<16>.data_ptr()<29><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>.data_ptr()<29>hj1
ubah}<7D>(h ]<5D>h"]<5D>j(
ah$]<5D>h&]<5D>h(]<5D>uh*j
hj
ubh<16> method.<2E><><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05> method.<2E>hj
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hKhj

ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*j
hh<>ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hKhh<>h
hubh<62>)<29><>}<7D>(hhh]<5D>(h<>)<29><>}<7D>(h<05>note<74>h]<5D>h<16>Note<74><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjZ
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjW
hh<>hKubj
)<29><>}<7D>(h<05><>This function will be compiled and run on the GPU. It will only have access to:

* python primitives,
* objects within the triton.language package,
* arguments to this function,
* other jit'd functions
<EFBFBD>h]<5D>(h<>)<29><>}<7D>(h<05>OThis function will be compiled and run on the GPU. It will only have access to:<3A>h]<5D>h<16>OThis function will be compiled and run on the GPU. It will only have access to:<3A><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hjn
hjl
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hKhjh
ubh	<09>bullet_list<73><74><EFBFBD>)<29><>}<7D>(hhh]<5D>(h	<09>	list_item<65><6D><EFBFBD>)<29><>}<7D>(h<05>python primitives,<2C>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>
h]<5D>h<16>python primitives,<2C><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>
hj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hKhj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*j
hj|
ubj<62>
)<29><>}<7D>(h<05>+objects within the triton.language package,<2C>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>
h]<5D>h<16>+objects within the triton.language package,<2C><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>
hj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hKhj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*j
hj|
ubj<62>
)<29><>}<7D>(h<05>arguments to this function,<2C>h]<5D>h<EFBFBD>)<29><>}<7D>(hj<>
h]<5D>h<16>arguments to this function,<2C><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>
hj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hK	hj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*j
hj|
ubj<62>
)<29><>}<7D>(h<05>other jit'd functions
<EFBFBD>h]<5D>h<EFBFBD>)<29><>}<7D>(h<05>other jit'd functions<6E>h]<5D>h<16>other jit’d functions<6E><73><EFBFBD><EFBFBD><EFBFBD>}<7D>(hj<>
hj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hK
hj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*j
hj|
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>bullet<65><74>
*<2A>uh*jz
hh<>hKhjh
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*j
hjW
ubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh<>hKhh<>h
hubh<62>)<29><>}<7D>(hhh]<5D>(h<>)<29><>}<7D>(h<05>
Parameters<EFBFBD>h]<5D>h<16>
Parameters<EFBFBD><EFBFBD><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj<>
ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hj<>
hh\hKubj
)<29><>}<7D>(hhh]<5D>h<EFBFBD>)<29><>}<7D>(h<05>0fn (Callable) -- the function to be jit-compiled<65>h]<5D>(h<00>literal_strong<6E><67><EFBFBD>)<29><>}<7D>(h<05>fn<66>h]<5D>h<16>fn<66><6E><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>refspecific<69><63><EFBFBD>	py:module<6C>h،py:class<73>Nuh*j
hjubh<16> (<28><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubh<00>pending_xref<65><66><EFBFBD>)<29><>}<7D>(hhh]<5D>h<00>literal_emphasis<69><73><EFBFBD>)<29><>}<7D>(h<05>Callable<6C>h]<5D>h<16>Callable<6C><65><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhj(ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*j&hj#ubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D><>	refdomain<69><6E>py<70><79>refexplicit<69><74><EFBFBD>reftype<70>hٌ	reftarget<65>j*j<00>jh<>jNuh*j!hjubh<16>
)<29><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubh<16> – <20><><EFBFBD><EFBFBD><EFBFBD>}<7D>(hhhjubh<16>the function to be jit-compiled<65><64><EFBFBD><EFBFBD><EFBFBD>}<7D>(h<05>the function to be jit-compiled<65>hjh
hhNhNubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hjubah}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*j
hj<>
ubeh}<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*h<>hh<>h
hhNhNubeh}<7D>(h ]<5D>h"]<5D>h$]<5D>h&]<5D>h(]<5D>uh*h<>hh?h
hhh\hNubeh}<7D>(h ]<5D>h"]<5D>(j=<00>function<6F>eh$]<5D>h&]<5D>h(]<5D><>domain<69>j=<00>objtype<70>jo<00>desctype<70>jo<00>noindex<65><78>uh*h=h
hhhhNhNubeh}<7D>(h ]<5D><>
triton-jit<69>ah"]<5D>h$]<5D><>
triton.jit<69>ah&]<5D>h(]<5D>uh*h
hhh
hhh+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><73>entry<72><79>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>j<EFBFBD><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><>nameids<64>}<7D>j|jys<>	nametypes<65>}<7D>j|Nsh }<7D>(jyhh;hDu<44>
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><><EFBFBD>R<EFBFBD><52>parse_messages<65>]<5D><>transform_messages<65>]<5D><>transformer<65>N<EFBFBD>
decoration<EFBFBD>Nh
hub.