Spaces:
Sleeping
Sleeping
| <html class="no-js" lang="en"> | |
| <head><meta charset="utf-8"/> | |
| <meta name="viewport" content="width=device-width,initial-scale=1"/> | |
| <meta name="color-scheme" content="light dark"><meta name="generator" content="Docutils 0.19: https://docutils.sourceforge.io/" /> | |
| <link rel="index" title="Index" href="genindex.html" /><link rel="search" title="Search" href="search.html" /><link rel="next" title="CUTLASS" href="cutlass.html" /><link rel="prev" title="Contributing" href="contribute.html" /> | |
| <link rel="canonical" href="docs/modules.html" /> | |
| <!-- Generated with Sphinx 6.1.3 and Furo 2023.03.27 --> | |
| <title>CUTLASS Python API - CUTLASS Python</title> | |
| <link rel="stylesheet" type="text/css" href="_static/pygments.css" /> | |
| <link rel="stylesheet" type="text/css" href="_static/styles/furo.css?digest=fad236701ea90a88636c2a8c73b44ae642ed2a53" /> | |
| <link rel="stylesheet" type="text/css" href="_static/copybutton.css" /> | |
| <link rel="stylesheet" type="text/css" href="_static/tabs.css" /> | |
| <link rel="stylesheet" type="text/css" href="_static/styles/furo-extensions.css?digest=30d1aed668e5c3a91c3e3bf6a60b675221979f0e" /> | |
| <style> | |
| body { | |
| --color-code-background: #eeffcc; | |
| --color-code-foreground: black; | |
| --color-brand-primary: #76B900; | |
| --color-brand-content: #76B900; | |
| } | |
| @media not print { | |
| body[data-theme="dark"] { | |
| --color-code-background: #272822; | |
| --color-code-foreground: #f8f8f2; | |
| --color-brand-primary: #76B900; | |
| --color-brand-content: #76B900; | |
| } | |
| @media (prefers-color-scheme: dark) { | |
| body:not([data-theme="light"]) { | |
| --color-code-background: #272822; | |
| --color-code-foreground: #f8f8f2; | |
| --color-brand-primary: #76B900; | |
| --color-brand-content: #76B900; | |
| } | |
| } | |
| } | |
| </style></head> | |
| <body> | |
| <script> | |
| document.body.dataset.theme = localStorage.getItem("theme") || "auto"; | |
| </script> | |
| <svg xmlns="http://www.w3.org/2000/svg" style="display: none;"> | |
| <symbol id="svg-toc" viewBox="0 0 24 24"> | |
| <title>Contents</title> | |
| <svg stroke="currentColor" fill="currentColor" stroke-width="0" viewBox="0 0 1024 1024"> | |
| <path d="M408 442h480c4.4 0 8-3.6 8-8v-56c0-4.4-3.6-8-8-8H408c-4.4 0-8 3.6-8 8v56c0 4.4 3.6 8 8 8zm-8 204c0 4.4 3.6 8 8 8h480c4.4 0 8-3.6 8-8v-56c0-4.4-3.6-8-8-8H408c-4.4 0-8 3.6-8 8v56zm504-486H120c-4.4 0-8 3.6-8 8v56c0 4.4 3.6 8 8 8h784c4.4 0 8-3.6 8-8v-56c0-4.4-3.6-8-8-8zm0 632H120c-4.4 0-8 3.6-8 8v56c0 4.4 3.6 8 8 8h784c4.4 0 8-3.6 8-8v-56c0-4.4-3.6-8-8-8zM115.4 518.9L271.7 642c5.8 4.6 14.4.5 14.4-6.9V388.9c0-7.4-8.5-11.5-14.4-6.9L115.4 505.1a8.74 8.74 0 0 0 0 13.8z"/> | |
| </svg> | |
| </symbol> | |
| <symbol id="svg-menu" viewBox="0 0 24 24"> | |
| <title>Menu</title> | |
| <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" | |
| stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather-menu"> | |
| <line x1="3" y1="12" x2="21" y2="12"></line> | |
| <line x1="3" y1="6" x2="21" y2="6"></line> | |
| <line x1="3" y1="18" x2="21" y2="18"></line> | |
| </svg> | |
| </symbol> | |
| <symbol id="svg-arrow-right" viewBox="0 0 24 24"> | |
| <title>Expand</title> | |
| <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" | |
| stroke-width="2" stroke-linecap="round" stroke-linejoin="round" class="feather-chevron-right"> | |
| <polyline points="9 18 15 12 9 6"></polyline> | |
| </svg> | |
| </symbol> | |
| <symbol id="svg-sun" viewBox="0 0 24 24"> | |
| <title>Light mode</title> | |
| <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" | |
| stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round" class="feather-sun"> | |
| <circle cx="12" cy="12" r="5"></circle> | |
| <line x1="12" y1="1" x2="12" y2="3"></line> | |
| <line x1="12" y1="21" x2="12" y2="23"></line> | |
| <line x1="4.22" y1="4.22" x2="5.64" y2="5.64"></line> | |
| <line x1="18.36" y1="18.36" x2="19.78" y2="19.78"></line> | |
| <line x1="1" y1="12" x2="3" y2="12"></line> | |
| <line x1="21" y1="12" x2="23" y2="12"></line> | |
| <line x1="4.22" y1="19.78" x2="5.64" y2="18.36"></line> | |
| <line x1="18.36" y1="5.64" x2="19.78" y2="4.22"></line> | |
| </svg> | |
| </symbol> | |
| <symbol id="svg-moon" viewBox="0 0 24 24"> | |
| <title>Dark mode</title> | |
| <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" | |
| stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round" class="icon-tabler-moon"> | |
| <path stroke="none" d="M0 0h24v24H0z" fill="none" /> | |
| <path d="M12 3c.132 0 .263 0 .393 0a7.5 7.5 0 0 0 7.92 12.446a9 9 0 1 1 -8.313 -12.454z" /> | |
| </svg> | |
| </symbol> | |
| <symbol id="svg-sun-half" viewBox="0 0 24 24"> | |
| <title>Auto light/dark mode</title> | |
| <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24" fill="none" stroke="currentColor" | |
| stroke-width="1.5" stroke-linecap="round" stroke-linejoin="round" class="icon-tabler-shadow"> | |
| <path stroke="none" d="M0 0h24v24H0z" fill="none"/> | |
| <circle cx="12" cy="12" r="9" /> | |
| <path d="M13 12h5" /> | |
| <path d="M13 15h4" /> | |
| <path d="M13 18h1" /> | |
| <path d="M13 9h4" /> | |
| <path d="M13 6h1" /> | |
| </svg> | |
| </symbol> | |
| </svg> | |
| <input type="checkbox" class="sidebar-toggle" name="__navigation" id="__navigation"> | |
| <input type="checkbox" class="sidebar-toggle" name="__toc" id="__toc"> | |
| <label class="overlay sidebar-overlay" for="__navigation"> | |
| <div class="visually-hidden">Hide navigation sidebar</div> | |
| </label> | |
| <label class="overlay toc-overlay" for="__toc"> | |
| <div class="visually-hidden">Hide table of contents sidebar</div> | |
| </label> | |
| <div class="page"> | |
| <header class="mobile-header"> | |
| <div class="header-left"> | |
| <label class="nav-overlay-icon" for="__navigation"> | |
| <div class="visually-hidden">Toggle site navigation sidebar</div> | |
| <i class="icon"><svg><use href="#svg-menu"></use></svg></i> | |
| </label> | |
| </div> | |
| <div class="header-center"> | |
| <a href="index.html"><div class="brand">CUTLASS Python</div></a> | |
| </div> | |
| <div class="header-right"> | |
| <div class="theme-toggle-container theme-toggle-header"> | |
| <button class="theme-toggle"> | |
| <div class="visually-hidden">Toggle Light / Dark / Auto color theme</div> | |
| <svg class="theme-icon-when-auto"><use href="#svg-sun-half"></use></svg> | |
| <svg class="theme-icon-when-dark"><use href="#svg-moon"></use></svg> | |
| <svg class="theme-icon-when-light"><use href="#svg-sun"></use></svg> | |
| </button> | |
| </div> | |
| <label class="toc-overlay-icon toc-header-icon no-toc" for="__toc"> | |
| <div class="visually-hidden">Toggle table of contents sidebar</div> | |
| <i class="icon"><svg><use href="#svg-toc"></use></svg></i> | |
| </label> | |
| </div> | |
| </header> | |
| <aside class="sidebar-drawer"> | |
| <div class="sidebar-container"> | |
| <div class="sidebar-sticky"><a class="sidebar-brand" href="index.html"> | |
| <div class="sidebar-logo-container"> | |
| <img class="sidebar-logo only-light" src="_static/cutlass-logo-small.png" alt="Light Logo"/> | |
| <img class="sidebar-logo only-dark" src="_static/cutlass-logo-small.png" alt="Dark Logo"/> | |
| </div> | |
| <span class="sidebar-brand-text">CUTLASS Python</span> | |
| </a><form class="sidebar-search-container" method="get" action="search.html" role="search"> | |
| <input class="sidebar-search" placeholder="Search" name="q" aria-label="Search"> | |
| <input type="hidden" name="check_keywords" value="yes"> | |
| <input type="hidden" name="area" value="default"> | |
| </form> | |
| <div id="searchbox"></div><div class="sidebar-scroll"><div class="sidebar-tree"> | |
| <ul> | |
| <li class="toctree-l1"><a class="reference internal" href="index.html">Home</a></li> | |
| </ul> | |
| <p class="caption" role="heading"><span class="caption-text">Getting Started:</span></p> | |
| <ul> | |
| <li class="toctree-l1"><a class="reference internal" href="install.html">Installation</a></li> | |
| <li class="toctree-l1"><a class="reference internal" href="externals/00_basic_gemm.html">Getting Started</a></li> | |
| <li class="toctree-l1"><a class="reference internal" href="contribute.html">Contributing</a></li> | |
| </ul> | |
| <p class="caption" role="heading"><span class="caption-text">Python Documentation:</span></p> | |
| <ul class="current"> | |
| <li class="toctree-l1 current has-children current-page"><a class="current reference internal" href="#">CUTLASS Python API</a><input checked="" class="toctree-checkbox" id="toctree-checkbox-1" name="toctree-checkbox-1" role="switch" type="checkbox"/><label for="toctree-checkbox-1"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> | |
| <li class="toctree-l2 has-children"><a class="reference internal" href="cutlass.html">CUTLASS</a><input class="toctree-checkbox" id="toctree-checkbox-2" name="toctree-checkbox-2" role="switch" type="checkbox"/><label for="toctree-checkbox-2"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.emit.html">Emitters</a></li> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.op.html">Operations</a></li> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.utils.html">Utilities</a></li> | |
| </ul> | |
| </li> | |
| </ul> | |
| </li> | |
| </ul> | |
| <p class="caption" role="heading"><span class="caption-text">Examples and Tutorials:</span></p> | |
| <ul> | |
| <li class="toctree-l1 has-children"><a class="reference internal" href="examples.html">Examples</a><input class="toctree-checkbox" id="toctree-checkbox-3" name="toctree-checkbox-3" role="switch" type="checkbox"/><label for="toctree-checkbox-3"><div class="visually-hidden">Toggle child pages in navigation</div><i class="icon"><svg><use href="#svg-arrow-right"></use></svg></i></label><ul> | |
| <li class="toctree-l2"><a class="reference internal" href="externals/00_basic_gemm.html">Basic GEMM</a></li> | |
| <li class="toctree-l2"><a class="reference internal" href="externals/01_epilogue.html">Epilogue</a></li> | |
| <li class="toctree-l2"><a class="reference internal" href="externals/02_pytorch_extension_grouped_gemm.html">PyTorch Extension</a></li> | |
| </ul> | |
| </li> | |
| </ul> | |
| <p class="caption" role="heading"><span class="caption-text">Reference:</span></p> | |
| <ul> | |
| <li class="toctree-l1"><a class="reference external" href="https://github.com/NVIDIA/cutlass">Github</a></li> | |
| </ul> | |
| </div> | |
| </div> | |
| </div> | |
| </div> | |
| </aside> | |
| <div class="main"> | |
| <div class="content"> | |
| <div class="article-container"> | |
| <a href="#" class="back-to-top muted-link"> | |
| <svg xmlns="http://www.w3.org/2000/svg" viewBox="0 0 24 24"> | |
| <path d="M13 20h-2V8l-5.5 5.5-1.42-1.42L12 4.16l7.92 7.92-1.42 1.42L13 8v12z"></path> | |
| </svg> | |
| <span>Back to top</span> | |
| </a> | |
| <div class="content-icon-container"> | |
| <div class="theme-toggle-container theme-toggle-content"> | |
| <button class="theme-toggle"> | |
| <div class="visually-hidden">Toggle Light / Dark / Auto color theme</div> | |
| <svg class="theme-icon-when-auto"><use href="#svg-sun-half"></use></svg> | |
| <svg class="theme-icon-when-dark"><use href="#svg-moon"></use></svg> | |
| <svg class="theme-icon-when-light"><use href="#svg-sun"></use></svg> | |
| </button> | |
| </div> | |
| <label class="toc-overlay-icon toc-content-icon no-toc" for="__toc"> | |
| <div class="visually-hidden">Toggle table of contents sidebar</div> | |
| <i class="icon"><svg><use href="#svg-toc"></use></svg></i> | |
| </label> | |
| </div> | |
| <article role="main"> | |
| <section id="cutlass-python-api"> | |
| <h1>CUTLASS Python API<a class="headerlink" href="#cutlass-python-api" title="Permalink to this heading">#</a></h1> | |
| <div class="toctree-wrapper compound"> | |
| <ul> | |
| <li class="toctree-l1"><a class="reference internal" href="cutlass.html">CUTLASS</a><ul> | |
| <li class="toctree-l2"><a class="reference internal" href="cutlass.html#subpackages">Subpackages</a><ul> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.emit.html">Emitters</a><ul> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.emit.html#module-cutlass.emit.common">Common</a></li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.emit.html#module-cutlass.emit.pytorch">PyTorch</a><ul> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.emit.html#cutlass.emit.pytorch.pytorch"><code class="docutils literal notranslate"><span class="pre">pytorch()</span></code></a></li> | |
| </ul> | |
| </li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.op.html">Operations</a><ul> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.op.html#module-cutlass.op.gemm">GEMM</a><ul> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.op.html#cutlass.op.gemm.Gemm"><code class="docutils literal notranslate"><span class="pre">Gemm</span></code></a></li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.op.html#module-cutlass.op.gemm_grouped">Grouped GEMM</a><ul> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.op.html#cutlass.op.gemm_grouped.GroupedGemm"><code class="docutils literal notranslate"><span class="pre">GroupedGemm</span></code></a></li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.op.html#module-cutlass.op.op">Operation</a><ul> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.op.html#cutlass.op.op.OperationBase"><code class="docutils literal notranslate"><span class="pre">OperationBase</span></code></a></li> | |
| </ul> | |
| </li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.utils.html">Utilities</a><ul> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.utils.html#module-cutlass.utils.check">Checks</a><ul> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.check.alignment_or_default"><code class="docutils literal notranslate"><span class="pre">alignment_or_default()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.check.calculate_smem_usage"><code class="docutils literal notranslate"><span class="pre">calculate_smem_usage()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.check.calculate_smem_usage_per_stage"><code class="docutils literal notranslate"><span class="pre">calculate_smem_usage_per_stage()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.check.valid_cluster_shape"><code class="docutils literal notranslate"><span class="pre">valid_cluster_shape()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.check.valid_kernel_schedule"><code class="docutils literal notranslate"><span class="pre">valid_kernel_schedule()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.check.valid_stage_count"><code class="docutils literal notranslate"><span class="pre">valid_stage_count()</span></code></a></li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.utils.html#module-cutlass.utils.datatypes">Data Types</a><ul> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.backend_math_operation"><code class="docutils literal notranslate"><span class="pre">backend_math_operation()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.bfloat16_library_type"><code class="docutils literal notranslate"><span class="pre">bfloat16_library_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.bfloat16_type"><code class="docutils literal notranslate"><span class="pre">bfloat16_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.binding_layout"><code class="docutils literal notranslate"><span class="pre">binding_layout()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.binding_library_type"><code class="docutils literal notranslate"><span class="pre">binding_library_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.binding_opclass"><code class="docutils literal notranslate"><span class="pre">binding_opclass()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.binding_type"><code class="docutils literal notranslate"><span class="pre">binding_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.construct_backend_td"><code class="docutils literal notranslate"><span class="pre">construct_backend_td()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.cupy_library_type"><code class="docutils literal notranslate"><span class="pre">cupy_library_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.cupy_type"><code class="docutils literal notranslate"><span class="pre">cupy_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.get_datatype_and_layout"><code class="docutils literal notranslate"><span class="pre">get_datatype_and_layout()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.has_binding_type"><code class="docutils literal notranslate"><span class="pre">has_binding_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.library_layout"><code class="docutils literal notranslate"><span class="pre">library_layout()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.library_to_binding"><code class="docutils literal notranslate"><span class="pre">library_to_binding()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.library_type"><code class="docutils literal notranslate"><span class="pre">library_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.numpy_library_type"><code class="docutils literal notranslate"><span class="pre">numpy_library_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.numpy_type"><code class="docutils literal notranslate"><span class="pre">numpy_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.td_from_profiler_op"><code class="docutils literal notranslate"><span class="pre">td_from_profiler_op()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.td_from_profiler_td"><code class="docutils literal notranslate"><span class="pre">td_from_profiler_td()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.torch_library_type"><code class="docutils literal notranslate"><span class="pre">torch_library_type()</span></code></a></li> | |
| <li class="toctree-l5"><a class="reference internal" href="cutlass.utils.html#cutlass.utils.datatypes.torch_type"><code class="docutils literal notranslate"><span class="pre">torch_type()</span></code></a></li> | |
| </ul> | |
| </li> | |
| </ul> | |
| </li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l2"><a class="reference internal" href="cutlass.html#module-cutlass.epilogue">Epilogue</a><ul> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.html#cutlass.epilogue.get_activation_epilogue"><code class="docutils literal notranslate"><span class="pre">get_activation_epilogue()</span></code></a></li> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.html#cutlass.epilogue.get_activations"><code class="docutils literal notranslate"><span class="pre">get_activations()</span></code></a></li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l2"><a class="reference internal" href="cutlass.html#module-cutlass.library_defaults">Library Defaults</a><ul> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.ArchOptions"><code class="docutils literal notranslate"><span class="pre">ArchOptions</span></code></a><ul> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.ArchOptions.opclass_supports_combination"><code class="docutils literal notranslate"><span class="pre">ArchOptions.opclass_supports_combination()</span></code></a></li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.ArchOptions.operations"><code class="docutils literal notranslate"><span class="pre">ArchOptions.operations()</span></code></a></li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.ArchOptions.supporting_opclasses"><code class="docutils literal notranslate"><span class="pre">ArchOptions.supporting_opclasses()</span></code></a></li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.KernelsForDataType"><code class="docutils literal notranslate"><span class="pre">KernelsForDataType</span></code></a><ul> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.KernelsForDataType.add"><code class="docutils literal notranslate"><span class="pre">KernelsForDataType.add()</span></code></a></li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.KernelsForDataType.alignments"><code class="docutils literal notranslate"><span class="pre">KernelsForDataType.alignments</span></code></a></li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.KernelsForDataType.all_operations"><code class="docutils literal notranslate"><span class="pre">KernelsForDataType.all_operations</span></code></a></li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.KernelsForDataType.find_alignment"><code class="docutils literal notranslate"><span class="pre">KernelsForDataType.find_alignment()</span></code></a></li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.KernelsForDataType.operations"><code class="docutils literal notranslate"><span class="pre">KernelsForDataType.operations()</span></code></a></li> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.KernelsForDataType.sort"><code class="docutils literal notranslate"><span class="pre">KernelsForDataType.sort()</span></code></a></li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.OptionRegistry"><code class="docutils literal notranslate"><span class="pre">OptionRegistry</span></code></a><ul> | |
| <li class="toctree-l4"><a class="reference internal" href="cutlass.html#cutlass.library_defaults.OptionRegistry.options_for_cc"><code class="docutils literal notranslate"><span class="pre">OptionRegistry.options_for_cc()</span></code></a></li> | |
| </ul> | |
| </li> | |
| </ul> | |
| </li> | |
| <li class="toctree-l2"><a class="reference internal" href="cutlass.html#module-cutlass.swizzle">Swizzle</a><ul> | |
| <li class="toctree-l3"><a class="reference internal" href="cutlass.html#cutlass.swizzle.get_swizzling_functors"><code class="docutils literal notranslate"><span class="pre">get_swizzling_functors()</span></code></a></li> | |
| </ul> | |
| </li> | |
| </ul> | |
| </li> | |
| </ul> | |
| </div> | |
| </section> | |
| </article> | |
| </div> | |
| <footer> | |
| <div class="related-pages"> | |
| <a class="next-page" href="cutlass.html"> | |
| <div class="page-info"> | |
| <div class="context"> | |
| <span>Next</span> | |
| </div> | |
| <div class="title">CUTLASS</div> | |
| </div> | |
| <svg class="furo-related-icon"><use href="#svg-arrow-right"></use></svg> | |
| </a> | |
| <a class="prev-page" href="contribute.html"> | |
| <svg class="furo-related-icon"><use href="#svg-arrow-right"></use></svg> | |
| <div class="page-info"> | |
| <div class="context"> | |
| <span>Previous</span> | |
| </div> | |
| <div class="title">Contributing</div> | |
| </div> | |
| </a> | |
| </div> | |
| <div class="bottom-of-page"> | |
| <div class="left-details"> | |
| <div class="copyright"> | |
| Copyright © 2023, NVIDIA | |
| </div> | |
| Made with <a href="https://www.sphinx-doc.org/">Sphinx</a> and <a class="muted-link" href="https://pradyunsg.me">@pradyunsg</a>'s | |
| <a href="https://github.com/pradyunsg/furo">Furo</a> | |
| </div> | |
| <div class="right-details"> | |
| <div class="icons"> | |
| <a class="muted-link " href="https://github.com/NVIDIA/cutlass" aria-label="GitHub"> | |
| <svg stroke="currentColor" fill="currentColor" stroke-width="0" viewBox="0 0 16 16"> | |
| <path fill-rule="evenodd" d="M8 0C3.58 0 0 3.58 0 8c0 3.54 2.29 6.53 5.47 7.59.4.07.55-.17.55-.38 0-.19-.01-.82-.01-1.49-2.01.37-2.53-.49-2.69-.94-.09-.23-.48-.94-.82-1.13-.28-.15-.68-.52-.01-.53.63-.01 1.08.58 1.23.82.72 1.21 1.87.87 2.33.66.07-.52.28-.87.51-1.07-1.78-.2-3.64-.89-3.64-3.95 0-.87.31-1.59.82-2.15-.08-.2-.36-1.02.08-2.12 0 0 .67-.21 2.2.82.64-.18 1.32-.27 2-.27.68 0 1.36.09 2 .27 1.53-1.04 2.2-.82 2.2-.82.44 1.1.16 1.92.08 2.12.51.56.82 1.27.82 2.15 0 3.07-1.87 3.75-3.65 3.95.29.25.54.73.54 1.48 0 1.07-.01 1.93-.01 2.2 0 .21.15.46.55.38A8.013 8.013 0 0 0 16 8c0-4.42-3.58-8-8-8z"></path> | |
| </svg> | |
| </a> | |
| </div> | |
| </div> | |
| </div> | |
| </footer> | |
| </div> | |
| <aside class="toc-drawer no-toc"> | |
| </aside> | |
| </div> | |
| </div><script data-url_root="./" id="documentation_options" src="_static/documentation_options.js"></script> | |
| <script src="_static/doctools.js"></script> | |
| <script src="_static/sphinx_highlight.js"></script> | |
| <script src="_static/scripts/furo.js"></script> | |
| <script src="_static/clipboard.min.js"></script> | |
| <script src="_static/copybutton.js"></script> | |
| <script src="_static/tabs.js"></script> | |
| <script crossorigin="anonymous" integrity="sha256-Ae2Vz/4ePdIu6ZyI/5ZGsYnb+m0JlOmKPjt6XZ9JJkA=" src="https://cdnjs.cloudflare.com/ajax/libs/require.js/2.3.4/require.min.js"></script> | |
| </body> | |
| </html> |