
    <h                         % S SK Jr   S SKJrJrJrJrJrJr  Sr	S\" SSS90S\" SS	S90\" S
SS9\R                  \" S
SS90S.S\" SSS90\R                  \" SSS9\R                  \" SSS90\R                  \" SSS90S.S.r\\\\\\4   \4   4   \S'   \" \5        S r/ SQrg! \ a    S r " S S5      rS rS rSr	 N%f = f)    )Union)DeviceLayerRepositoryModeregister_kernel_mappingreplace_kernel_forward_from_hubuse_kernel_forward_from_hubTcudaz!kernels-community/deformable-detrMultiScaleDeformableAttention)repo_id
layer_namezkernels-community/moeLlama4TextMoezkernels-community/liger_kernelsLigerRMSNorm)r
   rocmzmedmekk/triton-llama-mlpTritonLlamaMLPzkernels-community/megablocksMegaBlocksMoeMLPzahadnagy/megablocks)r   r   RMSNormMLPr   _KERNEL_MAPPINGc                      S nU$ )Nc                     U $ N )clss    ]/var/www/html/shao/venv/lib/python3.13/site-packages/transformers/integrations/hub_kernels.py	decorator.use_kernel_forward_from_hub.<locals>.decoratorZ   s    J    r   )argskwargsr   s      r   r	   r	   Y   s    	 r   c                       \ rS rSrS rSrg)r   _   c                     [        S5      e)NzNLayerRepository requires `kernels` to be installed. Run `pip install kernels`.RuntimeError)selfr   r    s      r   __init__LayerRepository.__init__`   s    oppr   r   N)__name__
__module____qualname____firstlineno__r'   __static_attributes__r   r   r   r   r   _   s    	qr   r   c                      [        S5      e)Nz^replace_kernel_forward_from_hub requires `kernels` to be installed. Run `pip install kernels`.r$   r   r    s     r   r   r   c   s    l
 	
r   c                      [        S5      e)NzVregister_kernel_mapping requires `kernels` to be installed. Run `pip install kernels`.r$   r/   s     r   r   r   h   s    sttr   Fc                      [         $ r   )_hub_kernels_availabler   r   r   is_hub_kernels_availabler3   n   s    !!r   )r   r3   r	   r   r   N)typingr   kernelsr   r   r   r   r   r	   r2   	INFERENCETRAININGr   dictstr__annotations__ImportErrorr3   __all__r   r   r   <module>r=      sa   Z#  " O;:*
 O/*
 $9) =-!
 O2+
 :1  :1!	 11!
E4MOT#tE&#+$6$GHHI 4l O,6"9  #q q

u #+#s   BB+ +C	C	