o
    ÉÙ¾i»  ã                   @   sD   d Z ddlmZ ddlmZ ddlmZ dd„ Zd	d
„ Zdd„ Z	dS )zDECORD TVM bridgeé    )Úabsolute_importé   )Ú_from_dlpacké   ©Ú
try_importc                  C   s   d} t d| ƒS )zmTry import tvm at runtime.

    Returns
    -------
    tvm module if found. Raise ImportError otherwise
    zqtvm is required, for installation guide, please checkout:
         https://tvm.apache.org/docs/install/index.htmlÚtvmr   )Úmsg© r
   úE/home/ubuntu/.local/lib/python3.10/site-packages/decord/bridge/tvm.pyÚtry_import_tvm   s   
r   c                 C   s   t ƒ }|j |  ¡ ¡S )zfrom decord to tvm, no copy)r   ÚndÚfrom_dlpackÚ	to_dlpack)Ú
decord_arrr   r
   r
   r   Úto_tvm   s   r   c                 C   s   t |  ¡ ƒS )zfrom tvm to decord, no copy)r   r   )Útvm_arrr
   r
   r   Úfrom_tvm   s   r   N)
Ú__doc__Ú
__future__r   Ú_ffi._ctypes.ndarrayr   Úutilsr   r   r   r   r
   r
   r
   r   Ú<module>   s    