pytorch/.github/scripts
Nirav Mehta e81ab046bd Skip stale check when facebook-github-bot is merging (#79572)
# Summary

ShipIt jobs triggered by co-development workflows are failing to merge PRs due to stale checks.  This diff skips the stale check when merge is triggered by `facebook-github-bot`.

Sample merge failure: https://github.com/pytorch/pytorch/pull/78654#issuecomment-1155607617
Pull Request resolved: https://github.com/pytorch/pytorch/pull/79572
Approved by: https://github.com/bigfootjon, https://github.com/seemethere, https://github.com/malfet
2022-06-15 03:24:51 +00:00
..
build_publish_nightly_docker.sh ci: Fix cudatoolkit issue, make docker builds testable 2022-03-09 19:49:05 +00:00
convert_lintrunner_annotations_to_github.py [ci] use lintrunner in CI 2022-04-15 00:08:21 +00:00
ensure_actions_will_cancel.py Workflow consolidation for GitHub actions 2022-03-21 14:15:03 +00:00
export_pytorch_labels.py
generate_binary_build_matrix.py Upgrade CI to ROCm 5.1 2022-04-28 00:02:07 +00:00
generate_ci_workflows.py [ci] remove ciflow/all (#78317) 2022-05-26 16:25:25 +00:00
generate_pytorch_version.py
get_workflow_job_id.py [ci] Fix bug in get_workflow_job_id.py 2022-04-08 21:09:38 +00:00
gitutils.py [GHF] Retry push unconditionally (#77426) 2022-05-13 17:09:01 +00:00
gql_mocks.json [mergebot] add additional checks (#79317) 2022-06-13 19:28:29 +00:00
install_nvidia_utils_linux.sh Cuda 11.6 Disable failing tests (#75420) 2022-04-07 22:43:15 +00:00
kill_active_ssh_sessions.ps1
lint_native_functions.py [lint] upgrade mypy to latest version 2022-05-03 20:51:34 +00:00
parse_ref.py
print_latest_commits.py Refactored print_latest_commits.py to return the latest viable commit from master (#79422) 2022-06-13 18:30:38 +00:00
process_commit.py Fix alignment, make sure release labels are included (#73739) 2022-03-03 16:30:55 +00:00
README.md [BE] add readme for .github, located in .github/scripts 2022-04-01 19:53:44 +00:00
report_git_status.sh
run_torchbench.py Add userbenchmark support to TorchBench CI (#78794) 2022-06-03 21:07:36 +00:00
test_gitutils.py Add gitutils tests (#71580) 2022-01-20 14:33:41 -08:00
test_print_latest_commits.py Refactored print_latest_commits.py to return the latest viable commit from master (#79422) 2022-06-13 18:30:38 +00:00
test_trymerge.py [mergebot] add additional checks (#79317) 2022-06-13 19:28:29 +00:00
test_tryrebase.py rebase via comment - rebase to any branch (#78772) 2022-06-03 17:13:11 +00:00
trymerge.py Skip stale check when facebook-github-bot is merging (#79572) 2022-06-15 03:24:51 +00:00
tryrebase.py Add comment link for Job Run URL (#78823) 2022-06-03 20:31:19 +00:00
wait_for_ssh_to_drain.ps1
wait_for_ssh_to_drain.sh

pytorch/.github

NOTE: This README contains information for the .github directory but cannot be located there because it will overwrite the repo README.

This directory contains workflows and scripts to support our CI infrastructure that runs on Github Actions.

Workflows

  • Pull CI (pull.yml) is run on PRs and on master.
  • Trunk CI (trunk.yml) is run on trunk to validate incoming commits. Trunk jobs are usually more expensive to run so we do not run them on PRs unless specified.
  • Scheduled CI (periodic.yml) is a subset of trunk CI that is run every few hours on master.
  • Binary CI is run to package binaries for distribution for all platforms.

Templates

Templates written in Jinja are located in the .github/templates directory and used to generate workflow files for binary jobs found in the .github/workflows/ directory. These are also a couple of utility templates used to discern common utilities that can be used amongst different templates.

(Re)Generating workflow files

You will need jinja2 in order to regenerate the workflow files which can be installed using:

pip install -r .github/requirements.txt

Workflows can be generated / regenerated using the following command:

.github/regenerate.sh

Adding a new generated binary workflow

New generated binary workflows can be added in the .github/scripts/generate_ci_workflows.py script. You can reference examples from that script in order to add the workflow to the stream that is relevant to what you particularly care about.

Different parameters can be used to acheive different goals, i.e. running jobs on a cron, running only on trunk, etc.

ciflow (trunk)

The label ciflow/trunk can be used to run trunk only workflows. This is especially useful if trying to re-land a PR that was reverted for failing a non-default workflow.

Infra

Currently most of our self hosted runners are hosted on AWS, for a comprehensive list of available runner types you can reference .github/scale-config.yml.

Exceptions to AWS for self hosted:

  • ROCM runners

Adding new runner types

New runner types can be added by committing changes to .github/scale-config.yml. Example: https://github.com/pytorch/pytorch/pull/70474

NOTE: New runner types can only be used once the changes to .github/scale-config.yml have made their way into the default branch