e e2e diffusion latency)</p>
<div class="highlight-python notranslate"><div class="highlight"><pre><span></span><span class="kn">import</span><span class="w"> </span><span class="nn">register_sdpa</span>  <span class="c1"># Register SDPA as a standalone operator</span>
</pre></div>
</div>
<section id="import-the-following-libraries">
<h2>Import the following libraries<a class="headerlink" href="#import-the-following-libraries" title="Permalink to this heading">¶</a></h2>
<div class="highlight-python notran