Skip to content
Open
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
8 changes: 4 additions & 4 deletions .github/workflows/build-ami.yml
Original file line number Diff line number Diff line change
Expand Up @@ -34,16 +34,16 @@ jobs:
AWS_REGION: us-east-1
steps:
- name: Checkout
uses: actions/checkout@df4cb1c069e1874edd31b4311f1884172cec0e10 # v6.0.3
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1

- name: Setup Packer
uses: hashicorp/setup-packer@3286471d6cc6756d056a0b199fea5e0becdbc189 # main
uses: hashicorp/setup-packer@ce93c3c08a6c2ff2275bf4b54ff0d9a75f6c9789 # v3.4.0
id: setup-packer
with:
version: "1.10.1"

- name: configure aws credentials
uses: aws-actions/configure-aws-credentials@e7f100cf4c008499ea8adda475de1042d6975c7b # v6.2.0
uses: aws-actions/configure-aws-credentials@e1253824e5c10ff9df46874f81ed3ec929e19cfd # v6.3.0
with:
role-to-assume: arn:aws:iam::907797767998:role/optimum-neuron-build-ami
aws-region: ${{ env.AWS_REGION }}
Expand Down Expand Up @@ -84,7 +84,7 @@ jobs:

- name: Slack Notification on Failure
id: slack
uses: slackapi/slack-github-action@45a88b9581bfab2566dc881e2cd66d334e621e2c # v3.0.3
uses: slackapi/slack-github-action@dcb1066f776dd043e64d0e8ba94ca15cc7e1875d # v4.0.0
if: ${{ failure() && github.event_name == 'schedule' }}
with:
channel-id: 'C06GAEQJLNN' #copied from slack channel
Expand Down
10 changes: 5 additions & 5 deletions .github/workflows/build-vllm-image.yml
Original file line number Diff line number Diff line change
Expand Up @@ -16,30 +16,30 @@ jobs:
steps:
-
name: Checkout sources
uses: actions/checkout@df4cb1c069e1874edd31b4311f1884172cec0e10 # v6.0.3
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
-
name: Extract version
run: |
export ON_VERSION=$(make version)
echo "ON_VERSION=${ON_VERSION}">> $GITHUB_ENV
-
name: Set up Docker Buildx
uses: docker/setup-buildx-action@d7f5e7f509e45cec5c76c4d5afdd7de93d0b3df5 # v4.1.0
uses: docker/setup-buildx-action@f87e5991a6d7451dcb8d9637bfbc97413f497069 # v4.4.1
-
name: Login to ghcr.io
uses: docker/login-action@650006c6eb7dba73a995cc03b0b2d7f5ca915bee # v4.2.0
uses: docker/login-action@dbcb813823bdd20940b903addbd779551569679f # v4.6.0
with:
registry: ghcr.io
username: ${{ github.actor }}
password: ${{ secrets.GITHUB_TOKEN }}
- name: Login to DockerHub
uses: docker/login-action@650006c6eb7dba73a995cc03b0b2d7f5ca915bee # v4.2.0
uses: docker/login-action@dbcb813823bdd20940b903addbd779551569679f # v4.6.0
with:
username: ${{ secrets.DOCKERHUB_USERNAME }}
password: ${{ secrets.DOCKERHUB_PASSWORD }}
-
name: Build and push
uses: docker/build-push-action@f9f3042f7e2789586610d6e8b85c8f03e5195baf # v7.2.0
uses: docker/build-push-action@c3c9e263c25d99ce0380d002d59b67737d91b0dc # v7.4.0
with:
context: .
file: docker/vllm/Dockerfile
Expand Down
10 changes: 8 additions & 2 deletions .github/workflows/cache_diffusion.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,19 +22,25 @@ concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.ref_name || github.run_id }}
cancel-in-progress: true

permissions: {}

jobs:
sanity:
permissions:
contents: read
name: Sanity
runs-on: ubuntu-24.04
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Run sanity check
uses: ./.github/actions/sanity-check
with:
hf_token: ${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CACHE }}

