span class="n">gpu_id</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">0</span><span class="p">;</span><span class="w"></span>

<span class="cp"># Set the DLA core id</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">device</span><span class="p">.</span><span class="n">dla_core</span><span class="w"> </span><span class="o">=</span><span class="w"> </span><span class="mi">1</span><span class="p">;</span><span class="w"></span>

<sp