Star855

Star855

cann-robotfix anti_mx_quant

da9ca691创建于 6月25日历史提交

文件	最后提交记录	最后更新时间
docs	Delete redundant spaces in the .md files in the activation dirrectory Co-authored-by: zhouwenfang<zhouwenfang3@h-partners.com> # message auto-generated for no-merge-commit merge: !6045 merge master into master Delete redundant spaces in the .md files in the activation dirrectory Created-by: zhouwenfang Commit-by: zhouwenfang Merged-by: cann-robot Description: ## 描述 Delete redundant spaces in the .md files in the activation dirrectory ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/3317 ## 测试 NA ## 文档更新 *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!6045	2 个月前
examples	New op AntiMxQuant Co-authored-by: yeeyanghsia<xiayiyang@huawei.com> # message auto-generated for no-merge-commit merge: !4545 merge AntiMxQuant0429 into master New op AntiMxQuant Created-by: yeeyanghsia Commit-by: yeeyanghsia Merged-by: cann-robot Description: ## 描述新增AntiMxQuant算子，将调用 DynamicMxQuant 量化得到的 FLOAT4/FLOAT8 的 Tensor 反量化为 FLOAT16/BFLOAT16/FLOAT32 格式。 ## 关联的Issue 关联Issue [#2905](https://gitcode.com/cann/ops-nn/issues/2905) ## 测试本地通过精度测试，通过算子冒烟测试。 ## 文档更新涉及docs/zh/op_api_list.md、docs/zh/op_list.md、quant/anti_mx_quant/README.md、quant/anti_mx_quant/docs/aclnnAntiMxQuant.md。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4545	2 个月前
op_api	AntiMxQuant warning and bug fix Co-authored-by: yeeyanghsia<xiayiyang@huawei.com> # message auto-generated for no-merge-commit merge: !5754 merge AntiMxQuant0605 into master AntiMxQuant warning and bug fix Created-by: yeeyanghsia Commit-by: yeeyanghsia Merged-by: cann-robot Description: ## 描述修复AntiMxQuant算子部分异常用例拦截失败和语言规则告警问题。 ## 关联的Issue 关联Issue [#3144](https://gitcode.com/cann/ops-nn/issues/3144) ## 测试通过在线compile、ttk测试、冒烟验证。 ## 文档更新不涉及。 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!5754	2 个月前
op_graph	New op AntiMxQuant Co-authored-by: yeeyanghsia<xiayiyang@huawei.com> # message auto-generated for no-merge-commit merge: !4545 merge AntiMxQuant0429 into master New op AntiMxQuant Created-by: yeeyanghsia Commit-by: yeeyanghsia Merged-by: cann-robot Description: ## 描述新增AntiMxQuant算子，将调用 DynamicMxQuant 量化得到的 FLOAT4/FLOAT8 的 Tensor 反量化为 FLOAT16/BFLOAT16/FLOAT32 格式。 ## 关联的Issue 关联Issue [#2905](https://gitcode.com/cann/ops-nn/issues/2905) ## 测试本地通过精度测试，通过算子冒烟测试。 ## 文档更新涉及docs/zh/op_api_list.md、docs/zh/op_list.md、quant/anti_mx_quant/README.md、quant/anti_mx_quant/docs/aclnnAntiMxQuant.md。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4545	2 个月前
op_host	fix anti_mx_quant Co-authored-by: peng_yq<pengyongqiang5@huawei.com> # message auto-generated for no-merge-commit merge: !6511 merge AntiMxQuantDTSc11 into 9.1.0 fix anti_mx_quant Created-by: pyq17870581669 Commit-by: peng_yq Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 当axis参数输入是非尾轴时，有概率出现aic error，添加对axis是非尾轴场景的拦截 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/3547 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 本地压测问题无复现，本地精度测试。 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!6511	1 个月前
op_kernel	New op AntiMxQuant Co-authored-by: yeeyanghsia<xiayiyang@huawei.com> # message auto-generated for no-merge-commit merge: !4545 merge AntiMxQuant0429 into master New op AntiMxQuant Created-by: yeeyanghsia Commit-by: yeeyanghsia Merged-by: cann-robot Description: ## 描述新增AntiMxQuant算子，将调用 DynamicMxQuant 量化得到的 FLOAT4/FLOAT8 的 Tensor 反量化为 FLOAT16/BFLOAT16/FLOAT32 格式。 ## 关联的Issue 关联Issue [#2905](https://gitcode.com/cann/ops-nn/issues/2905) ## 测试本地通过精度测试，通过算子冒烟测试。 ## 文档更新涉及docs/zh/op_api_list.md、docs/zh/op_list.md、quant/anti_mx_quant/README.md、quant/anti_mx_quant/docs/aclnnAntiMxQuant.md。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4545	2 个月前
tests	New op AntiMxQuant Co-authored-by: yeeyanghsia<xiayiyang@huawei.com> # message auto-generated for no-merge-commit merge: !4545 merge AntiMxQuant0429 into master New op AntiMxQuant Created-by: yeeyanghsia Commit-by: yeeyanghsia Merged-by: cann-robot Description: ## 描述新增AntiMxQuant算子，将调用 DynamicMxQuant 量化得到的 FLOAT4/FLOAT8 的 Tensor 反量化为 FLOAT16/BFLOAT16/FLOAT32 格式。 ## 关联的Issue 关联Issue [#2905](https://gitcode.com/cann/ops-nn/issues/2905) ## 测试本地通过精度测试，通过算子冒烟测试。 ## 文档更新涉及docs/zh/op_api_list.md、docs/zh/op_list.md、quant/anti_mx_quant/README.md、quant/anti_mx_quant/docs/aclnnAntiMxQuant.md。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4545	2 个月前
CMakeLists.txt	New op AntiMxQuant Co-authored-by: yeeyanghsia<xiayiyang@huawei.com> # message auto-generated for no-merge-commit merge: !4545 merge AntiMxQuant0429 into master New op AntiMxQuant Created-by: yeeyanghsia Commit-by: yeeyanghsia Merged-by: cann-robot Description: ## 描述新增AntiMxQuant算子，将调用 DynamicMxQuant 量化得到的 FLOAT4/FLOAT8 的 Tensor 反量化为 FLOAT16/BFLOAT16/FLOAT32 格式。 ## 关联的Issue 关联Issue [#2905](https://gitcode.com/cann/ops-nn/issues/2905) ## 测试本地通过精度测试，通过算子冒烟测试。 ## 文档更新涉及docs/zh/op_api_list.md、docs/zh/op_list.md、quant/anti_mx_quant/README.md、quant/anti_mx_quant/docs/aclnnAntiMxQuant.md。 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!4545	2 个月前
README.md	Delete redundant spaces in the .md files in the activation dirrectory Co-authored-by: zhouwenfang<zhouwenfang3@h-partners.com> # message auto-generated for no-merge-commit merge: !6045 merge master into master Delete redundant spaces in the .md files in the activation dirrectory Created-by: zhouwenfang Commit-by: zhouwenfang Merged-by: cann-robot Description: ## 描述 Delete redundant spaces in the .md files in the activation dirrectory ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/3317 ## 测试 NA ## 文档更新 *.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!6045	2 个月前

