ó
    ¤3j %  ã                   óŽ   • / S Qr SSKJr  SSKJrJrJrJrJrJ	r	J
r
JrJr  SSKJrJrJrJr  SSKJrJr  S\4S jrS r\" 5         g	)
)Ú
apply_ropeÚapply_rope1Úapply_rope_split_halfÚapply_rope_split_half1Údequantize_mxfp8Údequantize_nvfp4Údequantize_per_tensor_fp8Úgemv_awq_w4a16Úquantize_mxfp8Úquantize_nvfp4Úquantize_per_tensor_fp8Úquantize_svdquant_w4a4Úscaled_mm_mxfp8Úscaled_mm_nvfp4Úscaled_mm_svdquant_w4a4Ústochastic_rounding_fp8é   )r	   )	r   r   r   r
   r   r   r   r   r   )r   r   r   r   )r   r   Úreturnc                  óì  • SS K n SSKJnJnJn  [        1 Sk5      n[        U R                  U R                  U R                  15      nU" U" US9U" [        U R                  15      S9U" [        U R                  U R                  15      S9S.US9U" U" [        U R                  U R                  15      S9U" US9U" US9S.US9U" U" US9U" [        U R                  15      S9U" [        U R                  U R                  15      S9S.US9U" U" UU" S5      4S	9U" [        U R                  15      S9S
.US9U" U" [        U R                  15      U" S5      4S	9U" [        U R                  15      S9U" [        U R                  15      S9U" US9S.US9U" U" [        U R                  15      U" S5      4S	9U" [        U R                  15      U" S5      4S	9U" [        U R                  15      S9U" [        U R                  15      S9U" [        U R                  15      S9U" [        U R                  15      S9U" US9S.US9U" U" US9U" US9S.US9U" U" US9U" US9U" US9S.US9U" U" US9U" US9S.US9U" U" US9U" US9U" US9S.US9U" U" XQ" S5      4S	9U" XQ" S5      4S	9U" XQ" S5      4S	9S.US9U" U" [        U R                  U R                  15      U" S5      4S	9U" [        U R                  15      S9U" XQ" S5      4S	9U" US9U" UU" S5      4S	9U" US9S.US9U" U" US9U" [        U R                  15      U" S5      4S	9U" XQ" S5      4S	9U" XQ" S5      4S	9S.US9S.n[        U S5      (       aæ  U" SU" UU" S5      4S	90US9US'   U" U" [        U R                  15      U" S5      4S	9U" [        U R                  15      S9U" US9S.US9US'   U" U" [        U R                  15      U" S5      4S	9U" [        U R                  15      U" S5      4S	9U" [        U R                  15      S9U" [        U R                  15      S9U" US9S.US9US'   U$ )Né    )Ú	ExactDimsÚFunctionConstraintsÚParamConstraint>   Ú*ÚcpuÚhpuÚmpsÚxpuÚcudaÚmeta)Údtypes)ÚxÚscaleÚoutput_type)ÚparamsÚdefault_devices)r!   Úrngr#   é   )r    Úshape_rules)r!   Úper_tensor_scale)Úqxr)   Úblock_scalesr#   )ÚaÚbÚtensor_scale_aÚtensor_scale_bÚblock_scale_aÚblock_scale_bÚ	out_dtype)r!   Ú	freqs_cis)ÚxqÚxkr3   r   )r!   ÚsmoothÚ	lora_down)ÚactÚwgtÚascalesÚwscalesÚlora_act_inÚlora_up)r!   Úqweightr;   Úwzeros)r   r   r   r   r   r   r   r   r   r   r   r   r	   Úfloat8_e8m0fnur!   r
   )r*   r+   r#   r   )r,   r-   r0   r1   r2   r   )ÚtorchÚcomfy_kitchen.constraintsr   r   r   Ú	frozensetÚfloat32Úfloat16Úbfloat16Úfloat8_e4m3fnÚfloat8_e5m2Úuint8Úint8Úhasattrr@   )rA   r   r   r   Úall_devicesÚstandard_floatsÚouts          Ú_/home/wildlama/miniconda3/lib/python3.13/site-packages/comfy_kitchen/backends/eager/__init__.pyÚ_build_constraintsrP   $   s¯  € Û÷ñ ô ÒMÓN€KÜ §¡°·±¸u¿~¹~Ð NÓO€Oñ $7á$¨OÑ<Ù(´	¸5¿=¹=¸/Ó0JÑKÙ.Ü$ e×&9Ñ&9¸5×;LÑ;LÐ%MÓNñ ñð (ñ	$
ñ &9á$Ü$ e×&9Ñ&9¸5×;LÑ;LÐ%MÓNññ )°Ñ@Ù.°oÑFñð (ñ	&
ñ $7á$¨OÑ<Ù&¬i¸¿¹¸Ó.FÑGÙ.Ü$ e×&9Ñ&9¸5×;LÑ;LÐ%MÓNñ ñð (ñ	$
ñ .á$Ø*Ù!*¨1£ ññ %4¼9ÀeÇmÁmÀ_Ó;UÑ$Vñð (ñ	
ñ 0á%Ü$ e§k¡k ]Ó3Ù!*¨1£ ññ %4¼9ÀeÇmÁmÀ_Ó;UÑ$VÙ /Ü$ e×&9Ñ&9Ð%:Ó;ñ!ñ  /°oÑFñ
ð (ñ
ñ /á$Ü$ e§k¡k ]Ó3Ù!*¨1£ ññ %Ü$ e§k¡k ]Ó3Ù!*¨1£ ññ #2¼ÀEÇMÁMÀ?Ó9SÑ"TÙ"1¼ÀEÇMÁMÀ?Ó9SÑ"TÙ!0Ü$ e×&9Ñ&9Ð%:Ó;ñ"ñ "1Ü$ e×&9Ñ&9Ð%:Ó;ñ"ñ -°OÑDñ#ð& (ñ)
ñ, +á$¨OÑ<Ù,°OÑDñð (ñ
ñ *á%¨_Ñ=Ù%¨_Ñ=Ù,°OÑDñð
 (ñ
ñ #6á$¨OÑ<Ù,°OÑDñð (ñ#
ñ "5á%¨_Ñ=Ù%¨_Ñ=Ù,°OÑDñð
 (ñ"
ñ #6á$¨OÈ)ÐTUË,ÈÑYÙ)°ÈyÐYZË|ÈoÑ^Ù,Ø*¸À1»¸ññð (ñ	#
ñ $7á&Ü$ e§j¡j°%·+±+Ð%>Ó?Ù!*¨1£ ññ 'Ü$ e§j¡j \Ó2ññ +°/ÐPYÐZ[ÓP\ÈÑ_Ù*°/ÑBÙ.Ø*Ù!*¨1£ ñ ñ +°/ÑBñð  (ñ#$
ñ& .á$¨OÑ<Ù*Ü$ e§j¡j \Ó2Ù!*¨1£ ññ +°/ÐPYÐZ[ÓP\ÈÑ_Ù)°ÈyÐYZË|ÈoÑ^ñð (ñ
ñQT€Côl ˆuÐ&×'Ñ'Ù 3à™Ø.Ù%.¨q£\ Oñðð !,ñ!-ˆÐÑñ #6á)Ü(¨%×*=Ñ*=Ð)>Ó?Ù%.¨q£\ Oññ %4Ü(¨%×*>Ñ*>Ð)?Ó@ñ%ñ $3¸/Ñ#Jñ	ð !,ñ#-ˆÐÑñ "5á(Ü(¨%×*=Ñ*=Ð)>Ó?Ù%.¨q£\ Oññ )Ü(¨%×*=Ñ*=Ð)>Ó?Ù%.¨q£\ Oññ &5Ü(¨%×*>Ñ*>Ð)?Ó@ñ&ñ &5Ü(¨%×*>Ñ*>Ð)?Ó@ñ&ñ "1¸Ñ!Hñð" !,ñ%"-ˆÐÑð( €Jó    c                  ób   • SSK Jn   U R                  S[        [        [
        S9[        5       S9  g )Nr   ©ÚregistryÚeager)Úfromlist)ÚnameÚmoduleÚcapabilities)Úcomfy_kitchen.registryrT   ÚregisterÚ
__import__Ú__name__Ú__all__rP   rS   s    rO   Ú	_registerr_   ô   s+   € Ý/à×ÑØÜœ(¬WÑ5Ü'Ó)ð ò rQ   N)r^   Úawqr	   Úquantizationr   r   r   r
   r   r   r   r   r   Úroper   r   r   r   Úsvdquantr   r   ÚdictrP   r_   © rQ   rO   Ú<module>rf      sG   ðò€õ&  ÷
÷ 
õ 
÷ YÓ Xß EðM˜Dô Mò`ñ 
…rQ   