QuantMax

产品支持情况

产品	是否支持
Ascend 950PR&950DT系列产品	√
Atlas A3系列产品	×
Atlas A2系列产品	×
Atlas 200I/500 A2推理产品	×
Atlas推理系列产品	×
Atlas训练系列产品	×

功能说明

• 算子功能：根据输入的scale对输入x进行量化，并计算输入x的绝对值的最大值amax。

• 计算公式：

  $$y=\text{Cast}(x\times scale,\text{dst\_type},\text{round\_mode})$$

  $$amax=\max (|x|)$$

  其中，$x$表示输入张量，$scale$表示量化缩放因子，$\text{dst\_type}$指定输出类型，$\text{round\_mode}$指定舍入模式，$amax$为输入张量所有元素绝对值的最大值。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待量化的输入张量，对应公式中的x。支持1-8维任意shape。	FLOAT、FLOAT16、BFLOAT16	ND
scale	输入	量化缩放因子，对应公式中的scale。shape固定为[1]。	FLOAT	ND
y	输出	量化后的输出张量，对应公式中的y。shape与x相同，数据类型由dst_type决定。	HIFLOAT8、FLOAT8_E5M2、FLOAT8_E4M3	ND
amax	输出	x绝对值的最大值，对应公式中的amax。shape固定为[1]，数据类型与x一致。	FLOAT、FLOAT16、BFLOAT16	ND
round_mode	属性	取值需为Cast API支持的有效舍入模式。	string	-
dst_type	属性	指定输出量化数据类型。34=HIFLOAT8，35=FLOAT8_E5M2，36=FLOAT8_E4M3。	int	-

约束说明

• round_mode与dst_dtype的匹配关系：
  • 当dst_dtype为35或36时，round_mode必须为"rint"。
  • 当dst_dtype为34时，round_mode必须为"round"或"hybrid"。

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_quant_max	通过aclnnQuantMax接口方式调用QuantMax算子。
图模式调用	-	通过算子IR构图方式调用QuantMax算子。