aclnnFlatQuant

产品支持情况

[object Object]undefined

功能说明

• 接口功能：该融合算子为输入矩阵x一次进行两次小矩阵乘法，即右乘输入矩阵kroneckerP2，左乘输入矩阵kroneckerP1，然后针对矩阵乘的结果进行per-token量化处理。

• 计算公式：

  1.输入x右乘kroneckerP2：

  $$x' = x @ kroneckerP2$$

  2.kroneckerP1左乘x'：

  $$x'' = kroneckerP1@x'$$

  3.沿着x''的0维计算最大绝对值并除以(7 / clipRatio)以计算需量化为INT4格式的量化因子：

  $$quantScale = [max(abs(x''[0,:,:])),max(abs(x''[1,:,:])),...,max(abs(x''[K,:,:]))]/(7 / clipRatio)$$

  4.计算输出的out：

  $$out = x'' / quantScale$$

函数原型

每个算子分为，必须先调用[object Object]接口获取计算所需workspace大小以及包含了算子计算流程的执行器，再调用[object Object]接口执行计算。

aclnnFlatQuantGetWorkspaceSize

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

  第一段接口完成入参校验，出现以下场景时报错：

  [object Object]

aclnnFlatQuant

• 参数说明：

  [object Object]

• 返回值：

  aclnnStatus：返回状态码，具体参见。

约束说明

• 确定性计算：
  • aclnnFlatQuant默认确定性实现。

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考。