Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
12 changes: 9 additions & 3 deletions .github/workflows/collect-results.yml
Original file line number Diff line number Diff line change
Expand Up @@ -19,6 +19,14 @@ jobs:
token: ${{ secrets.REPO_PAT }}
fetch-depth: 0

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install -q matplotlib

- name: Download JSON artifacts
uses: actions/download-artifact@v6
with:
Expand All @@ -38,9 +46,7 @@ jobs:
path: agg_${{ inputs.exp-name || 'all' }}.json

- name: Plot performance
run: |
pip install -q matplotlib
python3 utils/plot_perf.py results/ ${{ inputs.exp-name || 'all' }}
run: python3 utils/plot_perf.py results/ ${{ inputs.exp-name || 'all' }}
- name: Upload performance graphs
uses: actions/upload-artifact@v5
with:
Expand Down
20 changes: 16 additions & 4 deletions .github/workflows/e2e-tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -18,9 +18,16 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

- id: get-jobs
run: |
pip install pydantic
CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py ${{ inputs.generate-cli-command }})
echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT

Expand Down Expand Up @@ -70,15 +77,20 @@ jobs:
token: ${{ secrets.REPO_PAT }}
fetch-depth: 0

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install PyGithub

- name: Download results artifacts
uses: actions/download-artifact@v6
with:
path: ${{ env.RESULTS_DIR }}
pattern: results_*

- name: Install python dependencies
run: pip install PyGithub

- name: Calculate success rate
run: python3 utils/calc_success_rate.py $STATS_FILENAME

Expand Down
29 changes: 24 additions & 5 deletions .github/workflows/full-sweep-1k1k-scheduler.yml
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,16 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

- id: get-dsr1-configs
run: |
pip install pydantic
CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py full-sweep --config-files ${GITHUB_WORKSPACE}/.github/configs/nvidia-master.yaml ${GITHUB_WORKSPACE}/.github/configs/amd-master.yaml --seq-lens 1k1k --model-prefix dsr1)
echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT

Expand All @@ -28,9 +35,16 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

- id: get-gptoss-configs
run: |
pip install pydantic
CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py full-sweep --config-files ${GITHUB_WORKSPACE}/.github/configs/nvidia-master.yaml ${GITHUB_WORKSPACE}/.github/configs/amd-master.yaml --seq-lens 1k1k --model-prefix gptoss)
echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT

Expand Down Expand Up @@ -159,15 +173,20 @@ jobs:
token: ${{ secrets.REPO_PAT }}
fetch-depth: 0

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install PyGithub

- name: Download results artifacts
uses: actions/download-artifact@v6
with:
path: ${{ env.RESULTS_DIR }}
pattern: results_*

- name: Install python dependencies
run: pip install PyGithub

- name: Calculate success rate
run: python3 utils/calc_success_rate.py $STATS_FILENAME

Expand Down
29 changes: 24 additions & 5 deletions .github/workflows/full-sweep-1k8k-scheduler.yml
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,16 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

- id: get-dsr1-configs
run: |
pip install pydantic
CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py full-sweep --config-files ${GITHUB_WORKSPACE}/.github/configs/nvidia-master.yaml ${GITHUB_WORKSPACE}/.github/configs/amd-master.yaml --seq-lens 1k8k --model-prefix dsr1)
echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT

Expand All @@ -28,9 +35,16 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

- id: get-gptoss-configs
run: |
pip install pydantic
CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py full-sweep --config-files ${GITHUB_WORKSPACE}/.github/configs/nvidia-master.yaml ${GITHUB_WORKSPACE}/.github/configs/amd-master.yaml --seq-lens 1k8k --model-prefix gptoss)
echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT

Expand Down Expand Up @@ -114,15 +128,20 @@ jobs:
token: ${{ secrets.REPO_PAT }}
fetch-depth: 0

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install PyGithub