AntiMxQuant

产品支持情况

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

功能说明

算子功能：将调用DynamicMxQuant量化得到的FLOAT4/FLOAT8的Tensor反量化为FLOAT16/BFLOAT16/FLOAT32格式。
反量化公式：

$Xdq=Xq×2sf−biasX_{dq} = X_q \times 2^{sf - bias}$
- 其中 $s f$ 是缩放因子，由输入mxscale提供； $b i a s$ 是指数位的偏移，对于FLOAT8_E8M0格式， $b i a s = 127$ ； $X_q$ 是量化得到的FLOAT4/FLOAT8张量； $X_{dq}$ 是反量化得到的FLOAT16/BFLOAT16/FLOAT32张量。

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
x	输入	待反量化的数据。由DynamicMxQuant量化得到的FLOAT4/FLOAT8张量。	FLOAT4_E2M1、FLOAT4_E1M2、FLOAT8_E4M3FN、FLOAT8_E5M2	ND
mxscale	输入	调用DynamicMxQuant计算得到的量化尺度。	FLOAT8_E8M0	ND
axis	属性	指定反量化轴。	INT64	-
dst_type	属性	指定反量化后输出y的类型。	INT64	-
y	输出	输入x反量化后的对应结果。shape与x一致。	FLOAT16、BFLOAT16、FLOAT32	ND

约束说明

关于x、mxscale的shape约束说明如下：
- 如果输入x的数据类型是float4_e2m1或float4_e1m2，x.shape[-1]必须是偶数。
- axis_change = axis if axis >= 0 else axis + rank(x)。
- mxscale.shape[axis_change] = (ceil(x.shape[axis] / 32) + 2 - 1) / 2。
- mxscale.shape[-1] = 2。
- rank(mxscale) = rank(x) + 1。
- 其它维度与输入x一致。

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_anti_mx_quant	通过aclnnAntiMxQuant接口方式调用AntiMxQuant算子。