cache:
permissions:
contents: read
name: Create optimum-neuron inference cache
needs: sanity
runs-on:
Expand All @@ -55,7 +61,7 @@ jobs:
]
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Install Neuronx runtime
uses: ./.github/actions/install_neuronx_runtime
- name: Setup virtual environment
Expand Down
10 changes: 8 additions & 2 deletions .github/workflows/cache_llm.yml
Original file line number Diff line number Diff line change
Expand Up @@ -22,19 +22,25 @@ concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.ref_name || github.run_id }}
cancel-in-progress: true

permissions: {}

jobs:
sanity:
permissions:
contents: read
name: Sanity
runs-on: ubuntu-24.04
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Run sanity check
uses: ./.github/actions/sanity-check
with:
hf_token: ${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CACHE }}

cache:
permissions:
contents: read
name: Create optimum-neuron LLM inference cache
needs: sanity
runs-on:
Expand All @@ -57,7 +63,7 @@ jobs:
]
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Install Neuronx runtime
uses: ./.github/actions/install_neuronx_runtime
- name: Setup virtual environment
Expand Down
8 changes: 6 additions & 2 deletions .github/workflows/doc-build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -15,8 +15,12 @@ on:
- '.github/workflows/doc-build.yml'
workflow_dispatch:

permissions: {}

jobs:
build_documentation:
permissions:
contents: read
runs-on: ubuntu-24.04
env:
COMMIT_SHA: ${{ github.event.pull_request.head.sha }}
Expand All @@ -25,8 +29,8 @@ jobs:
PR_CLONE_URL: ${{ github.event.pull_request.head.repo.clone_url }}

steps:
- uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
- uses: actions/setup-node@249970729cb0ef3589644e2896645e5dc5ba9c38 # v6.5.0
- uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- uses: actions/setup-node@949feb2413d6458794dcd2491c4babbbce0c15c1 # v7.1.0
with:
node-version: '22'
cache-dependency-path: "kit/package-lock.json"
Expand Down
10 changes: 7 additions & 3 deletions .github/workflows/doc-pr-build.yml
Original file line number Diff line number Diff line change
Expand Up @@ -14,8 +14,12 @@ concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.run_id }}
cancel-in-progress: true

permissions: {}

jobs:
build_documentation:
permissions:
contents: read
runs-on: ubuntu-24.04
env:
COMMIT_SHA: ${{ github.event.pull_request.head.sha }}
Expand All @@ -24,8 +28,8 @@ jobs:
PR_CLONE_URL: ${{ github.event.pull_request.head.repo.clone_url }}

steps:
- uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
- uses: actions/setup-node@249970729cb0ef3589644e2896645e5dc5ba9c38 # v6.5.0
- uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- uses: actions/setup-node@949feb2413d6458794dcd2491c4babbbce0c15c1 # v7.1.0
with:
node-version: '22'
cache-dependency-path: "kit/package-lock.json"
Expand Down Expand Up @@ -88,7 +92,7 @@ jobs:
echo ${{ env.COMMIT_SHA }} > ./commit_sha
echo ${{ env.PR_NUMBER }} > ./pr_number

- uses: actions/upload-artifact@043fb46d1a93c77aae656e7c1c64a875d1fc6a0a # v7.0.1
- uses: actions/upload-artifact@cf430e030ddbb5b0abf93d22962f4752f3646cd9 # v7.0.2
with:
name: doc-build-artifact
path: neuron-doc-build/
2 changes: 1 addition & 1 deletion .github/workflows/doc-pr-upload.yml
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,7 @@ on:

jobs:
build:
uses: huggingface/doc-builder/.github/workflows/upload_pr_documentation.yml@9fc8a41d7729c424fdb2192f2f85ff63102ec9a3 # main
uses: huggingface/doc-builder/.github/workflows/upload_pr_documentation.yml@90ddd199e07144a13408c523680c76987343c95f # main
with:
package_name: optimum-neuron
secrets:
Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/security.yml
Original file line number Diff line number Diff line change
Expand Up @@ -11,10 +11,10 @@ jobs:
runs-on: ubuntu-latest
steps:
- name: Checkout code
uses: actions/checkout@df4cb1c069e1874edd31b4311f1884172cec0e10 # v6.0.3
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
with:
fetch-depth: 0
- name: Secret Scanning
uses: trufflesecurity/trufflehog@d411fff7b8879a62509f3fa98c07f247ac089a51 # main
uses: trufflesecurity/trufflehog@b2b0a92070f206ab7b5a1105d82a7f2f48d92341 # v3.99.0
with:
extra_args: --results=verified,unknown
9 changes: 5 additions & 4 deletions .github/workflows/stale.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -4,15 +4,16 @@ on:
- cron: '0 8 * * *'
workflow_dispatch:

