test mi300 node

.github/workflows/amd_workflow.yml

@@ -13,10 +13,18 @@ on:
 
 jobs:
   run:
-    runs-on: [amdgpu-mi250-x86-64]
+    runs-on: ${{ matrix.runs-on }}
+    strategy:
+      fail-fast: false
+      matrix:
+        include:
+          - name: mi300
+            runs-on: amdgpu-mi300-x86-64
+            venv-dir: ${{ github.workspace }}
+          - name: mi250
+            runs-on: amdgpu-mi300-x86-64
+            venv-dir: /groups/aig_sharks/pytorch_venv
     timeout-minutes: 10
-    env:
-      VENV_DIR: /groups/aig_sharks/pytorch_venv
     steps:
     - uses: actions/checkout@v3
     - name: Setup Python
@@ -35,8 +43,8 @@ jobs:
     - name: Setup Virtual Environment and Install Dependencies
       shell: bash
       run: |
-        python -m venv ${VENV_DIR}
-        source ${VENV_DIR}/bin/activate
+        python -m venv ${{ matrix.venv-dir }}
+        source ${{ matrix.venv-dir }}/bin/activate
         pip install --upgrade pip
 
         if [[ -n "${{ github.event.inputs.requirements }}" ]]; then