- name: Download results artifacts
uses: actions/download-artifact@v6
with:
path: ${{ env.RESULTS_DIR }}
pattern: results_*

- name: Install python dependencies
run: pip install PyGithub

- name: Calculate success rate
run: python3 utils/calc_success_rate.py $STATS_FILENAME

Expand Down
29 changes: 24 additions & 5 deletions .github/workflows/full-sweep-8k1k-scheduler.yml
Original file line number Diff line number Diff line change
Expand Up @@ -14,9 +14,16 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

- id: get-dsr1-configs
run: |
pip install pydantic
CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py full-sweep --config-files ${GITHUB_WORKSPACE}/.github/configs/nvidia-master.yaml ${GITHUB_WORKSPACE}/.github/configs/amd-master.yaml --seq-lens 8k1k --model-prefix dsr1)
echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT

Expand All @@ -28,9 +35,16 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

- id: get-gptoss-configs
run: |
pip install pydantic
CONFIG_JSON=$(python3 ${GITHUB_WORKSPACE}/utils/matrix-logic/generate_sweep_configs.py full-sweep --config-files ${GITHUB_WORKSPACE}/.github/configs/nvidia-master.yaml ${GITHUB_WORKSPACE}/.github/configs/amd-master.yaml --seq-lens 8k1k --model-prefix gptoss)
echo "search-space-config=$CONFIG_JSON" >> $GITHUB_OUTPUT

Expand Down Expand Up @@ -159,15 +173,20 @@ jobs:
token: ${{ secrets.REPO_PAT }}
fetch-depth: 0

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install PyGithub

- name: Download results artifacts
uses: actions/download-artifact@v6
with:
path: ${{ env.RESULTS_DIR }}
pattern: results_*

- name: Install python dependencies
run: pip install PyGithub

- name: Calculate success rate
run: python3 utils/calc_success_rate.py $STATS_FILENAME

Expand Down
20 changes: 16 additions & 4 deletions .github/workflows/full-sweep-test.yml
Original file line number Diff line number Diff line change
Expand Up @@ -48,11 +48,18 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

# This looks complicated, but it is just calling generate_sweep_configs.py conditioned on
# discrete inputs (i.e., run_1k1k, run_h100, etc.) to split the test sweep into discrete jobs
- id: generate-configs
run: |
pip install pydantic

set -x
# Build runner type filters based on inputs
Expand Down Expand Up @@ -427,15 +434,20 @@ jobs:
token: ${{ secrets.REPO_PAT }}
fetch-depth: 0

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install PyGithub

- name: Download results artifacts
uses: actions/download-artifact@v6
with:
path: ${{ env.RESULTS_DIR }}
pattern: results_*

- name: Install python dependencies
run: pip install PyGithub

- name: Calculate success rate
run: python3 utils/calc_success_rate.py $STATS_FILENAME

Expand Down
20 changes: 16 additions & 4 deletions .github/workflows/label-validation.yml
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,14 @@ jobs:
- name: Checkout code
uses: actions/checkout@v5

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install pydantic pyyaml

- id: get-jobs
shell: python
run: |
Expand Down Expand Up @@ -49,7 +57,6 @@ jobs:
# Generate configs for standard labels
all_configs = []
if matching:
subprocess.run(['pip', 'install', 'pydantic'], check=True)

for label in matching:
result = subprocess.run([
Expand Down Expand Up @@ -125,15 +132,20 @@ jobs:
token: ${{ secrets.REPO_PAT }}
fetch-depth: 0

- name: Set up Python
uses: actions/setup-python@v6
with:
python-version: '3.12'

- name: Install dependencies
run: pip install PyGithub

- name: Download results artifacts
uses: actions/download-artifact@v6
with:
path: ${{ env.RESULTS_DIR }}
pattern: results_*

- name: Install python dependencies
run: pip install PyGithub

- name: Calculate success rate
run: python3 utils/calc_success_rate.py $STATS_FILENAME

Expand Down