o
    پif                     @   sT   d dl Z d dlZd dlmZ d dlmZ G dd deZedkr(e jdd dS dS )	    N)DynamicGradMode)CustomTestCasec                   @   s,   e Zd Zdd Zdd Zdd Zdd Zd	S )
TestDynamicGradModec                 C   8   t d t  dd }| }| |j o|  d S )NTc                   S   
   t dS Nr   torchempty r   r   V/home/ubuntu/.local/lib/python3.10/site-packages/sglang/test/test_dynamic_grad_mode.pycreate_tensor_x      
z;TestDynamicGradMode.test_inference.<locals>.create_tensor_xr   set_inference_mode
assertTruerequires_gradis_inference)selfr   Xr   r   r   test_inference
   s
   

z"TestDynamicGradMode.test_inferencec                 C   s:   t d t  dd }| }| |j o|   d S )NFc                   S   r   r   r   r   r   r   r   create_tensor_y   r   z9TestDynamicGradMode.test_no_grad.<locals>.create_tensor_yr   )r   r   Yr   r   r   test_no_grad   s
   

z TestDynamicGradMode.test_no_gradc                 C   r   )NFc                   S   6   t   t dW  d    S 1 sw   Y  d S r   )r	   inference_moder
   r   r   r   r   create_tensor_z$      
$zBTestDynamicGradMode.test_nested_inference.<locals>.create_tensor_zr   )r   r   Zr   r   r   test_nested_inference    
   

z)TestDynamicGradMode.test_nested_inferencec                 C   r   )NTc                   S   r   r   )r	   no_gradr
   r   r   r   r   create_tensor_w0   r   z@TestDynamicGradMode.test_nested_no_grad.<locals>.create_tensor_wr   )r   r"   Wr   r   r   test_nested_no_grad,   r    z'TestDynamicGradMode.test_nested_no_gradN)__name__
__module____qualname__r   r   r   r$   r   r   r   r   r   	   s
    r   __main__   )	verbosity)	unittestr	   sglang.srt.utilsr   sglang.test.test_utilsr   r   r%   mainr   r   r   r   <module>   s    0