Skip to content

Commit

Permalink
Initial Enablement of CI on MI300 (#3168)
Browse files Browse the repository at this point in the history
  • Loading branch information
saienduri authored Feb 5, 2025
1 parent a322051 commit 04d8cd2
Showing 1 changed file with 10 additions and 3 deletions.
13 changes: 10 additions & 3 deletions .github/workflows/pr-test-amd.yml
Original file line number Diff line number Diff line change
Expand Up @@ -20,16 +20,23 @@ concurrency:
jobs:
accuracy-test-1-gpu:
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:
- name: Checkout code
uses: actions/checkout@v3

- name: Install dependencies
run: |
pip install --upgrade pip
pip install -e "python[runtime_common, test]"
pip install -e "python" --no-deps
cd sgl-kernel
python3 setup_rocm.py install
cd ..
pip install -e "python[dev_hip]"
git clone https://github.com/merrymercy/human-eval.git
cd human-eval
Expand Down

0 comments on commit 04d8cd2

Please sign in to comment.