-
Notifications
You must be signed in to change notification settings - Fork 84
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.
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
Enable linux_job.yml workflow for ROCm #6003
Conversation
The latest updates on your projects. Learn more about Vercel for Git ↗︎
|
No ciflow labels are configured for this repo. |
.github/workflows/linux_job.yml
Outdated
if: ${{ inputs.rocm == false }} | ||
|
||
- name: Setup ROCm | ||
uses: amdfaa/pytorch/.github/actions/setup-rocm@torchao_experiment |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Temporary change until pytorch/pytorch#143590 is merged
env_file = f"{ os.environ.get('RUNNER_TEMP', '') }/github_env_{ os.environ.get('GITHUB_RUN_ID', '') }" | ||
if f"{ os.environ.get('ROCM', '') }" == "true": | ||
env_file = f"/tmp/github_env_{ os.environ.get('GITHUB_RUN_ID', '') }" | ||
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This shouldn't be needed if we update setup-rocm.yml
to export all the required env vars to ${RUNNER_TEMP}/github_env_${GITHUB_RUN_ID}
like in actions/setup-linux/action.yml
-v "{ os.environ.get('RUNNER_TEST_RESULTS_DIR', '') }:/test-results" \ | ||
-v "{ os.environ.get('RUNNER_TEMP', '') }/artifacts:/artifacts" \ | ||
-v "{ os.environ.get('RUNNER_TEMP', '') }/docs:/docs" \ | ||
-v "{ os.environ.get('RUNNER_TEMP', '') }/test-results:/test-results" \ |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's define these RUNNER_
env vars in setup-rocm.yml similar to
RUNNER_ARTIFACT_DIR="${RUNNER_TEMP}/artifacts" |
.github/workflows/linux_job.yml
Outdated
@@ -60,7 +60,7 @@ on: | |||
type: string | |||
docker-image: | |||
description: Identifies the Docker image by name. | |||
default: "pytorch/conda-builder" | |||
default: "pytorch/almalinux-builder" |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Let's make changes to linux_job_v2.yml
instead, which already has these almalinux-based changes, so the only updates would be for the ROCm conditionalization.
379b577
to
998e2f9
Compare
Needed for pytorch/test-infra#6003 and pytorch/ao#999 Pull Request resolved: #143590 Approved by: https://github.com/atalman Co-authored-by: Jithun Nair <[email protected]>
998e2f9
to
4796068
Compare
Relates to pytorch/ao#999
This PR has been closed in favor of: #6104