MarlinQQQTensor¶
- class torchao.dtypes.MarlinQQQTensor(tensor_impl: AQTTensorImpl, block_size: Tuple[int, ...], shape: Size, quant_min: Optional[Union[int, float]] = None, quant_max: Optional[Union[int, float]] = None, zero_point_domain: ZeroPointDomain = ZeroPointDomain.INT, dtype=None, strides=None)[源代码]¶
MarlinQQQ 量化张量子类,继承自 AffineQuantizedTensor 类。
要了解 _choose_qparams_and_quantize_affine_qqq、Marlin QQQ 量化的量化和反量化过程中发生的情况,请查看 https://github.com/pytorch/ao/blob/main/torchao/quantization/quant_primitives.py,并检查两个量化原始操作:_choose_qparams_and_quantize_affine_qqq 和 _dequantize_affine_qqq。