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.
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.
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.
Let's define these RUNNER_ env vars in setup-rocm.yml similar to
.github/workflows/linux_job.yml
Outdated
| docker-image: | ||
| description: Identifies the Docker image by name. | ||
| default: "pytorch/conda-builder" | ||
| default: "pytorch/almalinux-builder" |
There was a problem hiding this comment.
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 <37884920+jithunnair-amd@users.noreply.github.com>
998e2f9 to
4796068
Compare
Relates to pytorch/ao#999
This PR has been closed in favor of: #6104