Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
change
sglang
Commits
f9a377f6
Unverified
Commit
f9a377f6
authored
Nov 08, 2024
by
HAI
Committed by
GitHub
Nov 08, 2024
Browse files
[Release, ROCm] release ROCm docker build for AMD MI GPUs (#1957)
parent
4ade15dd
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
55 additions
and
0 deletions
+55
-0
.github/workflows/release-docker-amd.yml
.github/workflows/release-docker-amd.yml
+55
-0
No files found.
.github/workflows/release-docker-amd.yml
0 → 100644
View file @
f9a377f6
name
:
Release Docker Images (AMD)
on
:
push
:
branches
:
-
main
paths
:
-
"
python/sglang/version.py"
workflow_dispatch
:
jobs
:
publish
:
if
:
github.repository == 'sgl-project/sglang'
runs-on
:
docker-builder-amd
environment
:
'
prod'
strategy
:
matrix
:
rocm_version
:
[
'
6.2.0'
]
build_type
:
[
'
all'
,
'
srt'
]
steps
:
-
name
:
Delete huge unnecessary tools folder
run
:
rm -rf /opt/hostedtoolcache
-
name
:
Checkout repository
uses
:
actions/checkout@v3
-
name
:
Login to Docker Hub
uses
:
docker/login-action@v2
with
:
username
:
${{ secrets.DOCKERHUB_USERNAME }}
password
:
${{ secrets.DOCKERHUB_TOKEN }}
-
name
:
Build and Push
run
:
|
version=$(cat python/sglang/version.py | cut -d'"' -f2)
if [ "${{ matrix.rocm_version }}" = "6.2.0" ]; then
rocm_tag="rocm620"
else
echo "Unsupported ROCm version"
exit 1
fi
tag=v${version}-${rocm_tag}
if [ "${{ matrix.build_type }}" = "all" ]; then
tag_suffix=""
elif [ "${{ matrix.build_type }}" = "srt" ]; then
tag_suffix="-srt"
else
echo "Unsupported build type"
exit 1
fi
docker build . -f docker/Dockerfile.rocm --build-arg BUILD_TYPE=${{ matrix.build_type }} -t lmsysorg/sglang:${tag}${tag_suffix} --no-cache
docker push lmsysorg/sglang:${tag}${tag_suffix}
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment