| # @generated DO NOT EDIT MANUALLY |
| |
| # Template is at: .github/templates/windows_binary_build_workflow.yml.j2 |
| # Generation script: .github/scripts/generate_ci_workflows.py |
| name: windows-binary-libtorch-debug |
| |
| on: |
| push: |
| # NOTE: Meta Employees can trigger new nightlies using: https://fburl.com/trigger_pytorch_nightly_build |
| branches: |
| - nightly |
| tags: |
| # NOTE: Binary build pipelines should only get triggered on release candidate builds |
| # Release candidate tags look like: v1.11.0-rc1 |
| - v[0-9]+.[0-9]+.[0-9]+-rc[0-9]+ |
| - 'ciflow/binaries/*' |
| - 'ciflow/binaries_libtorch/*' |
| workflow_dispatch: |
| |
| env: |
| # Needed for conda builds |
| ALPINE_IMAGE: "308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/alpine" |
| ANACONDA_USER: pytorch |
| AWS_DEFAULT_REGION: us-east-1 |
| BUILD_ENVIRONMENT: windows-binary-libtorch-debug |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| IN_CI: 1 |
| IS_GHA: 1 |
| PR_LABELS: ${{ toJson(github.event.pull_request.labels.*.name) }} |
| PR_NUMBER: ${{ github.event.pull_request.number }} |
| SHA1: ${{ github.event.pull_request.head.sha || github.sha }} |
| SKIP_ALL_TESTS: 1 |
| concurrency: |
| group: windows-binary-libtorch-debug-${{ github.event.pull_request.number || github.sha }}-${{ github.event_name == 'workflow_dispatch' }} |
| cancel-in-progress: true |
| |
| jobs: |
| libtorch-cpu-shared-with-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cpu-shared-with-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cpu-shared-with-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cpu-shared-with-deps-debug-build |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cpu-shared-with-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cpu-shared-with-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cpu-shared-with-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cpu-shared-with-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cpu-shared-without-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cpu-shared-without-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cpu-shared-without-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cpu-shared-without-deps-debug-build |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cpu-shared-without-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cpu-shared-without-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cpu-shared-without-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cpu-shared-without-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cpu-static-with-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cpu-static-with-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cpu-static-with-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cpu-static-with-deps-debug-build |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cpu-static-with-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cpu-static-with-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cpu-static-with-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cpu-static-with-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cpu-static-without-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cpu-static-without-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cpu-static-without-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cpu-static-without-deps-debug-build |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cpu-static-without-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cpu-static-without-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cpu-static-without-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cpu |
| GPU_ARCH_TYPE: cpu |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cpu-static-without-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cuda11_3-shared-with-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cuda11_3-shared-with-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_3-shared-with-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_3-shared-with-deps-debug-build |
| runs-on: windows.8xlarge.nvidia.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_3-shared-with-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_3-shared-with-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_3-shared-with-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_3-shared-with-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cuda11_3-shared-without-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cuda11_3-shared-without-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_3-shared-without-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_3-shared-without-deps-debug-build |
| runs-on: windows.8xlarge.nvidia.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_3-shared-without-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_3-shared-without-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_3-shared-without-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_3-shared-without-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cuda11_3-static-with-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cuda11_3-static-with-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_3-static-with-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_3-static-with-deps-debug-build |
| runs-on: windows.8xlarge.nvidia.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_3-static-with-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_3-static-with-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_3-static-with-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_3-static-with-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cuda11_3-static-without-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cuda11_3-static-without-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_3-static-without-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_3-static-without-deps-debug-build |
| runs-on: windows.8xlarge.nvidia.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_3-static-without-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_3-static-without-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_3-static-without-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu113 |
| GPU_ARCH_VERSION: 11.3 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_3-static-without-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cuda11_6-shared-with-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cuda11_6-shared-with-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_6-shared-with-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_6-shared-with-deps-debug-build |
| runs-on: windows.8xlarge.nvidia.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_6-shared-with-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_6-shared-with-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_6-shared-with-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_6-shared-with-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cuda11_6-shared-without-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cuda11_6-shared-without-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_6-shared-without-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_6-shared-without-deps-debug-build |
| runs-on: windows.8xlarge.nvidia.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_6-shared-without-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_6-shared-without-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_6-shared-without-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: shared-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_6-shared-without-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cuda11_6-static-with-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cuda11_6-static-with-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_6-static-with-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_6-static-with-deps-debug-build |
| runs-on: windows.8xlarge.nvidia.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_6-static-with-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_6-static-with-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_6-static-with-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-with-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_6-static-with-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
| libtorch-cuda11_6-static-without-deps-debug-build: |
| if: ${{ github.repository_owner == 'pytorch' }} |
| runs-on: windows.4xlarge |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Build PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_build.sh" |
| - uses: seemethere/upload-artifact-s3@v5 |
| if: always() |
| with: |
| name: libtorch-cuda11_6-static-without-deps-debug |
| retention-days: 14 |
| if-no-files-found: error |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_6-static-without-deps-debug-test: # Testing |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_6-static-without-deps-debug-build |
| runs-on: windows.8xlarge.nvidia.gpu |
| timeout-minutes: 240 |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Display EC2 information |
| shell: bash |
| run: | |
| set -euo pipefail |
| function get_ec2_metadata() { |
| # Pulled from instance metadata endpoint for EC2 |
| # see https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/instancedata-data-retrieval.html |
| category=$1 |
| curl -fsSL "http://169.254.169.254/latest/meta-data/${category}" |
| } |
| echo "ami-id: $(get_ec2_metadata ami-id)" |
| echo "instance-id: $(get_ec2_metadata instance-id)" |
| echo "instance-type: $(get_ec2_metadata instance-type)" |
| echo "system info $(uname -a)" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| # Needed for binary builds, see: https://github.com/pytorch/pytorch/issues/73339#issuecomment-1058981560 |
| - name: Enable long paths on Windows |
| shell: powershell |
| run: | |
| Set-ItemProperty -Path "HKLM:\\SYSTEM\CurrentControlSet\Control\FileSystem" -Name "LongPathsEnabled" -Value 1 |
| # Since it's just a defensive command, the workflow should continue even the command fails |
| - name: Disables Windows Defender scheduled and real-time scanning for files in pytorch directory. |
| shell: powershell |
| run: | |
| Add-MpPreference -ExclusionPath $(Get-Location).tostring() -ErrorAction Ignore |
| # NOTE: These environment variables are put here so that they can be applied on every job equally |
| # They are also here because setting them at a workflow level doesn't give us access to the |
| # runner.temp variable, which we need. |
| - name: Populate binary env |
| shell: bash |
| run: | |
| echo "BINARY_ENV_FILE=${RUNNER_TEMP}/env" >> "${GITHUB_ENV}" |
| echo "PYTORCH_FINAL_PACKAGE_DIR=${RUNNER_TEMP}/artifacts" >> "${GITHUB_ENV}" |
| echo "WIN_PACKAGE_WORK_DIR=${RUNNER_TEMP}" |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_6-static-without-deps-debug |
| path: "${{ env.PYTORCH_FINAL_PACKAGE_DIR }}" |
| - name: Checkout PyTorch |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }} |
| submodules: recursive |
| path: pytorch |
| - name: Clean PyTorch checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: pytorch |
| - name: Checkout pytorch/builder |
| uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9 |
| with: |
| ref: main |
| submodules: recursive |
| repository: pytorch/builder |
| path: builder |
| - name: Clean pytorch/builder checkout |
| run: | |
| # Remove any artifacts from the previous checkouts |
| git clean -fxd |
| working-directory: builder |
| - name: Populate binary env |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_populate_env.sh" |
| - name: Test PyTorch binary |
| shell: bash |
| run: | |
| "${PYTORCH_ROOT}/.circleci/scripts/binary_windows_test.sh" |
| - name: Wait until all sessions have drained |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| timeout-minutes: 120 |
| run: | |
| .github\scripts\wait_for_ssh_to_drain.ps1 |
| - name: Kill active ssh sessions if still around (Useful if workflow was cancelled) |
| shell: powershell |
| working-directory: pytorch |
| if: always() |
| run: | |
| .github\scripts\kill_active_ssh_sessions.ps1 |
| libtorch-cuda11_6-static-without-deps-debug-upload: # Uploading |
| runs-on: linux.2xlarge # self hosted runner to download ec2 artifacts |
| if: ${{ github.repository_owner == 'pytorch' }} |
| needs: libtorch-cuda11_6-static-without-deps-debug-test |
| env: |
| PYTORCH_ROOT: ${{ github.workspace }}/pytorch |
| BUILDER_ROOT: ${{ github.workspace }}/builder |
| PACKAGE_TYPE: libtorch |
| # TODO: This is a legacy variable that we eventually want to get rid of in |
| # favor of GPU_ARCH_VERSION |
| DESIRED_CUDA: cu116 |
| GPU_ARCH_VERSION: 11.6 |
| GPU_ARCH_TYPE: cuda |
| SKIP_ALL_TESTS: 1 |
| LIBTORCH_CONFIG: debug |
| LIBTORCH_VARIANT: static-without-deps |
| # This is a dummy value for libtorch to work correctly with our batch scripts |
| # without this value pip does not get installed for some reason |
| DESIRED_PYTHON: "3.7" |
| steps: |
| - name: Checkout PyTorch |
| uses: pytorch/pytorch/.github/actions/checkout-pytorch@master |
| - name: Setup Linux |
| uses: ./.github/actions/setup-linux |
| - name: Chown workspace |
| run: | |
| retry () { |
| "$@" || (sleep 1 && "$@") || (sleep 2 && "$@") |
| } |
| retry docker pull "${ALPINE_IMAGE}" |
| # Ensure the working directory gets chowned back to the current user |
| docker run --pull=never --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Clean workspace |
| run: | |
| rm -rf "${GITHUB_WORKSPACE}" |
| mkdir "${GITHUB_WORKSPACE}" |
| - name: "[FB EMPLOYEES] Enable SSH (Click me for login details)" |
| uses: seemethere/add-github-ssh-key@v1 |
| with: |
| GITHUB_TOKEN: ${{ secrets.GITHUB_TOKEN }} |
| - name: Clone pytorch/pytorch |
| uses: actions/checkout@v2 |
| - uses: seemethere/download-artifact-s3@v4 |
| name: Download Build Artifacts |
| with: |
| name: libtorch-cuda11_6-static-without-deps-debug |
| path: "${{ runner.temp }}/artifacts/" |
| - name: Set DRY_RUN (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && (github.event.ref == 'refs/heads/nightly' || (startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/'))) }} |
| run: | |
| echo "DRY_RUN=disabled" >> "$GITHUB_ENV" |
| - name: Set UPLOAD_CHANNEL (only for tagged pushes) |
| if: ${{ github.event_name == 'push' && startsWith(github.event.ref, 'refs/tags/') && !startsWith(github.event.ref, 'refs/tags/ciflow/') }} |
| run: | |
| # reference ends with an RC suffix |
| if [[ ${GITHUB_REF_NAME} = *-rc[0-9]* ]]; then |
| echo "UPLOAD_CHANNEL=test" >> "$GITHUB_ENV" |
| fi |
| - name: Upload binaries |
| env: |
| PKG_DIR: "${{ runner.temp }}/artifacts" |
| UPLOAD_SUBFOLDER: "${{ env.DESIRED_CUDA }}" |
| # When running these on pull_request events these should be blank |
| AWS_ACCESS_KEY_ID: ${{ secrets.AWS_PYTORCH_UPLOADER_ACCESS_KEY_ID }} |
| AWS_SECRET_ACCESS_KEY: ${{ secrets.AWS_PYTORCH_UPLOADER_SECRET_ACCESS_KEY }} |
| ANACONDA_API_TOKEN: ${{ secrets.CONDA_PYTORCHBOT_TOKEN }} |
| run: | |
| docker run --rm -i \ |
| -e ANACONDA_API_TOKEN \ |
| -e AWS_ACCESS_KEY_ID \ |
| -e AWS_SECRET_ACCESS_KEY \ |
| -e DRY_RUN \ |
| -e PACKAGE_TYPE \ |
| -e PKG_DIR=/artifacts \ |
| -e UPLOAD_CHANNEL \ |
| -e UPLOAD_SUBFOLDER \ |
| -v "${RUNNER_TEMP}/artifacts:/artifacts" \ |
| -v "${GITHUB_WORKSPACE}:/v" \ |
| -w /v \ |
| 308535385114.dkr.ecr.us-east-1.amazonaws.com/tool/miniconda3:4.10.3 \ |
| bash -c '.circleci/scripts/binary_upload.sh' |
| - name: Hold runner for 2 hours or until ssh sessions have drained |
| # Always hold for active ssh sessions |
| if: always() |
| run: .github/scripts/wait_for_ssh_to_drain.sh |
| - name: Chown workspace |
| if: always() |
| run: | |
| # Ensure the working directory gets chowned back to the current user |
| docker run --rm -v "$(pwd)":/v -w /v "${ALPINE_IMAGE}" chown -R "$(id -u):$(id -g)" . |
| - name: Kill containers, clean up images |
| if: always() |
| run: | |
| # ignore expansion of "docker ps -q" since it could be empty |
| # shellcheck disable=SC2046 |
| docker stop $(docker ps -q) || true |
| # Prune all of the docker images |
| docker system prune -af |
