
    rh                         U d dl mZ 	 d dlmZmZmZmZmZmZ dZ	d edd      id edd	      i ed
d      ej                   ed
d      idd edd      iej                   edd      ej                   edd      iej                   edd      iddZeeeeeef   ef   f   ed<    ee       d Zg dZy# e$ r d Z G d d      Zd Zd ZdZ	Y $w xY w)    )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                      d }|S )Nc                     | S N )clss    x/var/www/html/ai-insurance-compliance-backend/venv/lib/python3.12/site-packages/transformers/integrations/hub_kernels.py	decoratorz.use_kernel_forward_from_hub.<locals>.decoratorZ   s    J    r   )argskwargsr   s      r   r	   r	   Y   s    	 r   c                       e Zd Zd Zy)r   c                     t        d      )NzNLayerRepository requires `kernels` to be installed. Run `pip install kernels`.RuntimeError)selfr   r   s      r   __init__zLayerRepository.__init__`   s    oppr   N)__name__
__module____qualname__r%   r   r   r   r   r   _   s    	qr   r   c                      t        d      )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                      t        d      )NzVregister_kernel_mapping requires `kernels` to be installed. Run `pip install kernels`.r"   r*   s     r   r   r   h   s    sttr   Fc                      t         S r   )_hub_kernels_availabler   r   r   is_hub_kernels_availabler.   n   s    !!r   )r   r.   r	   r   r   N)typingr   kernelsr   r   r   r   r   r	   r-   	INFERENCETRAININGr   dictstr__annotations__ImportErrorr.   __all__r   r   r   <module>r8      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   B2C C C 