o
    پi                     @   s2   d dl Z d dlmZ dd Zedkre  dS dS )    N)add_constantc                  C   s>   d} t jddt jd }t|| }t |||  ksJ d S )Ni   r   i  )dtype)torcharangeint32cudar   all)csrcdst r   ]/home/ubuntu/.local/lib/python3.10/site-packages/sglang/jit_kernel/tests/test_add_constant.pymain   s   
r   __main__)r   sglang.jit_kernel.add_constantr   r   __name__r   r   r   r   <module>   s    
