o
    ¡ µi	  ã                
   @   s    d dl Z d dlZd dlZd dlZej ej e¡¡Zdd„ Z	dd„ Z
ej dd¡ej dd¡gZeD ]Ze
ej eejjeej d	¡rEd
nd¡ƒ q2e
eƒ dS )é    Nc                 C   s&   zt  | ¡ W d S  ty   Y d S w )N)ÚctypesÚCDLLÚOSError)Úlibrary© r   úJ/home/ubuntu/.local/lib/python3.10/site-packages/tensorrt_libs/__init__.pyÚtry_load   s
   ÿr   c                 C   sH   t  tj | d¡¡D ]}t|ƒ q
t  tj | d¡¡D ]}t|ƒ qd S )Nz*.so*z*.dll*)ÚglobÚiglobÚosÚpathÚjoinr   )r   Úlibr   r   r   Útry_load_libs_from_dir!   s
   

ÿr   ÚnvidiaÚcuda_runtimeÚ
cuda_nvrtcÚwinÚbinr   )r   r	   r   Úsysr   ÚrealpathÚdirnameÚ__file__ÚCURDIRr   r   r   ÚDEPENDENCY_PATHSÚdep_pathÚpardirÚplatformÚ
startswithr   r   r   r   Ú<module>   s(   þüÿ