    can have a performance impact so they are therefore opt-in. Used to suppress</span>
<span class="sd">    the warning about running unsafely in a multi-device context.</span>

<span class="sd">    Arguments:</span>
<span class="sd">        mode (bool): Enable (``True``) or disable (``False``) multi-device checks</span>

<span class="sd">    Example:</span>

<span class="sd">        .. code-block:: py</span>

<span class="sd">            with torch_tensorrt.runtime.set_multi_device_safe_mode(True):</span>