mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-21 14:52:20 +00:00
.
This commit is contained in:
parent
65b2efc585
commit
fc4404d9d2
4
.github/workflows/build.yaml
vendored
4
.github/workflows/build.yaml
vendored
@ -52,9 +52,9 @@ jobs:
|
|||||||
export dockerfile="Dockerfile_amd"
|
export dockerfile="Dockerfile_amd"
|
||||||
export label_extension="-rocm"
|
export label_extension="-rocm"
|
||||||
export docker_devices="/dev/kfd,/dev/dri"
|
export docker_devices="/dev/kfd,/dev/dri"
|
||||||
export runs_on="ubuntu-latest"
|
export runs_on="[self-hosted, amd-gpu , multi-gpu]"
|
||||||
;;
|
;;
|
||||||
rocm)
|
intel)
|
||||||
export dockerfile="Dockerfile_intel"
|
export dockerfile="Dockerfile_intel"
|
||||||
export label_extension="-intel"
|
export label_extension="-intel"
|
||||||
export docker_devices=""
|
export docker_devices=""
|
||||||
|
2
.github/workflows/ci_build.yaml
vendored
2
.github/workflows/ci_build.yaml
vendored
@ -27,7 +27,7 @@ jobs:
|
|||||||
strategy:
|
strategy:
|
||||||
# super important if you want to see all results, even if one fails
|
# super important if you want to see all results, even if one fails
|
||||||
# fail-fast is true by default
|
# fail-fast is true by default
|
||||||
fail-fast: true
|
fail-fast: false
|
||||||
matrix:
|
matrix:
|
||||||
hardware: ["cuda", "rocm", "intel"]
|
hardware: ["cuda", "rocm", "intel"]
|
||||||
uses: ./.github/workflows/build.yaml # calls the one above ^
|
uses: ./.github/workflows/build.yaml # calls the one above ^
|
||||||
|
Loading…
Reference in New Issue
Block a user