lass="p">{</span><span class="n">torch</span><span class="o">.</span><span class="n">float</span><span class="p">,</span> <span class="n">torch</span><span class="o">.</span><span class="n">half</span><span class="p">},</span> <span class="c1"># Enabling FP16 kernels</span>
    <span class="c1"># List of datatypes that should be configured for each input. Supported options torch.{float|half|int8|int32|bool}.</span>
    <span class="s2">"disable_tf32"</span><span class="p">:</span> <span class="kc">False</sp