o wi @sUddlmZzEddlmZmZmZmZmZdZdedddidedd dided d dided d didZ e e e eee feffe d<ee Wne yjddZGdddZddZddZdZYnwddZgdZdS))Union)DeviceLayerRepositoryregister_kernel_mappingreplace_kernel_forward_from_hubuse_kernel_forward_from_hubTcudaz!kernels-community/deformable-detrMultiScaleDeformableAttention)repo_id layer_namezkernels-community/moe Llama4TextMoezkernels-community/liger_kernels LigerRMSNormzmedmekk/triton-llama-mlpTritonLlamaMLP)r r RMSNormMLP_KERNEL_MAPPINGcOs dd}|S)NcSs|SN)clsrrb/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/transformers/integrations/hub_kernels.py decorator@z.use_kernel_forward_from_hub..decoratorr)argskwargsrrrrr?src@seZdZddZdS)rcOtd)NzNLayerRepository requires `kernels` to be installed. Run `pip install kernels`. RuntimeError)selfrrrrr__init__FzLayerRepository.__init__N)__name__ __module__ __qualname__rrrrrrEs rcOr)Nz^replace_kernel_forward_from_hub requires `kernels` to be installed. Run `pip install kernels`.rrrrrrrIsrcOr)NzVregister_kernel_mapping requires `kernels` to be installed. Run `pip install kernels`.rr#rrrrNrrFcCstSr)_hub_kernels_availablerrrris_hub_kernels_availableTrr%)rr%rrrN)typingrkernelsrrrrrr$rdictstr__annotations__ ImportErrorr%__all__rrrrsF &