Skip to content

(feat): add ROLL PPU dockerfile with SAIL pypi source, full megatron-core and FLA for Qwen3.5 - #489

Merged
HuangJoJo merged 1 commit into
alibaba:mainfrom
SendoRay:feat/ppu-dockerfile
Aug 24, 2026
Merged

(feat): add ROLL PPU dockerfile with SAIL pypi source, full megatron-core and FLA for Qwen3.5#489
HuangJoJo merged 1 commit into
alibaba:mainfrom
SendoRay:feat/ppu-dockerfile

Conversation

@SendoRay

@SendoRay SendoRay commented Aug 24, 2026

Copy link
Copy Markdown
Contributor

Summary

  • Add docker/Dockerfile.torch2100.vllm.ppu for PPU hardware: SAIL internal PyPI source, pinned PPU-custom torch/vllm/flash-attn builds via PIP_CONSTRAINT
  • Install full megatron-core (via mcore_adapter) to replace the base image's stripped stub
  • FLA for Qwen3.5 hybrid attention: back up PPU kda shims, install fla-org build, restore shims
  • Add docker/constraints_ppu.txt and requirements_torch2100_vllm_ppu.txt (transformers>=5.2.0 for qwen3_5)

Test plan

  • Build image and verify import checks in Dockerfile pass (torch/vllm/roll, transformers qwen3_5, megatron DDP + mcore_adapter, fla modules + kda shims)

@CLAassistant

CLAassistant commented Aug 24, 2026

Copy link
Copy Markdown

CLA assistant check
All committers have signed the CLA.

@SendoRay
SendoRay marked this pull request as draft August 24, 2026 10:04
@SendoRay
SendoRay force-pushed the feat/ppu-dockerfile branch from 969a08c to 7ed0f9c Compare August 24, 2026 10:06
@SendoRay
SendoRay marked this pull request as ready for review August 24, 2026 10:07
@HuangJoJo
HuangJoJo merged commit 5304da2 into alibaba:main Aug 24, 2026
1 check passed
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.

3 participants