sort_kernel

文件	最后提交记录	最后更新时间
csrc	refactor: Remove SFA/SFAG/SLI Co-authored-by: JialiZheng<jializheng@huawei.com> # message auto-generated for no-merge-commit merge: !3591 merge master into master refactor: Remove SFA/SFAG/SLI Created-by: JialiZheng1 Commit-by: JialiZheng Merged-by: ascend-robot Description: 删除临时版本算子适配，相关功能通过正式算子承载 See merge request: Ascend/MindSpeed!3591	3 个月前
triton	fix(triton): harden sort_chunks_by_idx memory access	2 个月前
__init__.py	!371 MoE MLP not use mc2 linear and remove ops from init Merge pull request !371 from Jializheng/master	2 年前
dropout_add_layer_norm.py	!892 公开接口声明+Readme更新 Merge pull request !892 from TJJ/master	2 年前
dropout_add_rms_norm.py	!359 change ascendspeed to mindspeed Merge pull request !359 from 邓佳/master	2 年前
ffn.py	!937 add base st Merge pull request !937 from 王源升/master	2 年前
fusion_attention_v2.py	!892 公开接口声明+Readme更新 Merge pull request !892 from TJJ/master	2 年前
gmm.py	!1297 bugfix_GmmAdd_MmAdd Merge pull request !1297 from zengshu/master	1 年前
gmm_mxfp8.py	feat: add mxfp8 gmm Co-authored-by: EVA1<jingsiyu1@huawei.com> # message auto-generated for no-merge-commit merge: !3291 merge master into master feat: add mxfp8 gmm Created-by: EVA1 Commit-by: EVA1 Merged-by: ascend-robot Description: 添加MXFP8低精gmm模块 See merge request: Ascend/MindSpeed!3291	7 个月前
gmm_w4a8.py	feat: support W4A8-QAT Co-authored-by: xusiyang<xusiyang2@huawei.com> # message auto-generated for no-merge-commit merge: !3664 merge master into master feat: support W4A8-QAT Created-by: weixin_44492126 Commit-by: xusiyang Merged-by: ascend-robot Description: ## What this PR does / why we need it? 本PR 新增了对 W4A8-QAT（权重MXFP4/激活 MXFP8 量化感知训练，仅限 MoE 场景），并量化粒度分别支持：32(标准MXFP4量化粒度)/128(deepseek-V4采用量化粒度） ## Does this PR introduce any user-facing change? Yes. 用户可以通过在训练命令中添加以下参数来启用W4A8功能，默认量化粒度32： --qat-scheme w4a8-moe-only --qat-quant-block-size 128 ## How was this patch tested? 添加 --qat-scheme w4a8-moe-only 参数启动量化粒度为32的W4A8-QAT训练，添加--qat-quant-block-size 128启动量化粒度为128的W4A8-QAT训练 实现方案和实验结果如下：https://wiki.huawei.com/domains/171785/wiki/358154/WIKI2026071511870516 昇腾 950DT上测试通过：基于减层DeepSeek-V4，分别训练100step对比BF16 loss误差：量化粒度为32时loss误差在0.4%，量化粒度为128时loss误差在0.79% See merge request: Ascend/MindSpeed!3664	3 个月前
grouped_matmul.py	feat:mindspeed fsdp support qwen3vl fused ep Co-authored-by: ryanyeee<yezhenyuan@huawei.com> # message auto-generated for no-merge-commit merge: !3254 merge master_0205 into master feat:mindspeed fsdp support qwen3vl fused ep Created-by: ryanyeee Commit-by: ryanyeee Merged-by: ascend-robot Description: feat:mindspeed fsdp support qwen3vl fused ep See merge request: Ascend/MindSpeed!3254	8 个月前
grouped_matmul_mc2.py	feat:mindspeed fsdp Co-authored-by: ryanyeee<yezhenyuan@huawei.com> # message auto-generated for no-merge-commit merge: !3198 merge master into master feat:mindspeed fsdp Created-by: ryanyeee Commit-by: ryanyeee Merged-by: ascend-robot Description: feat:mindspeed fsdp See merge request: Ascend/MindSpeed!3198	9 个月前
lcal_functional.py	!1769 2dtp支持lcoc融合算子 Merge pull request !1769 from kirliavc/2dtp_backward	1 年前
npu_all_to_all_all_gather_bmm.py	!1017 Mc2 Moe Op Bug Fix Merge pull request !1017 from yayahello/master	1 年前
npu_apply_fused_adamw_v2.py	!2341 add swap optimizer to core_r0.12.0 Merge pull request !2341 from wangyuansheng8/master	1 年前
npu_apply_fused_ema_adamw.py	!1603 add ema adamw optimizer Merge pull request !1603 from wangyuansheng8/master	1 年前
npu_bmm_reduce_scatter_all_to_all.py	!954 update moe MC2 torch api Merge pull request !954 from 陈建军/master	2 年前
npu_grouped_mat_mul_all_reduce.py	!937 add base st Merge pull request !937 from 王源升/master	2 年前
npu_groupmatmul_add.py	refactor(ops): generalize NPU version detection with NPUVersion enum Co-authored-by: Muu<koimuu@163.com> # message auto-generated for no-merge-commit merge: !3544 merge dev into master refactor(ops): generalize NPU version detection with NPUVersion enum Created-by: Muuyo Commit-by: Muu Merged-by: ascend-robot Description: refactor(ops): generalize NPU version detection with NPUVersion enum See merge request: Ascend/MindSpeed!3544	4 个月前
npu_matmul_add.py	feat: add A2G3 device type support across codebase Co-authored-by: fanlu5<fanlu5@huawei.com> # message auto-generated for no-merge-commit merge: !3587 merge master into master feat: add A2G3 device type support across codebase Created-by: fanlu5 Commit-by: fanlu5 Merged-by: ascend-robot Description: ## What this PR does / why we need it? feat: add A2G3 device type support across codebase. 添加A2G3设备类型的适配支持，将其与现有的910B设备同等对待，覆盖版本检测、搜索空间裁剪、通信硬件信息、MoE相关逻辑等多个模块 ## Does this PR introduce any user-facing change? NA ## How was this patch tested? NA See merge request: Ascend/MindSpeed!3587	3 个月前
npu_mm_all_reduce_add_rms_norm.py	!937 add base st Merge pull request !937 from 王源升/master	2 年前
npu_mm_all_reduce_add_rms_norm_.py	!937 add base st Merge pull request !937 from 王源升/master	2 年前
npu_moe_token_permute.py	!892 公开接口声明+Readme更新 Merge pull request !892 from TJJ/master	2 年前
npu_moe_token_unpermute.py	!892 公开接口声明+Readme更新 Merge pull request !892 from TJJ/master	2 年前
npu_ring_attention_update.py	!892 公开接口声明+Readme更新 Merge pull request !892 from TJJ/master	2 年前
npu_rotary_position_embedding.py	!892 公开接口声明+Readme更新 Merge pull request !892 from TJJ/master	2 年前
quant_gmm.py	!968 Add weight quantize gmm ops Merge pull request !968 from 洪炜杰/hong0927	1 年前
weight_quant_gmm.py	!968 Add weight quantize gmm ops Merge pull request !968 from 洪炜杰/hong0927	1 年前