amd_workflow.yml

name: AMD PyTorch Job
on:
  workflow_dispatch:
    inputs:
      payload:
        description: 'Content of the user submission, as json string'
        required: true
        type: string
      requirements:
        description: 'Contents for a requirements.txt file'
        required: false
        type: string

jobs:
  run:
    runs-on: [amdgpu-mi250-x86-64]
    timeout-minutes: 10
    env:
      VENV_DIR: /groups/aig_sharks/pytorch_venv
    steps:
    - uses: actions/checkout@v3
    - name: Setup Python
      uses: actions/setup-python@v5
      with:
        python-version: '3.10'


    - name: Create input files
      shell: bash
      run: |
        cat > "payload.json" <<'EOL'
        ${{ github.event.inputs.payload }}
        EOL

    - name: Setup Virtual Environment and Install Dependencies
      shell: bash
      run: |
        python -m venv ${VENV_DIR}
        source ${VENV_DIR}/bin/activate
        pip install --upgrade pip

        if [[ -n "${{ github.event.inputs.requirements }}" ]]; then
          cat > "requirements.txt" <<'EOL'
          ${{ github.event.inputs.requirements }}
        EOL
        pip install -r "requirements.txt"
        fi
    - name: Run script
      shell: bash
      run: |
        python3 .github/workflows/runner.py
        cat result.json  # Debug: show output

    - name: Upload training artifacts
      uses: actions/upload-artifact@v4
      if: always()
      with:
        name: run-result
        path: |
          result.json