Add Neuron to the distcheck matrix

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

.github/workflows/distcheck.yaml

@@ -2,6 +2,7 @@ name: PR CI
 on: [push, pull_request]
 env:
   APT_PACKAGES: >-
+    add-apt-repository
     build-essential
     clang
     gcc
@@ -16,16 +17,33 @@ 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 add-apt-repository 'deb https://apt.repos.neuron.amazonaws.com ${VERSION_CODENAME} main'
+          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-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