Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

migrate ci #662

Merged
merged 4 commits into from
Jul 24, 2024
Merged

migrate ci #662

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
573248f
migrate ci
XciD Jul 22, 2024
7ad03cc
test
XciD Jul 22, 2024
97e6472
trigger workflow
XciD Jul 22, 2024
8e791cf
test
XciD Jul 22, 2024
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
5 changes: 3 additions & 2 deletions .github/workflows/inference_cache.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -12,7 +12,8 @@ concurrency:
jobs:
cache:
name: Create optimum-neuron inference cache
runs-on: [self-hosted, 12-aws-inf2, 192-cpu, ci] # run the job on the newly created runner
runs-on:
group: aws-inf2-48xlarge
env:
AWS_REGION: us-east-1
strategy:
Expand Down Expand Up @@ -44,7 +45,7 @@ jobs:
uses: actions/checkout@v4
- name: Install python and create venv
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
7 changes: 3 additions & 4 deletions .github/workflows/test_inf1_export.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,8 @@ concurrency:
jobs:
do-the-job:
name: Run INF1 export tests
runs-on: [self-hosted, 4-aws-inf1, 24-cpu, ci]
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf1-6xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -28,7 +27,7 @@ jobs:
uses: actions/checkout@v2
- name: Install system packages
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
- name: Install python packages
run: |
python3 -m venv aws_neuron_venv_pytorch
Expand Down
7 changes: 3 additions & 4 deletions .github/workflows/test_inf1_full_export.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,8 @@ concurrency:
jobs:
do-the-job:
name: Run INF1 full export tests
runs-on: [self-hosted, 4-aws-inf1, 24-cpu, ci]
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf1-6xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -28,7 +27,7 @@ jobs:
uses: actions/checkout@v2
- name: Install system packages
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
- name: Install python packages
run: |
python3 -m venv aws_neuron_venv_pytorch
Expand Down
7 changes: 3 additions & 4 deletions .github/workflows/test_inf1_inference.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,8 @@ concurrency:
jobs:
do-the-job:
name: Run INF1 tests
runs-on: [self-hosted, 4-aws-inf1, 24-cpu, ci]
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf1-6xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -28,7 +27,7 @@ jobs:
uses: actions/checkout@v2
- name: Install system packages
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
- name: Install python packages
run: |
python3 -m venv aws_neuron_venv_pytorch
Expand Down
7 changes: 3 additions & 4 deletions .github/workflows/test_inf1_pipelines.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,8 @@ concurrency:
jobs:
do-the-job:
name: Run INF1 tests
runs-on: [self-hosted, 4-aws-inf1, 24-cpu, ci]
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf1-6xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -28,7 +27,7 @@ jobs:
uses: actions/checkout@v2
- name: Install system packages
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
- name: Install python packages
run: |
python3 -m venv aws_neuron_venv_pytorch
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/test_inf2.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,13 @@ on:
paths:
- "setup.py"
- "optimum/**.py"
- ".github/workflows/test_inf2.yml"
pull_request:
branches: [ main ]
paths:
- "setup.py"
- "optimum/**.py"
- ".github/workflows/test_inf2.yml"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand All @@ -19,9 +21,8 @@ concurrency:
jobs:
do-the-job:
name: Run INF2 tests
runs-on: [self-hosted, 1-aws-inf2, 32-cpu, ci] # run the job on the newly created runner
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf2-8xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -37,7 +38,7 @@ jobs:
uses: actions/checkout@v2
- name: Install python dependencies
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/test_inf2_export.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,13 @@ on:
paths:
- "setup.py"
- "optimum/**.py"
- ".github/workflows/test_inf2_export.yml"
pull_request:
branches: [ main ]
paths:
- "setup.py"
- "optimum/**.py"
- ".github/workflows/test_inf2_export.yml"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand All @@ -19,9 +21,8 @@ concurrency:
jobs:
do-the-job:
name: Run INF2 export tests
runs-on: [self-hosted, 1-aws-inf2, 32-cpu, ci] # run the job on the newly created runner
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf2-8xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -37,7 +38,7 @@ jobs:
uses: actions/checkout@v2
- name: Install python dependencies
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/test_inf2_full_export.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -5,10 +5,12 @@ on:
branches: [ main ]
paths:
- "optimum/exporters/neuron/*.py"
- ".github/workflows/test_inf2_full_export.yml"
pull_request:
branches: [ main ]
paths:
- "optimum/exporters/neuron/*.py"
- ".github/workflows/test_inf2_full_export.yml"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand All @@ -17,9 +19,8 @@ concurrency:
jobs:
do-the-job:
name: Run INF2 full export tests
runs-on: [self-hosted, 1-aws-inf2, 32-cpu, ci] # run the job on the newly created runner
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf2-8xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -35,7 +36,7 @@ jobs:
uses: actions/checkout@v2
- name: Install python dependencies
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/test_inf2_inference.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,13 @@ on:
paths:
- "setup.py"
- "optimum/**.py"
- ".github/workflows/test_inf2_inference.yml"
pull_request:
branches: [ main ]
paths:
- "setup.py"
- "optimum/**.py"
- ".github/workflows/test_inf2_inference.yml"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand All @@ -19,9 +21,8 @@ concurrency:
jobs:
do-the-job:
name: Run INF2 tests
runs-on: [self-hosted, 1-aws-inf2, 32-cpu, ci] # run the job on the newly created runner
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf2-8xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -37,7 +38,7 @@ jobs:
uses: actions/checkout@v2
- name: Install python dependencies
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/test_inf2_tgi.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -7,12 +7,14 @@ on:
- "setup.py"
- "optimum/**.py"
- "text-generation-inference/**"
- ".github/workflows/test_inf2_tgi.yml"
pull_request:
branches: [ main ]
paths:
- "setup.py"
- "optimum/**.py"
- "text-generation-inference/**"
- ".github/workflows/test_inf2_tgi.yml"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand All @@ -21,9 +23,8 @@ concurrency:
jobs:
do-the-job:
name: Run TGI tests
runs-on: [self-hosted, 1-aws-inf2, 32-cpu, ci] # run the job on the newly created runner
env:
AWS_REGION: us-east-1
runs-on:
group: aws-inf2-8xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -39,7 +40,7 @@ jobs:
uses: actions/checkout@v2
- name: Install python and create venv
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
8 changes: 5 additions & 3 deletions .github/workflows/test_trainium_common.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,13 @@ on:
paths:
- "setup.py"
- "optimum/**.py"
- ".github/workflows/test_trainium_common.yml"
pull_request:
branches: [ main ]
paths:
- "setup.py"
- "optimum/**.py"
- ".github/workflows/test_trainium_common.yml"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand All @@ -19,9 +21,9 @@ concurrency:
jobs:
optimum-neuron-tests:
name: Run common tests on Trainium 1
runs-on: [self-hosted, 16-aws-trn, 128-cpu, ci]
runs-on:
group: aws-trn1-32xlarge
env:
AWS_REGION: us-east-1
TESTS_TO_IGNORE_FLAGS: --ignore tests/distributed/ --ignore tests/test_examples.py
steps:
- name: Install Neuron runtime
Expand All @@ -38,7 +40,7 @@ jobs:
uses: actions/checkout@v2
- name: Install python dependencies
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
9 changes: 5 additions & 4 deletions .github/workflows/test_trainium_distributed.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,11 +6,13 @@ on:
paths:
- "setup.py"
- "optimum/neuron/distributed/**.py"
- ".github/workflows/test_trainium_distributed.yml"
pull_request:
branches: [ main ]
paths:
- "setup.py"
- "optimum/neuron/distributed/**.py"
- ".github/workflows/test_trainium_distributed.yml"

concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
Expand All @@ -20,9 +22,8 @@ concurrency:
jobs:
optimum-neuron-tests:
name: Run distributed tests on Trainium 1
runs-on: [self-hosted, 16-aws-trn, 128-cpu, ci]
env:
AWS_REGION: us-east-1
runs-on:
group: aws-trn1-32xlarge
steps:
- name: Install Neuron runtime
run: |
Expand All @@ -40,7 +41,7 @@ jobs:
run: echo "/home/ubuntu/.local/bin" >> $GITHUB_PATH
- name: Install python dependencies
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
6 changes: 3 additions & 3 deletions .github/workflows/test_trainium_examples.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -28,9 +28,9 @@ concurrency:
jobs:
optimum-neuron-tests:
name: Run example script tests on Trainium 1
runs-on: [self-hosted, 1-aws-trn, 8-cpu, ci] # run the job on the newly created runner
runs-on:
group: aws-trn1-2xlarge
env:
AWS_REGION: us-east-1
RUN_TINY: ${{ github.event.inputs.model_size == "tiny" && "1" || "0" }}
steps:
- name: Install Neuron runtime
Expand All @@ -49,7 +49,7 @@ jobs:
run: echo "/home/ubuntu/.local/bin" >> $GITHUB_PATH
- name: Install python dependencies
run: |
sudo apt install python3.8-venv python3-dev -y
sudo apt install python3-venv python3-dev -y
python3 -m venv aws_neuron_venv_pytorch
source aws_neuron_venv_pytorch/bin/activate
python -m pip install -U pip
Expand Down
Loading