+
    "Ý³j¾  ã                   óê  € ^ RI HtHt ^RIHtHtHtHtHtH	t	H
t
 ^RIHtHtHt ^RIHtHt ^RIHtHtHtHtHtHtHtHtHtHtHtHtHtH t H!t!H"t"H#t# ^RI$H%t%H&t&H't'H(t(H)t)H*t*H+t+ ^RI,H-t-H.t.H/t/H0t0H1t1H2t2H3t3H4t4H5t5H6t6 ^RI7H8t8 ^R	I9H:t: ]t;. R
NRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNRNR NR!NR"NR#NR$NR%NR&NR'NR(NR)NR*NR+NR,NR-NR.NR/NR0NR1NR2NR3NR4NR5NR6NR7NR8NR9NR:NR;NR<Nt<R=# )>é    )Úint_scaled_matmulÚsafe_int_mm)ÚGranularityÚPerAxisÚPerBlockÚPerGroupÚPerRowÚ	PerTensorÚPerToken)ÚInt4WeightOnlyQuantizerÚInt8DynActInt4WeightLinearÚInt8DynActInt4WeightQuantizer)ÚAffineQuantizedMinMaxObserverÚAffineQuantizedObserverBase)Ú)Float8DynamicActivationFloat8WeightConfigÚ'Float8DynamicActivationInt4WeightConfigÚFloat8MMConfigÚ(Float8StaticActivationFloat8WeightConfigÚFloat8WeightOnlyConfigÚFqnToConfigÚInt4WeightOnlyConfigÚ%Int8DynamicActivationInt8WeightConfigÚ%Int8DynamicActivationIntxWeightConfigÚ$Int8StaticActivationInt8WeightConfigÚInt8WeightOnlyConfigÚIntxWeightOnlyConfigÚModuleFqnToConfigÚfqn_matches_fqn_configÚ intx_quantization_aware_trainingÚ	quantize_Úswap_conv2d_1x1_to_linear)ÚMappingTypeÚTorchAODTypeÚZeroPointDomainÚchoose_qparams_affineÚ"choose_qparams_affine_with_min_maxÚdequantize_affineÚquantize_affine)
ÚFloat8TensorÚInt4PlainInt32TensorÚInt4PreshuffledTensorÚ
Int4TensorÚInt4TilePackedTo4dTensorÚ
Int8TensorÚIntxOpaqueTensorÚIntxUnpackedToInt8TensorÚ	NF4TensorÚto_nf4)Ú register_quantize_module_handler)Úcompute_errorr    r   r   r!   r   r   r   r   r   r   r   r   r   r   ÚAOPerModuleConfigr   r   r.   r,   r*   r+   r/   r0   r-   r)   r1   r2   r4   r   r   r%   r&   r(   r'   r   r   r3   r"   r$   r#   r   r
   r   r   r   r	   r   r   r   r   r   N)=Útorchao.kernelr   r   Úgranularityr   r   r   r   r	   r
   r   Úlinear_quant_modulesr   r   r   Úobserverr   r   Ú	quant_apir   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   Úquant_primitivesr"   r#   r$   r%   r&   r'   r(   Úquantize_.workflowsr)   r*   r+   r,   r-   r.   r/   r0   r1   r2   Útransform_moduler3   Úutilsr4   r5   Ú__all__© ó    Ú^/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchao/quantization/__init__.pyÚ<module>rC      s¢  ð÷÷
÷ ñ ÷ñ ÷
÷÷ ÷ ÷ õ ÷&÷ ñ ÷÷ ÷ õ ?õð
 &Ð ð;àð;ð 'ð;ð ð	;ð
  ð;ð ,ð;ð ,ð;ð +ð;ð ð;ð .ð;ð ð;ð ð;ð 0ð;ð /ð;ð ð;ð  ð!;ð" ð#;ð$ ð%;ð( ð);ð* ð+;ð, ð-;ð. ð/;ð0 ð1;ð2 ð3;ð4 ð5;ð6 ð7;ð8 ð9;ð: ð;;ð< ð=;ð@ $ðA;ðB "ðC;ðF ðG;ðH )ðI;ðJ ðK;ðL ðM;ðP ðQ;ðR ðS;ðV 'ðW;ðZ ð[;ð\ ð];ð^ ð_;ð` ða;ðb ðc;ðd ðe;ðf ðg;ðh ði;ðj ðk;ðl ðm;ðn ðo;ðp $ðq;ðr !ðs;ðt ðu;‚rA   