Skip to content

feat(vllm-tensorizer): Add DeepEP build with NVSHMEM support#149

Draft
kptkin wants to merge 1 commit into
mainfrom
katia/inf-210-deepep
Draft

feat(vllm-tensorizer): Add DeepEP build with NVSHMEM support#149
kptkin wants to merge 1 commit into
mainfrom
katia/inf-210-deepep

Conversation

@kptkin
Copy link
Copy Markdown

@kptkin kptkin commented Apr 8, 2026

Summary

  • Add DeepEP (deepseek-ai/DeepEP) to the vllm-tensorizer image for EP all2all communication on InfiniBand
  • Build from source with NVSHMEM 3.3.24 for IBGDA (InfiniBand GPUDirect Async) support
  • Enables --all2all-backend deepep_high_throughput in vLLM for MoE expert parallel workloads

Build args

  • DEEPEP_COMMIT: DeepEP git commit (default: 73b6ea4)
  • NVSHMEM_VER: NVSHMEM version (default: 3.3.24)

@kptkin kptkin force-pushed the katia/inf-210-deepep branch from 9f5e71c to ff1868f Compare April 8, 2026 18:59
@kptkin kptkin force-pushed the katia/inf-210-deepep branch from ff1868f to ae4dfbc Compare April 8, 2026 19:09
@github-actions
Copy link
Copy Markdown

github-actions Bot commented Apr 8, 2026

@kptkin Build complete, success: https://github.com/coreweave/ml-containers/actions/runs/24153470981
Image: ghcr.io/coreweave/ml-containers/vllm-tensorizer:katia-inf-210-deepep-ae4dfbc-v0.19.0

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.

1 participant