         <article itemprop="articleBody" id="pytorch-article" class="pytorch-article">
              
  <h1>Source code for torch_tensorrt.runtime.multi_device_safe_mode</h1><div class="highlight"><pre>
<span></span><span class="kn">import</span> <span class="nn">logging</span>
<span class="kn">from</span> <span class="nn">importlib.util</span> <span class="kn">import</span> <span class="n">find_spec</span>
<span class="kn">from</span> <span class="nn">typing</span> <span class="kn">import</span> <span clas