o
    'i                     @   s.   d dl Z ddlmZ e jdd Zdd ZdS )    N   )gen_trtllm_utils_modulec                   C   s
   t   S N)r   build_and_load r   r   R/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/flashinfer/tllm_utils.pyget_trtllm_utils_module   s   
r   c                 C   s   t  |  d S r   )r   delay_kernel)stream_delay_micro_secsr   r   r   r	      s   r	   )	functoolsjit.tllm_utilsr   cacher   r	   r   r   r   r   <module>   s
    
