an class="cp"># If a layer fails to run on DLA it will fallback to GPU</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">device</span><span class="p">.</span><span class="n">allow_gpu_fallback</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="nb">true</span><span class="p">;</span><span class="w"></span>

<span class="cp"># Set the workspace size</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">workspac