Initial Enablement of CI on MI300 (#3168)
This commit is contained in:
13
.github/workflows/pr-test-amd.yml
vendored
13
.github/workflows/pr-test-amd.yml
vendored
@@ -20,7 +20,12 @@ concurrency:
|
|||||||
jobs:
|
jobs:
|
||||||
accuracy-test-1-gpu:
|
accuracy-test-1-gpu:
|
||||||
if: github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request'
|
if: github.repository == 'sgl-project/sglang' || github.event_name == 'pull_request'
|
||||||
runs-on: 1-gpu-runner-amd
|
runs-on: linux-mi300-gpu-1
|
||||||
|
container:
|
||||||
|
image: lmsysorg/sglang:v0.4.2.post2-rocm630
|
||||||
|
options: --user root --device=/dev/kfd --device=/dev/dri --ipc=host --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined
|
||||||
|
env:
|
||||||
|
HF_TOKEN: ${{ secrets.HF_TOKEN }}
|
||||||
steps:
|
steps:
|
||||||
- name: Checkout code
|
- name: Checkout code
|
||||||
uses: actions/checkout@v3
|
uses: actions/checkout@v3
|
||||||
@@ -28,8 +33,10 @@ jobs:
|
|||||||
- name: Install dependencies
|
- name: Install dependencies
|
||||||
run: |
|
run: |
|
||||||
pip install --upgrade pip
|
pip install --upgrade pip
|
||||||
pip install -e "python[runtime_common, test]"
|
cd sgl-kernel
|
||||||
pip install -e "python" --no-deps
|
python3 setup_rocm.py install
|
||||||
|
cd ..
|
||||||
|
pip install -e "python[dev_hip]"
|
||||||
|
|
||||||
git clone https://github.com/merrymercy/human-eval.git
|
git clone https://github.com/merrymercy/human-eval.git
|
||||||
cd human-eval
|
cd human-eval
|
||||||
|
|||||||
Reference in New Issue
Block a user