| fix(attention): 修正三组 SMLA 及 metadata example 调用参数 Co-authored-by: wangzhaonan<wangzhaonan4@huawei.com> # message auto-generated for no-merge-commit merge: !14996 merge fix/attention-mla-examples-a5 into master fix(attention): 修正三组 SMLA 及 metadata example 调用参数 Created-by: wangzhaonan Commit-by: wangzhaonan Merged-by: cann-robot Description: ## 描述 修复以下六个算子的 example 调用参数:mixed_quant_sparse_flash_mla、mixed_quant_sparse_flash_mla_metadata、quant_sparse_flash_mla、quant_sparse_flash_mla_metadata、sparse_flash_mla、sparse_flash_mla_metadata。 - 三个计算 example 使用 INT32[2048] metadata,由 shape 计算 host 缓冲长度。 - 三个 metadata C++ example 修正2048长度、80/80核数打印布局,并按压缩比计算压缩长度。 - 三个 metadata Python example 统一实际长度、最大长度、压缩长度与余数,移除未使用的额外导入,增加同步及结果输出。 改动仅包含这六个算子的9个example文件。 ## 关联的Issue [#6995](https://gitcode.com/cann/ops-transformer/issues/6995) ## 测试 已从 wangzhaonan/ops-transformer_fix_xxSMLA_example 重新 fetch 并在独立干净checkout中检出本PR提交 e5310e18a16017a5cfab92f0ae64efaa241507aa。 - A5 Ascend950DT_9584(32 AIC / 64 AIV),docker_wzn / Conda wzn,已安装CANN 9.3.0(V100R001C13B023)。 - 6个ACLNN example均重新编译、执行、stream同步及拷贝完成,退出码全部为0。 - 3个metadata PyTorch example均执行、同步并输出结果,退出码全部为0。 - 9份运行日志均未出现161002。 - 当前提交 b2f89e82a46b6529564deac91f2bc5acfe924193 的完整 pre-commit 检查通过,覆盖本 PR 的9个example文件,包括 clang-format、Ruff、OAT 及基础检查。相比上述运行验证提交仅调整注释对齐,git diff -w 为空。 本次使用已安装CANN库验证,未从当前主线重编完整算子包,未做golden精度对比。 原始mixed_quant_sparse_flash_mla_metadata C++ example已复现161002,日志明确指出metadata长度1024,应为2048。 ## 文档更新 无。 ## 类型标签 - [x] 🐛 Bug修复 See merge request: cann/ops-transformer!14996 | 13 days ago |