
    3jx                         / S Qr SrSr SSKrSSKJrJrJrJrJ	r	  SSK
JrJrJrJr  S	\4S
 jrS r\" 5         g! \ a  rSrS\< 3r SrCN%SrCff = f))	
apply_ropeapply_rope1apply_rope_split_halfapply_rope_split_half1dequantize_nvfp4dequantize_per_tensor_fp8quantize_mxfp8quantize_nvfp4quantize_per_tensor_fp8TN       )r   r   r   r	   r
   )r   r   r   r   FzImportError: returnc                     SS K n SSKJnJnJn  [        S15      n[        SS1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 R                  15      S9U" U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" SU" UU" S5      4S	90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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S.	$ )Nr   )	ExactDimsFunctionConstraintsParamConstraintcudaxpu)dtypes)xscaleoutput_type)paramsdefault_devices   )r   shape_rules)r   per_tensor_scaler   )qxr   block_scalesr   )
   r   )r   r   min_compute_capability)r   	freqs_cis)xqxkr!   )	r
   r   r	   r   r   r   r   r   r   )torchcomfy_kitchen.constraintsr   r   r   	frozensetfloat32float16bfloat16float8_e4m3fnfloat8_e5m2uint8)r$   r   r   r   cuda_devicestriton_devicesstandard_floatss          `/home/wildlama/miniconda3/lib/python3.13/site-packages/comfy_kitchen/backends/triton/__init__.py_build_constraintsr1   !   sl     fX&L/Nu~~ NOO $7$O<(	5==/0JK.$e&9&95;L;L%MN  +	$
 &9$$e&9&95;L;L%MN )	5==/0JK.oF +	&
 .$*!*1 %49emm_;U$V )	
 ._*!*1 )
 0%$ekk]3!*1 %49emm_;U$V /$e&9&9%:;!  /oF
 )#*
 +$O<,OD +
 *%_=%_=,OD
 +
 #6$O<,OD +#
 "5%_=%_=,OD
 +"
]V V    c                     SS K n SSKJn  [        (       d   UR	                  S[
        =(       d    S5        g U R                  R                  5       n[        U S5      =(       a    U R                  R                  5       nU(       d  U(       d  UR	                  SS5        g UR                  S[        [        [        S9[        5       S9  g )	Nr   )registrytritonzTriton not availabler   z-Neither CUDA nor XPU available on this system)fromlist)namemodulecapabilities)r$   comfy_kitchen.registryr4   _TRITON_AVAILABLEmark_unavailable_TRITON_ERRORr   is_availablehasattrr   register
__import____name____all__r1   )r$   r4   has_cudahas_xpus       r0   	_registerrF      s    /!!(M,S=STzz&&(HeU#@		(>(>(@GG!!(,[\(W5')  r2   )rC   r;   r=   r5   quantizationr   r   r   r	   r
   roper   r   r   r   ImportErroredictr1   rF    r2   r0   <module>rM      st   
  *  ]\cD cL. 
E  *#A5)M*s   : A AA