ó
    Eñi  ã                   ó^   • S SK Jr  S SKJr  S SKr " S S\5      rS SKJr  Sr " S S	5      r	g)
é    )ÚEnum)ÚOptionalNc                   ó   • \ rS rSrSrSrg)Ú
EffectTypeé   ÚOrdered© N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__ÚORDEREDÚ__static_attributes__r	   ó    ÚS/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_library/effects.pyr   r      s   † ØƒGr   r   )ÚRegistrationHandle)z8__torch__.torch.classes.quantized.Conv2dPackedParamsBasez8__torch__.torch.classes.quantized.Conv3dPackedParamsBasez;__torch__.torch.classes.quantized.EmbeddingPackedParamsBasez8__torch__.torch.classes.quantized.LinearPackedParamsBasez/__torch__.torch.classes.xnnpack.Conv2dOpContextz/__torch__.torch.classes.xnnpack.LinearOpContextz8__torch__.torch.classes.xnnpack.TransposeConv2dOpContextc                   óŠ   • \ rS rSrSrS\4S jrSS jr\S\	\
   4S j5       r\R                  S	 5       rS
\	\
   S\4S jrSrg)ÚEffectHolderé   z2A holder where one can register an effect impl to.Úqualnamec                 ó0   • Xl         U R                  5         g ©N)r   Ú_set_default_effect)Úselfr   s     r   Ú__init__ÚEffectHolder.__init__   s   € Ø%ŒØ× Ñ Õ"r   ÚreturnNc                 ó”  • S U l         [        R                  R                  R	                  U R
                  5      u  pUR                  S5      n[        U5      S:”  a#  [        U5      S:w  a  [        SU S35      eUu  p$OSnUS:X  a  g U SU 3n[        R                  R                  X$5      b™  [        R                  R                  X$5      nUR                   Hi  n[        UR                  [        R                  5      (       d  M.  UR                  R!                  5       nU["        ;   a  MT  [$        R&                  U l           g    g g )	NÚ.é   é   zTried to split z' based on '.' but found more than 1 '.'Ú Úhigher_orderz::)Ú_effectÚtorchÚ_libraryÚutilsÚparse_namespacer   ÚsplitÚlenÚAssertionErrorÚ_CÚ_get_operation_overloadÚ_get_schemaÚ	argumentsÚ
isinstanceÚtypeÚ	ClassTypeÚstrÚskip_classesr   r   )r   Ú	namespaceÚopnamer)   ÚoverloadÚschemaÚargÚtype_strs           r   r   Ú EffectHolder._set_default_effect"   s  € Ø-1ˆŒô "ŸN™N×0Ñ0×@Ñ@ÀÇÁÓOÑˆ	Ø—‘˜SÓ!ˆÜˆu‹:˜‹>Ü�5‹z˜Q‹Ü$Ø% f XÐ-TÐUóð ð  %ÑˆF�HàˆHà˜Ó&Øà�;˜b  Ð)ˆÜ�8‰8×+Ñ+¨FÓ=ÑIô —X‘X×)Ñ)¨&Ó;ˆFØ×'Ô'�Ü˜cŸh™h¬¯©×8Ó8Ø"Ÿx™xŸ|™|›~�HØ¤<Ó/Ù Ü#-×#5Ñ#5�D”LÙò (ð	 Jr   c                 ó   • U R                   $ r   )r$   ©r   s    r   ÚeffectÚEffectHolder.effectA   s   € à�|‰|Ðr   c                 ó   • [        S5      e)NzUnable to directly set kernel.)ÚRuntimeError)r   Ú_s     r   r>   r?   E   s   € äÐ;Ó<Ð<r   r>   c                 ó8   ^ • UT l         U 4S jn[        U5      nU$ )z^Register an effect

Returns a RegistrationHandle that one can use to de-register this
effect.
c                  ó&   >• T R                  5         g r   )r   r=   s   €r   Úderegister_effectÚ0EffectHolder.register.<locals>.deregister_effectQ   s   ø€ Ø×$Ñ$Õ&r   )r$   r   )r   r>   rE   Úhandles   `   r   ÚregisterÚEffectHolder.registerI   s#   ø€ ð ˆŒõ	'ô $Ð$5Ó6ˆØˆr   )r$   r   )r   N)r
   r   r   r   Ú__doc__r3   r   r   Úpropertyr   r   r>   Úsetterr   rH   r   r	   r   r   r   r      sg   † Ù<ð# ô #ôð> ð˜ Ñ,ó ó ðð ‡]�]ñ=ó ð=ð˜x¨
Ñ3ð Ð8J÷ r   r   )
Úenumr   Útypingr   r%   r   Útorch._library.utilsr   r4   r   r	   r   r   Ú<module>rP      s0   ðÝ Ý ã ô�ô õ 4ð
€÷:ò :r   