2021-09-10 22:58:16 +00:00
<!DOCTYPE html>
< html class = "writer-html5" lang = "en" >
< head >
< meta charset = "utf-8" / >
< meta name = "viewport" content = "width=device-width, initial-scale=1.0" / >
< title > triton.jit — Triton documentation< / title >
< link rel = "stylesheet" href = "../../_static/css/theme.css" type = "text/css" / >
< link rel = "stylesheet" href = "../../_static/pygments.css" type = "text/css" / >
< link rel = "stylesheet" href = "../../_static/pygments.css" type = "text/css" / >
< link rel = "stylesheet" href = "../../_static/css/theme.css" type = "text/css" / >
< link rel = "stylesheet" href = "../../_static/css/custom.css" type = "text/css" / >
<!-- [if lt IE 9]>
< script src = "../../_static/js/html5shiv.min.js" > < / script >
<![endif]-->
< script type = "text/javascript" id = "documentation_options" data-url_root = "../../" src = "../../_static/documentation_options.js" > < / script >
< script data-url_root = "../../" id = "documentation_options" src = "../../_static/documentation_options.js" > < / script >
< script src = "../../_static/jquery.js" > < / script >
< script src = "../../_static/underscore.js" > < / script >
< script src = "../../_static/doctools.js" > < / script >
< script type = "text/javascript" src = "../../_static/js/theme.js" > < / script >
< link rel = "index" title = "Index" href = "../../genindex.html" / >
< link rel = "search" title = "Search" href = "../../search.html" / >
< link rel = "next" title = "triton.autotune" href = "triton.autotune.html" / >
< link rel = "prev" title = "triton" href = "../triton.html" / >
< / head >
< body class = "wy-body-for-nav" >
< div class = "wy-grid-for-nav" >
< nav data-toggle = "wy-nav-shift" class = "wy-nav-side" >
< div class = "wy-side-scroll" >
< div class = "wy-side-nav-search" >
< a href = "../../index.html" class = "icon icon-home" > Triton
< / a >
< div role = "search" >
< form id = "rtd-search-form" class = "wy-form" action = "../../search.html" method = "get" >
< input type = "text" name = "q" placeholder = "Search docs" / >
< input type = "hidden" name = "check_keywords" value = "yes" / >
< input type = "hidden" name = "area" value = "default" / >
< / form >
< / div >
< / div >
< div class = "wy-menu wy-menu-vertical" data-spy = "affix" role = "navigation" aria-label = "main navigation" >
< p class = "caption" role = "heading" > < span class = "caption-text" > Getting Started< / span > < / p >
< ul >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../getting-started/installation.html" > Installation< / a > < / li >
< / ul >
< p class = "caption" role = "heading" > < span class = "caption-text" > Python API< / span > < / p >
< ul class = "current" >
< li class = "toctree-l1 current" > < a class = "reference internal" href = "../triton.html" > triton< / a > < ul class = "current" >
< li class = "toctree-l2 current" > < a class = "current reference internal" href = "#" > triton.jit< / a > < / li >
< li class = "toctree-l2" > < a class = "reference internal" href = "triton.autotune.html" > triton.autotune< / a > < / li >
< li class = "toctree-l2" > < a class = "reference internal" href = "triton.heuristics.html" > triton.heuristics< / a > < / li >
< li class = "toctree-l2" > < a class = "reference internal" href = "triton.Config.html" > triton.Config< / a > < / li >
< / ul >
< / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../triton.language.html" > triton.language< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../triton.testing.html" > triton.testing< / a > < / li >
< / ul >
< p class = "caption" role = "heading" > < span class = "caption-text" > Programming Guide< / span > < / p >
< ul >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../programming-guide/chapter-1/introduction.html" > Introduction< / a > < / li >
< li class = "toctree-l1" > < a class = "reference internal" href = "../../programming-guide/chapter-2/related-work.html" > Related Work< / a > < / li >
< / ul >
< / div >
< / div >
< / nav >
< section data-toggle = "wy-nav-shift" class = "wy-nav-content-wrap" >
< nav class = "wy-nav-top" aria-label = "top navigation" >
< i data-toggle = "wy-nav-top" class = "fa fa-bars" > < / i >
< a href = "../../index.html" > Triton< / a >
< / nav >
< div class = "wy-nav-content" >
< div class = "rst-content" >
< div role = "navigation" aria-label = "breadcrumbs navigation" >
< ul class = "wy-breadcrumbs" >
< li > < a href = "../../index.html" class = "icon icon-home" > < / a > » < / li >
< li > < a href = "../triton.html" > triton< / a > » < / li >
< li > triton.jit< / li >
< li class = "wy-breadcrumbs-aside" >
< a href = "../../_sources/python-api/generated/triton.jit.rst.txt" rel = "nofollow" > View page source< / a >
< / li >
< / ul >
< hr / >
< / div >
< div role = "main" class = "document" itemscope = "itemscope" itemtype = "http://schema.org/Article" >
< div itemprop = "articleBody" >
< div class = "section" id = "triton-jit" >
< h1 > triton.jit< a class = "headerlink" href = "#triton-jit" title = "Permalink to this headline" > ¶< / a > < / h1 >
< dl class = "py function" >
< dt class = "sig sig-object py" id = "triton.jit" >
2021-09-24 00:44:24 +00:00
< span class = "sig-prename descclassname" > < span class = "pre" > triton.< / span > < / span > < span class = "sig-name descname" > < span class = "pre" > jit< / span > < / span > < span class = "sig-paren" > (< / span > < em class = "sig-param" > < span class = "o" > < span class = "pre" > *< / span > < / span > < span class = "n" > < span class = "pre" > args< / span > < / span > < / em > , < em class = "sig-param" > < span class = "o" > < span class = "pre" > **< / span > < / span > < span class = "n" > < span class = "pre" > kwargs< / span > < / span > < / em > < span class = "sig-paren" > )< / span > < a class = "headerlink" href = "#triton.jit" title = "Permalink to this definition" > ¶< / a > < / dt >
2021-09-10 22:58:16 +00:00
< dd > < p > Decorator for JIT-compiling a function using the Triton compiler.< / p >
< dl class = "field-list simple" >
< dt class = "field-odd" > Note< / dt >
< dd class = "field-odd" > < p > When a jit’ < code class = "code docutils literal notranslate" > < span class = "pre" > torch.tensor< / span > < / code > arguments are implicitly converted to pointers using the < code class = "code docutils literal notranslate" > < span class = "pre" > .data_ptr()< / span > < / code > method.< / p >
< / dd >
< dt class = "field-even" > Note< / dt >
< dd class = "field-even" > < p > This function will be compiled and run on the GPU. It will only have access to:< / p >
< ul class = "simple" >
< li > < p > python primitives,< / p > < / li >
< li > < p > objects within the triton.language package,< / p > < / li >
< li > < p > arguments to this function,< / p > < / li >
< li > < p > other jit’ < / p > < / li >
< / ul >
< / dd >
< dt class = "field-odd" > Parameters< / dt >
< dd class = "field-odd" > < p > < strong > fn< / strong > (< em > Callable< / em > ) – < / p >
< / dd >
< / dl >
< / dd > < / dl >
< / div >
< / div >
< / div >
< footer >
< div class = "rst-footer-buttons" role = "navigation" aria-label = "footer navigation" >
< a href = "triton.autotune.html" class = "btn btn-neutral float-right" title = "triton.autotune" accesskey = "n" rel = "next" > Next < span class = "fa fa-arrow-circle-right" aria-hidden = "true" > < / span > < / a >
< a href = "../triton.html" class = "btn btn-neutral float-left" title = "triton" accesskey = "p" rel = "prev" > < span class = "fa fa-arrow-circle-left" aria-hidden = "true" > < / span > Previous< / a >
< / div >
< hr / >
< div role = "contentinfo" >
< p >
© Copyright 2020, Philippe Tillet.
< / p >
< / div >
Built with < a href = "https://www.sphinx-doc.org/" > Sphinx< / a > using a
< a href = "https://github.com/readthedocs/sphinx_rtd_theme" > theme< / a >
provided by < a href = "https://readthedocs.org" > Read the Docs< / a > .
< / footer >
< / div >
< / div >
< / section >
< / div >
< script type = "text/javascript" >
jQuery(function () {
SphinxRtdTheme.Navigation.enable(true);
});
< / script >
< / body >
< / html >
