| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
grouped_dynamic_mx_quant算子支持group_index int64类型 Co-authored-by: TH<taohai2@huawei.com> # message auto-generated for no-merge-commit merge: !9023 merge grouped_dynamic_mx_quantAdaptINT64 into master grouped_dynamic_mx_quant算子支持group_index int64类型 Created-by: TH_HW Commit-by: TH Merged-by: cann-robot Description: ## 描述 使 GroupedDynamicMxQuant 算子（含 aclnnGroupedDynamicMxQuant 与 aclnnGroupedDynamicMxQuantV2 接口）的 groupIndex 输入不再仅支持 INT32，而是同时支持 INT32、INT64 两种数据类型，满足上层框架以 int64 索引调用算子的需求。同时保持对既有 INT32 场景的完整兼容。 主要改动如下： 1. **算子描述与接口校验**：op_graph/grouped_dynamic_mx_quant_proto.h 中 group_index 的 TensorType 由 {DT_Int32} 扩展为 {DT_Int32, DT_Int64}；op_api 与 op_host/op_api 头文件中补充 groupIndex 的 int64 支持说明；aclnn_grouped_dynamic_mx_quant.cpp / aclnn_grouped_dynamic_mx_quant_v2.cpp 的GROUP_INDEX_DTYPE_SUPPORT_LIST 增加 DT_INT64。 2. **Host 侧（Tiling/定义）**：tiling 校验集合 GROUPIDX_SUPPORT_DTYPE_SET 增加 DT_INT64，GroupedDynamicMxQuantTilingParam 新增 groupIndexType 字段，并在 SetTilingKeyParam 中将其编入 tiling key；grouped_dynamic_mx_quant_def.cpp 的数据类型组合表扩展为 int32/int64 与所有输入、输出类型的组合（共16组）。 3. **Binary 配置**：ascend950/grouped_dynamic_mx_quant_binary.json 为各 x 输入类型（fp16/bf16）与输出类型（fp8_e4m3fn、fp8_e5m2、fp4_e2m1、fp4_e1m2）组合补充了 group_index 为 int64 的 _int64 二进制算子配置。 4. **Kernel 侧**：GroupedDynamicMxQuantCombine 模板新增 GroupIndexT 类型参数，GlobalTensor<GroupIndexT> 读取分组索引；公共切分基类 GroupedSplitBase::GroupedSplit 同步模板化 GroupIndexT；grouped_dynamic_mx_quant_struct.h 新增 TPL_GROUP_INDEX_INT32/TPL_GROUP_INDEX_INT64 模板实参选择；grouped_dynamic_mx_quant.cpp 入口通过 if constexpr 依据 tiling key 中的 groupIndexType 分发实例化。 5. **文档更新**：README 与 aclnnGroupedDynamicMxQuant / aclnnGroupedDynamicMxQuantV2 接口文档中 groupIndex 支持类型由 INT32 更新为 INT32、INT64。 6. **单测适配**：test_grouped_dynamic_mx_quant.cpp 中 kernel 调用补充新增的 groupIndexType 模板实参。 ## 关联的Issue 无。 ## 测试 - 对 group_index 均为 int32、int64 时逐用例执行了 tests/ut/op_kernel/test_grouped_dynamic_mx_quant.cpp 中现有用例验证（bf16→fp8/fp4、fp16→fp8/fp4 等多种输入输出组合），结果均通过。 - 新增/回归验证了 tiling key 中 groupIndexType 的取整与算子 bin 选择在 asc 引擎上的绑定正确性。 ## 文档更新 更新了以下文档： - quant/grouped_dynamic_mx_quant/README.md - quant/grouped_dynamic_mx_quant/docs/aclnnGroupedDynamicMxQuant.md - quant/grouped_dynamic_mx_quant/docs/aclnnGroupedDynamicMxQuantV2.md ## 类型标签 <!-- [x] 表示选中 --> - [x] 新特性 - [ ] Bug修复 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [x] AI辅助编写 See merge request: cann/ops-nn!9023 | 21 天前 | |
重命名mxquant相关变量 Co-authored-by: ji-jun1<jijun1@huawei.com> # message auto-generated for no-merge-commit merge: !9285 merge rename_mx into master 重命名mxquant相关变量 Created-by: ji-jun1 Commit-by: ji-jun1 Merged-by: cann-robot Description: ## 描述 重命名mxquant相关变量，包括原型和md中的说明 ## 关联的Issue #5420 ## 测试 已通过st测试 ## 文档更新 更新了 quant/dynamic_mx_quant_with_dual_axis/README.md quant/dynamic_mx_quant_with_dual_axis/aclnnDynamicMxQuantWithDualAxis.md quant/dynamic_mx_quant_with_dual_axis/aclnnDynamicMxQuantWithDualAxisV2.md quant/grouped_dynamic_mx_quant_with_dual_axis/docs/aclnnGroupedDynamicMxQuantWithDualAxis.md ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： ## AI/Agent生成声明 <!-- [x] 表示选中 --> - [ ] AI辅助编写 See merge request: cann/ops-nn!9285 | 14 天前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 21 天前 | ||
| 14 天前 |