an class="w"> </span><span class="o">=</span><span class="w"> </span><span class="n">torch_tensorrt</span><span class="o">::</span><span class="n">CompileSpec</span><span class="o">::</span><span class="n">DeviceType</span><span class="o">::</span><span class="n">kDLA</span><span class="p">;</span><span class="w"></span>

<span class="cp"># Make sure the gpu id is set to Xavier id for DLA</span>
<span class="n">compile_spec</span><span class="p">.</span><span class="n">device</span><span class="p">.</span><