
    i                     L   S SK JrJr  SSKJrJrJrJrJrJ	r	J
r
  SSKJrJr  SSKJr  SSKJrJrJr  SSKJrJr  SSKJrJrJrJrJrJrJrJrJ r J!r!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/J0r0J1r1J2r2  SS
K3J4r4J5r5J6r6J7r7J8r8J9r9J:r:J;r;  SSK<J=r=  SSK>J?r?J@r@  SSKAJBrB  \$rC/ SQrDg)    )int_scaled_matmulsafe_int_mm   )GranularityPerAxisPerBlockPerGroupPerRow	PerTensorPerToken)LinearActivationQuantizedTensorto_linear_activation_quantized)6to_weight_tensor_with_linear_activation_scale_metadata)Int4WeightOnlyQuantizerInt8DynActInt4WeightLinearInt8DynActInt4WeightQuantizer)AffineQuantizedMinMaxObserverAffineQuantizedObserverBase))Float8DynamicActivationFloat8WeightConfig'Float8DynamicActivationInt4WeightConfigFloat8MMConfig(Float8StaticActivationFloat8WeightConfigFloat8WeightOnlyConfigFqnToConfigInt4WeightOnlyConfig%Int8DynamicActivationInt8WeightConfig%Int8DynamicActivationIntxWeightConfig$Int8StaticActivationInt8WeightConfigInt8WeightOnlyConfigIntxWeightOnlyConfigModuleFqnToConfigPlainLayoutTensorCoreTiledLayout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) register_quantize_module_handler)	QuantizerTwoStepQuantizer)compute_error)8r&   r%   r$   r'   r   r   r   r   r   r   r   r   r   r    AOPerModuleConfigr   r!   r4   r2   r0   r1   r5   r6   r3   r/   r:   r   r   r   r   r+   r,   r.   r-   r   r   r7   r(   r*   r)   r   r   r   r   r	   r
   r   r   r   r   r   r9   r8   r"   r#   r   N)Etorchao.kernelr   r   granularityr   r   r   r	   r
   r   r   "linear_activation_quantized_tensorr   r   linear_activation_scaler   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%   r&   r'   quant_primitivesr(   r)   r*   r+   r,   r-   r.   quantize_.workflowsr/   r0   r1   r2   r3   r4   r5   r6   transform_moduler7   unifiedr8   r9   utilsr:   r;   __all__     \/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/torchao/quantization/__init__.py<module>rL      s   
   
     *  	 	 	 ? 0
 & ArJ   