Update amd docker image. (#3654)
This commit is contained in:
8
.github/workflows/pr-test-amd.yml
vendored
8
.github/workflows/pr-test-amd.yml
vendored
@@ -35,12 +35,12 @@ jobs:
|
|||||||
else
|
else
|
||||||
DEVICE_FLAG="--device /dev/dri"
|
DEVICE_FLAG="--device /dev/dri"
|
||||||
fi
|
fi
|
||||||
docker pull lmsysorg/sglang:v0.4.3-rocm630
|
docker pull lmsysorg/sglang:v0.4.3.post2-rocm630
|
||||||
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
||||||
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
||||||
--cap-add=SYS_PTRACE -e HF_TOKEN=${HF_TOKEN} --security-opt seccomp=unconfined \
|
--cap-add=SYS_PTRACE -e HF_TOKEN=${HF_TOKEN} --security-opt seccomp=unconfined \
|
||||||
-w /sglang-checkout --name ci_sglang \
|
-w /sglang-checkout --name ci_sglang \
|
||||||
lmsysorg/sglang:v0.4.3-rocm630
|
lmsysorg/v0.4.3.post2-rocm630
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
@@ -71,12 +71,12 @@ jobs:
|
|||||||
else
|
else
|
||||||
DEVICE_FLAG="--device /dev/dri"
|
DEVICE_FLAG="--device /dev/dri"
|
||||||
fi
|
fi
|
||||||
docker pull lmsysorg/sglang:v0.4.3-rocm630
|
docker pull lmsysorg/sglang:v0.4.3.post2-rocm630
|
||||||
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
docker run -dt --user root --device=/dev/kfd $DEVICE_FLAG \
|
||||||
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
-v ${{ github.workspace }}:/sglang-checkout --ipc=host --group-add video \
|
||||||
--cap-add=SYS_PTRACE -e HF_TOKEN=${{ secrets.AMD_HF_TOKEN }} --security-opt seccomp=unconfined \
|
--cap-add=SYS_PTRACE -e HF_TOKEN=${{ secrets.AMD_HF_TOKEN }} --security-opt seccomp=unconfined \
|
||||||
-w /sglang-checkout --name ci_sglang \
|
-w /sglang-checkout --name ci_sglang \
|
||||||
lmsysorg/sglang:v0.4.3-rocm630
|
lmsysorg/sglang:v0.4.3.post2-rocm630
|
||||||
|
|
||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
|
|||||||
Reference in New Issue
Block a user