huggingface/text-generation-inference
5
0
Fork 0
mirror of https://github.com/huggingface/text-generation-inference.git synced 2025-06-07 18:02:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
Nicolas Patry 2024-06-07 20:18:57 +02:00
parent e79c83d7ba
commit eda299b84f
No known key found for this signature in database
GPG Key ID: 690EBED7128A9DC8
1 changed files with 14 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
.github/workflows/build.yaml vendored
View File

@ -42,24 +42,26 @@ jobs:
run: | run: |
case ${{ inputs.hardware }} in case ${{ inputs.hardware }} in
cuda) cuda)
dockerfile="Dockerfile" export dockerfile="Dockerfile"
label_extension="" export label_extension=""
docker_devices="" export docker_devices=""
runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]" export runs_on="[self-hosted, nvidia-gpu , multi-gpu, 4-a10, ci]"
;; ;;
rocm) rocm)
dockerfile="Dockerfile_amd" export dockerfile="Dockerfile_amd"
label_extension="-rocm" export label_extension="-rocm"
docker_devices="/dev/kfd,/dev/dri" export docker_devices="/dev/kfd,/dev/dri"
runs_on="ubuntu-latest" export runs_on="ubuntu-latest"
;; ;;
rocm) rocm)
dockerfile="Dockerfile_intel" export dockerfile="Dockerfile_intel"
label_extension="-intel" export label_extension="-intel"
docker_devices="" export docker_devices=""
runs_on="ubuntu-latest" export runs_on="ubuntu-latest"
;; ;;
esac esac
echo $dockerfile
echo $label_extension
echo "DOCKERFILE=${dockerfile}" >> $GITHUB_ENV echo "DOCKERFILE=${dockerfile}" >> $GITHUB_ENV
echo "LABEL=${label_extension}" >> $GITHUB_ENV echo "LABEL=${label_extension}" >> $GITHUB_ENV
echo "DOCKER_DEVICES=${docker_devices}" >> $GITHUB_ENV echo "DOCKER_DEVICES=${docker_devices}" >> $GITHUB_ENV