
    +jH                         S SK Jr  S SKJrJr  S SKJr  S r " S S\R                  5      r	 " S S\5      r
S	\
4S
 jrg)    )override)ComfyExtensionio)set_torch_compile_wrapperc                 L    U  Vs/ s H  nSUR                   ;  PM     sn$ s  snf )Ntransformer_options)name)guard_entriesentrys     @/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_torch_compile.pyskip_torch_compile_dictr      s$    CPQ=%"%**4=QQQs   !c                   l    \ rS rSr\S\R                  4S j5       r\S\R                  4S j5       r	Sr
g)TorchCompileModel   returnc                     [         R                  " SS[         R                  R                  S5      [         R                  R                  SSS/SS9/[         R                  R                  5       /SS	9$ )
Nr   experimentalmodelbackendinductor
cudagraphsT)optionsadvanced)node_idcategoryinputsoutputsis_experimental)r   SchemaModelInputComboOutput)clss    r   define_schemaTorchCompileModel.define_schema	   sg    yy'#w''6!   XX__&' 
 	
    c                 l    UR                  SS9n[        X2S[        0S9  [        R                  " U5      $ )NT)disable_dynamicguard_filter_fn)r   r   r   )cloner   r   r   
NodeOutput)r$   r   r   ms       r   executeTorchCompileModel.execute   s2    KKK-!EVXoDpq}}Qr'    N)__name__
__module____qualname____firstlineno__classmethodr   r   r%   r,   r.   __static_attributes__r0   r'   r   r   r      s>    
bii 
 
       r'   r   c                   L    \ rS rSr\S\\\R                        4S j5       r	Sr
g)TorchCompileExtension!   r   c                    #    [         /$ 7fN)r   )selfs    r   get_node_list#TorchCompileExtension.get_node_list"   s      
 	
s   
r0   N)r1   r2   r3   r4   r   listtyper   	ComfyNoder=   r6   r0   r'   r   r8   r8   !   s)    
T$r||*<%= 
 
r'   r8   r   c                     #    [        5       $ 7fr;   )r8   r0   r'   r   comfy_entrypointrC   )   s      ""s   N)typing_extensionsr   comfy_api.latestr   r   comfy_api.torch_helpersr   r   rA   r   r8   rC   r0   r'   r   <module>rG      s=    & / =R   2
N 
# 5 #r'   