diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index d9f58667..0863645d 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="/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}" diff --git a/.github/workflows/ci_build.yaml b/.github/workflows/ci_build.yaml index d8746b65..b5487191 100644 --- a/.github/workflows/ci_build.yaml +++ b/.github/workflows/ci_build.yaml @@ -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