| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
docs: weight_quant_preprocess 补充 A16F4/A16MXF4 数据流文档与 example Co-authored-by: maqijun<maqijun@h-partners.com> # message auto-generated for no-merge-commit merge: !5765 merge docs/weight-quant-preprocess-a16f4 into master docs: weight_quant_preprocess 补充 A16F4/A16MXF4 数据流文档与 example Created-by: maqijun Commit-by: maqijun Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 为 weight_quant_preprocess 的 A16F4/A16MXF4 数据流补充 aclnn 文档与调用示例。A16F4(pergroup)与 A16MXF4(非转置 NZ + 转置 ND 直拷)数据流分别随 cann/ops-math!4495、cann/ops-math!5241 合入 master(对应 issue #3197),文档一直缺失,本 PR 补齐: - aclnnWeightQuantPreprocess.md: - 功能说明、161002 错误码说明补充 MM_A16F4/MM_A16MXF4 数据流 - 约束说明新增 MM_A16F4 数据流块(pergroup 唯一模式,非转置→FRACTAL_NZ_C0_16,scale 为 FLOAT16/BFLOAT16)与 MM_A16MXF4 数据流块(kGroupSize 固定 32,scale 为 FLOAT8_E8M0 {ceildiv(K,32),N};非转置→FRACTAL_NZ_C0_16,转置(末两维严格转置 stride [1,K],K 为偶数)→ND 直拷透传),字段结构沿用 MM_A16S4 块 - 调用示例章节新增 MM_A16F4/A16MXF4 示例小节(含完整内联源码,与 examples 中源码一致) - 新增 example test_aclnn_weight_quant_preprocess_a16f4.cpp:覆盖 A16F4 pergroup 非转置→NZ_C0_16、A16MXF4 非转置→NZ_C0_16、A16MXF4 转置→ND 直拷三个场景,各带 golden 比对(NZ 输出按 NZ_C0_16 物理布局 host 侧 bit-exact 比对,ND 直拷做 bit-wise 透传校验) ## 关联的Issue https://gitcode.com/cann/ops-math/issues/3197 ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> example 在 Ascend950 实机编译运行通过(K=256/N=128,pergroup gs=64 / MX gs=32,x 为 FLOAT16): - TestA16F4PerGroupNz / TestA16MXF4NonTransNz / TestA16MXF4TransNd 三个场景 golden 比对全部 PASS ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> 更新了:conversion/weight_quant_preprocess/docs/aclnnWeightQuantPreprocess.md ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [x] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5765 | 27 天前 |