You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
<p><codeclass="docutils literal notranslate"><spanclass="pre">autocast(enabled=False)</span></code> subregions can be nested in autocast-enabled regions.
631
631
Locally disabling autocast can be useful, for example, if you want to force a subregion
632
632
to run in a particular <codeclass="docutils literal notranslate"><spanclass="pre">dtype</span></code>. Disabling autocast gives you explicit control over
633
-
the execution type. In the subregion, inputs from the surrounding region
633
+
the execution type. In the subregion, inputs from the surrounding region
634
634
should be cast to <codeclass="docutils literal notranslate"><spanclass="pre">dtype</span></code> before use:</p>
635
635
<divclass="highlight-default notranslate"><divclass="highlight"><pre><span></span><spanclass="c1"># Creates some tensors in default dtype (here assumed to be float32)</span>
0 commit comments