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

feat: add neuron case to build ci

Browse Source
This commit is contained in:
drbh 2025-02-17 15:22:02 +00:00 committed by David Corvoysier
parent 3bcc523e76
commit bb51c5138c
2 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -114,6 +114,16 @@ jobs:
export extra_pytest="-k test_flash_gemma_simple"
export target=""
;;
neuron)
export dockerfile="Dockerfile.neuron"
export label_extension="-neuron"
export docker_devices="/dev/neuron0"
export docker_volume="/mnt/cache"
export runs_on="aws-inf2-8xlarge"
export platform="cpu"
export extra_pytest="--neuron"
export target=""
;;
esac
echo $dockerfile
echo "Dockerfile=${dockerfile}"

2
.github/workflows/ci_build.yaml vendored
View File

@ -37,7 +37,7 @@ jobs:
# fail-fast is true by default
fail-fast: false
matrix:
hardware: ["cuda", "cuda-trtllm", "rocm", "intel-xpu", "intel-cpu"]
hardware: ["cuda", "cuda-trtllm", "rocm", "intel-xpu", "intel-cpu", "neuron"]
uses: ./.github/workflows/build.yaml # calls the one above ^
permissions:
contents: write