Skip to content

NVIDIA Job - 186d7de1-ef94-44e0-9159-e662a922d65f #486378

NVIDIA Job - 186d7de1-ef94-44e0-9159-e662a922d65f

NVIDIA Job - 186d7de1-ef94-44e0-9159-e662a922d65f #486378

name: NVIDIA PyTorch/CUDA Job
on:
workflow_dispatch:
inputs:
run_id:
description: 'Unique identifier for this run'
required: true
type: string
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
run-name: 'NVIDIA Job - ${{ github.event.inputs.run_id }}'
jobs:
run:
runs-on: [nvidia-docker-b200-8-x86-64]
timeout-minutes: 10
steps:
- uses: actions/checkout@v3
- name: Create input files
shell: bash
run: |
# Extract the payload content without printing it
apt-get update && apt-get install -y jq
PAYLOAD=$(jq -r '.inputs.payload' $GITHUB_EVENT_PATH)
# Apply mask to the extracted content
echo "::add-mask::$PAYLOAD"
# Now write to file (won't be logged since it's masked)
echo "$PAYLOAD" > payload.json
- name: Setup Virtual Environment and Install Dependencies
shell: bash
run: |
pip install --upgrade pip
pip install -r "requirements-dev.txt"
pip install -e .
- name: Run script
shell: bash
run: |
python3 src/runners/github-runner.py
- name: Upload training artifacts
uses: actions/upload-artifact@v4
if: always()
with:
name: run-result
path: result.json
- name: Upload profiling artifacts
uses: actions/upload-artifact@v4
if: always()
with:
name: profile-data
path: profile_data/*
retention-days: 1