feat: add npu dockerfile and useage#428
Merged
HuangJoJo merged 2 commits intoalibaba:mainfrom Apr 30, 2026
Merged
Conversation
3f6baab to
03a6c22
Compare
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
PR: feat: add NPU Dockerfile and Ascend documentation
Summary
Add Docker build files and documentation for running ROLL on Huawei Ascend NPU (Atlas 900 A2/A3 PODc), enabling users to deploy RLVR and other pipelines on Ascend hardware.
Changes
Docker
Dockerfile.A2for Atlas 900 A2 PODc (Ascend 910B1, CANN 8.5.1)Dockerfile.A3for Atlas 900 A3 PODc (Ascend 910_9391, CANN 8.5.1)Documentation (English + Chinese)
ascend_docker_usage.md— Docker image build/pull, container startup, environment verification, and RLVR pipeline exampleascend_usage.md— Revise Ascend support status and installation instructionsExample Config
examples/ascend_examples/qwen3_8b_rlvr_deepspeed.yaml— Qwen3-8B RLVR config using DeepSpeed ZeRO-3 + CPU offloading on NPUFiles Changed (7 files, +808 / -68)
docker/Dockerfile.A2docker/Dockerfile.A3docs_roll/docs/User Guides/Hardware Support/ascend_docker_usage.mddocs_roll/docs/User Guides/Hardware Support/ascend_usage.mddocs_roll/i18n/.../ascend_docker_usage.mddocs_roll/i18n/.../ascend_usage.mdexamples/ascend_examples/qwen3_8b_rlvr_deepspeed.yamlKey Notes
flash_attnpackage is not supported; usefa2viatransformersinstead