[AMD] feat: add DLLM support for AMD GPUs with LLaDA2 testing (#15560)
This commit is contained in:
40
.github/workflows/pr-test-amd.yml
vendored
40
.github/workflows/pr-test-amd.yml
vendored
@@ -796,6 +796,43 @@ jobs:
|
||||
run: |
|
||||
bash scripts/ci/amd_ci_exec.sh -e SGLANG_USE_AITER_AR=0 -e SGLANG_USE_AITER=0 -e HF_HUB_ENABLE_HF_TRANSFER=0 python3 test_moe_eval_accuracy_large.py
|
||||
|
||||
dllm-test-1-gpu-amd:
|
||||
needs: [check-changes, stage-a-test-1-amd]
|
||||
if: |
|
||||
always() &&
|
||||
(
|
||||
(inputs.target_stage == 'dllm-test-1-gpu-amd') ||
|
||||
(
|
||||
!inputs.target_stage &&
|
||||
(!failure() && !cancelled()) &&
|
||||
((needs.check-changes.outputs.main_package == 'true') || (needs.check-changes.outputs.sgl_kernel == 'true'))
|
||||
)
|
||||
)
|
||||
strategy:
|
||||
fail-fast: false
|
||||
matrix:
|
||||
runner: [linux-mi325-gpu-1]
|
||||
runs-on: ${{matrix.runner}}
|
||||
steps:
|
||||
- name: Checkout code
|
||||
uses: actions/checkout@v4
|
||||
|
||||
- name: Ensure VRAM is clear
|
||||
run: bash scripts/ensure_vram_clear.sh rocm
|
||||
|
||||
- name: Start CI container
|
||||
run: bash scripts/ci/amd_ci_start_container.sh
|
||||
env:
|
||||
GITHUB_WORKSPACE: ${{ github.workspace }}
|
||||
|
||||
- name: Install dependencies
|
||||
run: bash scripts/ci/amd_ci_install_dependency.sh
|
||||
|
||||
- name: Test DLLM (LLaDA2)
|
||||
timeout-minutes: 30
|
||||
run: |
|
||||
bash scripts/ci/amd_ci_exec.sh python3 dllm/test_llada2_mini_amd.py
|
||||
|
||||
pr-test-amd-finish:
|
||||
needs:
|
||||
[
|
||||
@@ -814,7 +851,8 @@ jobs:
|
||||
performance-test-1-gpu-part-2-amd,
|
||||
performance-test-2-gpu-amd,
|
||||
accuracy-test-1-gpu-amd,
|
||||
accuracy-test-2-gpu-amd, # Temporarily disabled
|
||||
accuracy-test-2-gpu-amd,
|
||||
dllm-test-1-gpu-amd,
|
||||
]
|
||||
if: always()
|
||||
runs-on: ubuntu-latest
|
||||
|
||||
Reference in New Issue
Block a user