Skip to content

[feat] Support ascend mindspeed&megatron and add CI tests#456

Open
shun001 wants to merge 1 commit into
alibaba:mainfrom
shun001:ascend-mindspeed-ut
Open

[feat] Support ascend mindspeed&megatron and add CI tests#456
shun001 wants to merge 1 commit into
alibaba:mainfrom
shun001:ascend-mindspeed-ut

Conversation

@shun001
Copy link
Copy Markdown
Contributor

@shun001 shun001 commented Jun 4, 2026

This PR contains #380
Add MindSpeed-based NPU initialization patches, platform detection fixes, Megatron optimizer/offload compatibility updates, and NPU-specific attention mask handling. Also add an Ascend MindSpeed CI workflow, Qwen3 DPO Megatron example config, and update Megatron offload tests for NPU environments.

@shun001 shun001 force-pushed the ascend-mindspeed-ut branch 2 times, most recently from bfeee03 to a2bc3b2 Compare June 4, 2026 07:32
@shun001 shun001 force-pushed the ascend-mindspeed-ut branch from a2bc3b2 to 6288207 Compare June 4, 2026 07:38
@shun001 shun001 changed the title [feat] Add ascend mindspeed ut [feat] Support ascend mindspeed&megatron and add CI tests Jun 4, 2026
@UsernameFull
Copy link
Copy Markdown
Contributor

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants