o ãÊiåã@sXdZddlmZmZmZddlZddlmZeGdd„deƒƒZeGdd„deƒƒZ dS) z7Protocols for some functionalities in tensor subclassesé)ÚOptionalÚProtocolÚruntime_checkableN)ÚQuantizeTensorKwargsc@s eZdZUdZeejed<dS)ÚSupportsActivationPreScalingaProtocol for activation scale that should be multiplied with activation before quantization, or before we use activation in matrix multiplications, used for algorithms like AWQ A class that have `act_pre_scale: Optional[torch.Tensor]` attribute implements the Protocol Ú act_pre_scaleN)Ú__name__Ú __module__Ú __qualname__Ú__doc__rÚtorchÚTensorÚ__annotations__©rrúb/home/ubuntu/.local/lib/python3.10/site-packages/torchao/quantization/quantize_/common/protocol.pyrs rc@s.eZdZUdZeejed<defdd„Z dS)ÚIsStaticQuantizationConfigzÖProtocol for static quantization configuration. A class that has `act_quant_scale: Optional[torch.Tensor]` attribute and `get_act_quant_kwargs() -> QuantizeTensorKwargs` method implements the Protocol Úact_quant_scaleÚreturncCsdS)Nr)ÚselfrrrÚget_act_quant_kwargs'sz/IsStaticQuantizationConfig.get_act_quant_kwargsN) rr r r rr r rrrrrrrrs r) r Útypingrrrr Ús