diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 720a13cb..5b3c4a71 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -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}" diff --git a/Dockerfile.neuron b/Dockerfile_neuron similarity index 100% rename from Dockerfile.neuron rename to Dockerfile_neuron