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

Allow ROCm runner to upload benchmark results if found #144710

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Allow ROCm runner to upload benchmark results if found #144710

Changes from all commits
Commits
Show all changes
1 commit
Select commit
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
Allow ROCm runner to upload benchmark results if found
  • Loading branch information
@huydhn
huydhn committed Jan 13, 2025
commit 7c3ec48c3be38a3bd251a7a9c9055ea713af5d03
17 changes: 17 additions & 0 deletions .github/workflows/_rocm-test.yml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -63,6 +63,7 @@ jobs:
fail-fast: false
timeout-minutes: ${{ matrix.mem_leak_check == 'mem_leak_check' && 600 || inputs.timeout-minutes }}
runs-on: ${{ matrix.runner }}
environment: upload-benchmark-results
steps:
# [see note: pytorch repo ref]
- name: Checkout PyTorch
Expand Down Expand Up @@ -282,5 +283,21 @@ jobs:
if-no-files-found: ignore
path: ./**/core.[1-9]*

- name: Authenticate with AWS
uses: aws-actions/configure-aws-credentials@v4
with:
role-to-assume: arn:aws:iam::308535385114:role/gha_workflow_upload-benchmark-results
# The max duration enforced by the server side
role-duration-seconds: 18000
aws-region: us-east-1

- name: Upload the benchmark results
uses: pytorch/test-infra/.github/actions/upload-benchmark-results@main
with:
benchmark-results-dir: test/test-reports
dry-run: false
schema-version: v3
github-token: ${{ secrets.GITHUB_TOKEN }}

- name: Teardown ROCm
uses: ./.github/actions/teardown-rocm
Loading