Add Neuron to the distcheck matrix

Signed-off-by: Raghu Raja <[email protected]>

.github/workflows/distcheck.yaml

@@ -16,16 +16,36 @@ jobs:
         cc:
           - gcc
           - clang
+        sdk:
+          - cuda
+          - neuron
       fail-fast: false
     steps:
       - name: Install Dependencies
         run: |
+          sudo apt-get update -y
           sudo apt-get install -y ${{ env.APT_PACKAGES }}
-      - name: Install CUDA
+      - name: Install CUDA SDK
+        if: matrix.sdk == 'cuda'
         run: |
           sudo apt-get install -y nvidia-cuda-toolkit
+      - name: Install Neuron SDK
+        if: matrix.sdk == 'neuron'
+        run: |
+          # Configure Linux for Neuron repository updates
+          . /etc/os-release
+          sudo tee /etc/apt/sources.list.d/neuron.list > /dev/null <<EOF
+          deb https://apt.repos.neuron.amazonaws.com ${VERSION_CODENAME} main
+          EOF
+          wget -qO - https://apt.repos.neuron.amazonaws.com/GPG-PUB-KEY-AMAZON-AWS-NEURON.PUB | sudo apt-key add -
+
+          # Install Neuron Runtime 
+          sudo apt-get install aws-neuronx-collectives=2.* -y
+          sudo apt-get install aws-neuronx-runtime-lib=2.* -y
+
       - uses: actions/checkout@v4
-      - name: Build and Distribution Checks
+      - name: Build and Distribution Checks (CUDA)
+        if: matrix.sdk == 'cuda'
         run: |
           set -x