feat: add neuron case to build ci

This commit is contained in:
drbh 2025-02-17 15:22:02 +00:00
parent 856d7682cf
commit 252f5468cc
2 changed files with 10 additions and 0 deletions

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="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}"