mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-19 22:02:06 +00:00
feat: add neuron case to build ci
This commit is contained in:
parent
856d7682cf
commit
252f5468cc
10
.github/workflows/build.yaml
vendored
10
.github/workflows/build.yaml
vendored
@ -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="none"
|
||||
export docker_volume="/mnt/cache"
|
||||
export runs_on="aws-inf2-8xlarge"
|
||||
export platform="cpu"
|
||||
export extra_pytest="-k test_flash_llama_simple"
|
||||
export target=""
|
||||
;;
|
||||
esac
|
||||
echo $dockerfile
|
||||
echo "Dockerfile=${dockerfile}"
|
||||
|
Loading…
Reference in New Issue
Block a user