pytorch/tools
Yuanyuan Chen a029675f6f More ruff SIM fixes (#164695)
This PR applies ruff `SIM` rules to more files. Most changes are about simplifying `dict.get` because `None` is already the default value.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164695
Approved by: https://github.com/ezyang
2025-10-09 03:24:50 +00:00
..
alerts More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
amd_build [ROCm] Fix test_cuda_synchronize failure on ROCm (#164735) 2025-10-07 01:14:24 +00:00
autograd More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
bazel_tools [BE] fix typos in tools/ (#156082) 2025-06-17 19:25:50 +00:00
build/bazel [CI] update flake8 and mypy lint dependencies (#158720) 2025-07-29 08:05:56 +00:00
build_defs [BE] fix typos in tools/ (#156082) 2025-06-17 19:25:50 +00:00
code_analyzer [BE] fix typos in tools/ (#156082) 2025-06-17 19:25:50 +00:00
code_coverage [BE] fix typos in tools/ (#156082) 2025-06-17 19:25:50 +00:00
config [BE] fix typos in tools/ (#156082) 2025-06-17 19:25:50 +00:00
coverage_plugins_package More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
dynamo [Code Clean] Remove deadcodes about Python3.9 [8/N] (#163728) 2025-09-25 05:12:46 +00:00
experimental/torchfuzz [torchfuzz] synthesize inputs for data dependent ops (#164716) 2025-10-07 05:40:32 +00:00
flight_recorder More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
gdb Add gdb print methods support same as pytorch-lldb (#140935) 2024-11-19 01:28:30 +00:00
github [BE][CI] bump ruff to 0.9.0: string quote styles (#144569) 2025-02-24 19:56:09 +00:00
iwyu [BE][CI] bump ruff to 0.9.0: string quote styles (#144569) 2025-02-24 19:56:09 +00:00
jit [Fix] Adding missing f prefixes to formatted strings [3/N] (#164067) 2025-09-29 04:35:23 +00:00
linter [Fix] Adding missing f prefixes to formatted strings [3/N] (#164067) 2025-09-29 04:35:23 +00:00
lite_interpreter Move use of c10::string_view to std::string_view (#152509) 2025-06-25 01:57:49 +00:00
lldb [BE] Remove torch deploy | remove torch deploy specific files (#158290) 2025-07-30 01:36:03 +00:00
packaging Replace setup.py bdist_wheel with python -m build --wheel (#156712) 2025-09-29 21:51:32 +00:00
pyi Add api info for torch._C._nn.pyi (#162936) 2025-09-24 04:55:57 +00:00
rules
rules_cc [BE] Fix incompatible-std-redefinition warning (#141630) 2024-11-27 05:06:36 +00:00
setup_helpers More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
shared
stats [testing] Better short job name during upload additional stats (#164287) 2025-10-01 19:56:20 +00:00
test docstring_linter: Fix #151692 and other issues (#156596) 2025-08-25 16:04:14 +00:00
testing More ruff SIM fixes (#164695) 2025-10-09 03:24:50 +00:00
__init__.py
bazel.bzl Move version.h to torch/headeronly (#164381) 2025-10-07 17:47:30 +00:00
BUCK.bzl [BE] Cleanup old ExecuTorch codegen and runtime code (#154165) 2025-06-07 06:54:12 +00:00
BUCK.oss
build_libtorch.py [BE][Easy] use pathlib.Path instead of dirname / ".." / pardir (#129374) 2024-12-29 17:23:13 +00:00
build_pytorch_libs.py Revert "[RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594)" 2025-09-25 13:47:46 +00:00
build_with_debinfo.py [build] modernize build-frontend: python setup.py develop/install -> [uv ]pip install --no-build-isolation [-e ]. (#156027) 2025-07-09 11:24:27 +00:00
download_mnist.py [BE] Remove not existing mnist mirror (#164238) 2025-09-30 15:15:13 +00:00
extract_scripts.py mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
gen_flatbuffers.sh
gen_vulkan_spv.py [BE][Ez]: FURB148 - remove useless enumerate calls (#145619) 2025-01-24 23:37:15 +00:00
generate_torch_version.py Add sdist handling to version finding (#160315) 2025-09-25 07:15:51 +00:00
generated_dirs.txt
git_add_generated_dirs.sh
git_reset_generated_dirs.sh
nightly_hotpatch.py [3/N] Apply py39 ruff fixes (#142115) 2024-12-11 17:50:10 +00:00
nightly.py Better path handling for nightly setup tool (#164215) 2025-10-02 07:59:17 +00:00
nvcc_fix_deps.py [BE] fix typos in tools/ (#156082) 2025-06-17 19:25:50 +00:00
optional_submodules.py [BE] Make Eigen an optional dependency (#155955) 2025-06-21 03:02:02 +00:00
README.md
render_junit.py
substitute.py
update_masked_docs.py
vscode_settings.py

This folder contains a number of scripts which are used as part of the PyTorch build process. This directory also doubles as a Python module hierarchy (thus the __init__.py).

Overview

Modern infrastructure:

  • autograd - Code generation for autograd. This includes definitions of all our derivatives.
  • jit - Code generation for JIT
  • shared - Generic infrastructure that scripts in tools may find useful.
    • module_loader.py - Makes it easier to import arbitrary Python files in a script, without having to add them to the PYTHONPATH first.

Build system pieces:

  • setup_helpers - Helper code for searching for third-party dependencies on the user system.
  • build_pytorch_libs.py - cross-platform script that builds all of the constituent libraries of PyTorch, but not the PyTorch Python extension itself.
  • build_libtorch.py - Script for building libtorch, a standalone C++ library without Python support. This build script is tested in CI.

Developer tools which you might find useful:

Important if you want to run on AMD GPU:

  • amd_build - HIPify scripts, for transpiling CUDA into AMD HIP. Right now, PyTorch and Caffe2 share logic for how to do this transpilation, but have separate entry-points for transpiling either PyTorch or Caffe2 code.
    • build_amd.py - Top-level entry point for HIPifying our codebase.

Tools which are only situationally useful: