o
    Æ µi%  ã                   @   sN   d dl Z d dlmZ d dlmZ ddd„Zdefdd„Zd	eddfd
d„Z	dS )é    N)Ú_C)Ú__version__Úreturnc                   C   s   t tƒ ƒ dS )zHPrints build information about the torch_tensorrt distribution to stdoutN)ÚprintÚget_build_info© r   r   úL/home/ubuntu/.local/lib/python3.10/site-packages/torch_tensorrt/ts/_utils.pyÚdump_build_info   s   r	   c                  C   s6   t  ¡ } tdttƒ d d ttjƒ d |  ƒ}|S )z¸Returns a string containing the build information of torch_tensorrt distribution

    Returns:
        str: String containing the build information for torch_tensorrt distribution
    zTorch-TensorRT Version: Ú
zUsing PyTorch Version: )r   r   Ústrr   Útorch)Úcore_build_infoÚ
build_infor   r   r   r      s"   ÿþýüûúÿ	r   Úgpu_idc                 C   s   t  | ¡ d S )N)r   Ú
set_device)r   r   r   r   r      s   r   )r   N)
r   Útorch_tensorrtr   Útorch_tensorrt._versionr   r	   r   r   Úintr   r   r   r   r   Ú<module>   s    
