Wwanggangguofix
032ac98d创建于 3 小时前历史提交
文件最后提交记录最后更新时间
feat(arch35): port causal_conv1d_bwd and layer_norm_gated_bwd kernels from flash-linear-attention into mindspeed_ops/arch35/triton/, add API wrappers with arch routing, and create ATK backward-only test suites for both operators 4 天前
feat: add chunk_gated_delta_rule_fwd_h operator Co-authored-by: LinShua<707894133@qq.com> # message auto-generated for no-merge-commit merge: !28 merge master_fwd_h into master feat: add chunk_gated_delta_rule_fwd_h operator Created-by: LinShua Commit-by: LinShua Merged-by: ascend-robot Description: ## What this PR does / why we need it? add chunk_gated_delta_rule_fwd_h operator [#related roadmap](https://gitcode.com/Ascend/MindSpeed-Ops/issues/1) ## Does this PR introduce any user-facing change? Reference Operator Markdown Description. ## How was this patch tested? UT and ATK ![image.png](https://raw.gitcode.com/user-images/assets/9612429/efc6d84c-5b47-46e5-8b4f-e47cffe169fe/image.png 'image.png') ![image.png](https://raw.gitcode.com/user-images/assets/9612429/0c0e8daa-1066-4034-9dfd-565b7e182c11/image.png 'image.png') See merge request: Ascend/MindSpeed-Ops!287 天前
fix 3 小时前
add ci& add ut Co-authored-by: shiyuan680<917935075@qq.com> 1 个月前
feat: prepare_wy_repr ops Co-authored-by: liuxi_<liuxi75@huawei.com> # message auto-generated for no-merge-commit merge: !20 merge prepare_wy_repr into master feat: prepare_wy_repr ops Created-by: liuxi_ Commit-by: liuxi_ Merged-by: ascend-robot Description: ## What this PR does / why we need it? prepare_wy_repr ops [#1](https://gitcode.com/Ascend/MindSpeed-Ops/issues/1) ## Does this PR introduce any user-facing change? 不涉及 ## How was this patch tested? TA 3.2.1 ut ![image.png](https://raw.gitcode.com/user-images/assets/9612429/e080ebd9-8a25-45ae-ad87-dccda27e1e77/image.png 'image.png') atk ![image.png](https://raw.gitcode.com/user-images/assets/9612429/78ebd5d6-8a38-4316-8ee3-ff1a055f75eb/image.png 'image.png') ![image.png](https://raw.gitcode.com/user-images/assets/9612429/2c25d24b-773d-43c0-a714-4461a6cf51e9/image.png 'image.png') See merge request: Ascend/MindSpeed-Ops!2018 天前