mirror of
https://github.com/huggingface/text-generation-inference.git
synced 2025-04-20 22:32:07 +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 extra_pytest="-k test_flash_gemma_simple"
|
||||||
export target=""
|
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
|
esac
|
||||||
echo $dockerfile
|
echo $dockerfile
|
||||||
echo "Dockerfile=${dockerfile}"
|
echo "Dockerfile=${dockerfile}"
|
||||||
|
Loading…
Reference in New Issue
Block a user