permissions:
issues: write
pull-requests: write
permissions: {}

jobs:
stale:
permissions:
issues: write
pull-requests: write
runs-on: ubuntu-latest
steps:
- uses: actions/stale@1e223db275d687790206a7acac4d1a11bd6fe629 # v10.4.0
- uses: actions/stale@4391f3da665fdf50b6810c1a66712fb9ba21aa93 # v11.0.0
with:
stale-issue-message: 'This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 5 days.'
stale-pr-message: 'This PR is stale because it has been open 15 days with no activity. Remove stale label or comment or this will be closed in 5 days.'
Expand Down
10 changes: 8 additions & 2 deletions .github/workflows/test_cpu_only.yml
Original file line number Diff line number Diff line change
Expand Up @@ -52,25 +52,31 @@ concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.ref_name || github.run_id }}
cancel-in-progress: true

permissions: {}

jobs:
sanity:
permissions:
contents: read
name: Sanity
runs-on: ubuntu-24.04
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Run sanity check
uses: ./.github/actions/sanity-check
with:
hf_token: ${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CACHE }}

test-cpu:
permissions:
contents: read
name: Run CPU Only Tests
needs: sanity
runs-on: ubuntu-24.04
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Install Neuronx runtime
uses: ./.github/actions/install_neuronx_runtime
- name: Setup virtual environment
Expand Down
14 changes: 11 additions & 3 deletions .github/workflows/test_inf2_diffusers.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,26 +36,32 @@ concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.ref_name || github.run_id }}
cancel-in-progress: true

permissions: {}

jobs:
sanity:
permissions:
contents: read
name: Sanity
runs-on: ubuntu-24.04
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Run sanity check
uses: ./.github/actions/sanity-check
with:
hf_token: ${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CACHE }}

diffusers-model-tests:
permissions:
contents: read
name: Run diffusers model tests
needs: sanity
runs-on:
group: aws-inf2-8xlarge
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Install Neuronx runtime
uses: ./.github/actions/install_neuronx_runtime
- name: Install cv2 dependencies
Expand All @@ -82,13 +88,15 @@ jobs:
HF_TOKEN=${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CI }} pytest tests/inference/diffusers/test_flux.py --forked -q

diffusers-cache-cli-tests:
permissions:
contents: read
name: Run diffusers cache and CLI tests
needs: sanity
runs-on:
group: aws-inf2-8xlarge
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Install Neuronx runtime
uses: ./.github/actions/install_neuronx_runtime
- name: Install cv2 dependencies
Expand Down
10 changes: 8 additions & 2 deletions .github/workflows/test_inf2_export.yml
Original file line number Diff line number Diff line change
Expand Up @@ -44,26 +44,32 @@ concurrency:
group: ${{ github.workflow }}-${{ github.head_ref || github.ref_name || github.run_id }}
cancel-in-progress: true

permissions: {}

jobs:
sanity:
permissions:
contents: read
name: Sanity
runs-on: ubuntu-24.04
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Run sanity check
uses: ./.github/actions/sanity-check
with:
hf_token: ${{ secrets.HF_TOKEN_OPTIMUM_NEURON_CACHE }}

do-the-job:
permissions:
contents: read
name: Run INF2 export tests
needs: sanity
runs-on:
group: aws-inf2-8xlarge
steps:
- name: Checkout
uses: actions/checkout@d23441a48e516b6c34aea4fa41551a30e30af803 # v6.1.0
uses: actions/checkout@3d3c42e5aac5ba805825da76410c181273ba90b1 # v7.0.1
- name: Install Neuronx runtime
uses: ./.github/actions/install_neuronx_runtime
- name: Setup virtual environment
Expand Down
Loading
Loading