7月社区任务-MatmulGatherScatter算子开发任务书【免费下载链接】cann-ops-competitions本仓库用于 CANN 开源社区各类竞赛、开源课题、社区任务等课题发布、开发者作品提交和展示。项目地址: https://gitcode.com/cann/cann-ops-competitions基础信息技术标签算子开发适配硬件Ascend 950PR/Ascend 950DT开源仓地址https://link.gitcode.com/i/856c62aae5d86f4af2db7b738a9f625eCANN 版本算子开源仓指定版本开发语言Ascend C任务概述本算子实现融合 Gather/Scatter 的矩阵乘法Matmul with Gather/Scatter Fusion。通过索引向量从 A 矩阵中 Gather 出指定行的子集参与 Matmul 计算然后将计算结果通过同一索引向量 Scatter 写回 D 矩阵的对应行。该融合方式适用于推荐系统等需要对稀疏特征行进行 gather 后做矩阵乘的场景。算子公式$$ \mathbf{D}[\text{indices}[i], j] \sum_{k0}^{K-1} \mathbf{A}[\text{indices}[i], k] \cdot \mathbf{B}[k, j] $$等价向量化形式$\mathbf{D}[\text{indices}, :] \mathbf{A}[\text{indices}, :] \times \mathbf{B}$D 的其他行不在 indices 中的位置保持为零。输出矩阵 D 需先做置零初始化。核心开发要求及验收标准功能实现要求实现融合的 Gather A按行 Matmul Scatter D按行单次 kernel launch 完成全部计算。索引向量由调用方保证不重复、在合法范围内算子端不做校验。Gather A 和 Scatter D 使用同一索引向量。输出 D 需先置零仅索引行被写入。参数说明参数名输入/输出/属性描述使用说明数据类型数据格式维度(shape)非连续TensorA输入Matmul 左矩阵形状 (M, K)按 indices gather 行RowMajor 布局FP16ND2—B输入Matmul 右矩阵形状 (K, N)RowMajor 布局FP16ND2—indices输入行索引向量长度 J元素为 [0, M-1] 的不重复整数INT32ND1—D输出Matmul 输出矩阵形状 (M, N)RowMajor 布局仅 indices 行被写入其余保持 0FP16ND2—算子约束限制无测试标准需参考 CPU 精度标杆自行设计自验证用例覆盖不同 index_size 比例、多种问题规模。需额外验证非索引行保持为零。自验证报告完整、可复现所有测试用例执行通过。性能要求性能标杆为CUTLASS的36_gather_scatter_fusion算子整体性能需与 0.8 倍 GPUH100持平。精度要求算子计算精度需满足 AscendOpTest 工具默认阈值。文档规范要求算子设计文档需根据参考模板填写内容完整、格式规范且必须通过评审自验证报告需要覆盖所有功能场景参考xxx算子自验证报告含测试用例执行日志/截图、整体测试通过截图、性能数据截图可清晰指导算子使用与测试README 文档内容完整、规范。验收交付件1, 自测用例、测试结果报告、测试步骤指导文档2, 算子代码的私仓邀请链接、代码仓路径、分支、算子目录PR 申请合入测试通过后在 Catlass 代码仓提交 PR 申请申请将开发完成的算子合入https://link.gitcode.com/i/856c62aae5d86f4af2db7b738a9f625e具体目录参考该PRhttps://link.gitcode.com/i/856c62aae5d86f4af2db7b738a9f625e/pull/678 。参考资料文档类Ascend C算子开发文档、CATLASS创新样例开发流程指南课程类Ascend C在线课程参考样例https://link.gitcode.com/i/856c62aae5d86f4af2db7b738a9f625e/blob/master/examples/44_quant_matmul_full_loadA_tla环境获取开源仓提供100小时免费时长请不使用时及时关闭用时耗尽前请务必保存相关资料建议及时提交备份。使用 hidevlab notebook 算力https://hidevlab.huawei.com/online-develop-intro?fromhiascend如需额外环境资源请联系昇腾小助手。特别注意事项开发过程需严格遵循 Ascend C 编程规范及算子开发相关要求所有交付件需提前完成自验证确认符合验收标准后再提交验收申请开发前请务必阅读【社区任务】流程及注意事项会例行更新。【免费下载链接】cann-ops-competitions本仓库用于 CANN 开源社区各类竞赛、开源课题、社区任务等课题发布、开发者作品提交和展示。项目地址: https://gitcode.com/cann/cann-ops-competitions创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
CANN社区任务-MatmulGatherScatter算子开发
发布时间：2026/7/5 19:56:51
7月社区任务-MatmulGatherScatter算子开发任务书【免费下载链接】cann-ops-competitions本仓库用于 CANN 开源社区各类竞赛、开源课题、社区任务等课题发布、开发者作品提交和展示。项目地址: https://gitcode.com/cann/cann-ops-competitions基础信息技术标签算子开发适配硬件Ascend 950PR/Ascend 950DT开源仓地址https://link.gitcode.com/i/856c62aae5d86f4af2db7b738a9f625eCANN 版本算子开源仓指定版本开发语言Ascend C任务概述本算子实现融合 Gather/Scatter 的矩阵乘法Matmul with Gather/Scatter Fusion。通过索引向量从 A 矩阵中 Gather 出指定行的子集参与 Matmul 计算然后将计算结果通过同一索引向量 Scatter 写回 D 矩阵的对应行。该融合方式适用于推荐系统等需要对稀疏特征行进行 gather 后做矩阵乘的场景。算子公式$$ \mathbf{D}[\text{indices}[i], j] \sum_{k0}^{K-1} \mathbf{A}[\text{indices}[i], k] \cdot \mathbf{B}[k, j] $$等价向量化形式$\mathbf{D}[\text{indices}, :] \mathbf{A}[\text{indices}, :] \times \mathbf{B}$D 的其他行不在 indices 中的位置保持为零。输出矩阵 D 需先做置零初始化。核心开发要求及验收标准功能实现要求实现融合的 Gather A按行 Matmul Scatter D按行单次 kernel launch 完成全部计算。索引向量由调用方保证不重复、在合法范围内算子端不做校验。Gather A 和 Scatter D 使用同一索引向量。输出 D 需先置零仅索引行被写入。参数说明参数名输入/输出/属性描述使用说明数据类型数据格式维度(shape)非连续TensorA输入Matmul 左矩阵形状 (M, K)按 indices gather 行RowMajor 布局FP16ND2—B输入Matmul 右矩阵形状 (K, N)RowMajor 布局FP16ND2—indices输入行索引向量长度 J元素为 [0, M-1] 的不重复整数INT32ND1—D输出Matmul 输出矩阵形状 (M, N)RowMajor 布局仅 indices 行被写入其余保持 0FP16ND2—算子约束限制无测试标准需参考 CPU 精度标杆自行设计自验证用例覆盖不同 index_size 比例、多种问题规模。需额外验证非索引行保持为零。自验证报告完整、可复现所有测试用例执行通过。性能要求性能标杆为CUTLASS的36_gather_scatter_fusion算子整体性能需与 0.8 倍 GPUH100持平。精度要求算子计算精度需满足 AscendOpTest 工具默认阈值。文档规范要求算子设计文档需根据参考模板填写内容完整、格式规范且必须通过评审自验证报告需要覆盖所有功能场景参考xxx算子自验证报告含测试用例执行日志/截图、整体测试通过截图、性能数据截图可清晰指导算子使用与测试README 文档内容完整、规范。验收交付件1, 自测用例、测试结果报告、测试步骤指导文档2, 算子代码的私仓邀请链接、代码仓路径、分支、算子目录PR 申请合入测试通过后在 Catlass 代码仓提交 PR 申请申请将开发完成的算子合入https://link.gitcode.com/i/856c62aae5d86f4af2db7b738a9f625e具体目录参考该PRhttps://link.gitcode.com/i/856c62aae5d86f4af2db7b738a9f625e/pull/678 。参考资料文档类Ascend C算子开发文档、CATLASS创新样例开发流程指南课程类Ascend C在线课程参考样例https://link.gitcode.com/i/856c62aae5d86f4af2db7b738a9f625e/blob/master/examples/44_quant_matmul_full_loadA_tla环境获取开源仓提供100小时免费时长请不使用时及时关闭用时耗尽前请务必保存相关资料建议及时提交备份。使用 hidevlab notebook 算力https://hidevlab.huawei.com/online-develop-intro?fromhiascend如需额外环境资源请联系昇腾小助手。特别注意事项开发过程需严格遵循 Ascend C 编程规范及算子开发相关要求所有交付件需提前完成自验证确认符合验收标准后再提交验收申请开发前请务必阅读【社区任务】流程及注意事项会例行更新。【免费下载链接】cann-ops-competitions本仓库用于 CANN 开源社区各类竞赛、开源课题、社区任务等课题发布、开发者作品提交和展示。项目地址: https://gitcode.com/cann/cann-ops-competitions创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考
