biondizzle/vllm
1
0
Fork 0
Code Issues Pull Requests Actions 2 Packages Projects Releases Wiki Activity

[AMD][Build] Add DeepEP to ROCm Dockerfile (#36086)

Browse Source 
Signed-off-by: Ryan Rock <ryan.rock@amd.com>
This commit is contained in:
Ryan Rock
2026-03-12 16:33:32 -05:00
committed by GitHub
parent cc8f1f4764
commit a79c1c2c80
2 changed files with 49 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
.buildkite/test-amd.yaml
View File

@@ -2071,6 +2071,14 @@ steps:
- pytest -v -s kernels/moe --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
parallelism: 2
- label: Kernels FP8 MoE Test
timeout_in_minutes: 60
mirror_hardwares: [amdexperimental, amdproduction]
agent_pool: mi325_2
optional: true
commands:
- pytest -v -s kernels/moe/test_deepep_moe.py
- label: Kernels Mamba Test # 31min
timeout_in_minutes: 45
mirror_hardwares: [amdexperimental, amdproduction]
@@ -3801,6 +3809,14 @@ steps:
- pytest -v -s kernels/moe --shard-id=$$BUILDKITE_PARALLEL_JOB --num-shards=$$BUILDKITE_PARALLEL_JOB_COUNT
parallelism: 2
- label: Kernels FP8 MoE Test
timeout_in_minutes: 60
mirror_hardwares: [amdexperimental, amdproduction]
agent_pool: mi355_2
optional: true
commands:
- pytest -v -s kernels/moe/test_deepep_moe.py
- label: Kernels Mamba Test # 31min
timeout_in_minutes: 45
mirror_hardwares: [amdexperimental, amdproduction]