pytorch/tools
Catherine Lee c332d58184 [testing] upload test stats: Add info to the invoking file summary and some other changes (#164016)
* Changes some internal logic for grouping so hopefully it's slightly less annoying write code for
* Changes the invoking file summary to just use file, which I think is correct most of the time
* Adds some fields to the file summary, like skips, errors, etc so I can reuse it for file report regression things

Output should be the same, maybe with slightly more fields since I got rid of some of the pops

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164016
Approved by: https://github.com/huydhn
2025-09-29 21:20:18 +00:00
..
alerts [BE][Easy] do not install yanked types-pkg-resources in lint environment (#156462) 2025-06-20 16:00:43 +00:00
amd_build Add a stable TORCH_LIBRARY to C shim (#148124) 2025-03-11 19:12:46 +00:00
autograd Add magic TORCH_MAKE_PYBIND_ENUM_FASTER macro (#163527) 2025-09-26 17:59:22 +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 [BE] fix PYPROJECT linting errors in test/ and tools/ (#156021) 2025-06-20 06:19:05 +00:00
dynamo [Code Clean] Remove deadcodes about Python3.9 [8/N] (#163728) 2025-09-25 05:12:46 +00:00
experimental/dynamic_shapes/torchfuzz [torchfuzz] fix bool propagation (#164003) 2025-09-27 04:51:29 +00:00
flight_recorder [fr] Skip the dtype check for some one to all or all to one collective (#163839) 2025-09-25 16:02:06 +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 [BE][EZ] Delete remains of split-build logic (#159990) 2025-08-07 01:59:30 +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 Reapply "Make functionalization ViewMeta serializable with pickle. (#143712)" (#163769) 2025-09-25 10:27:37 +00:00
shared
stats [testing] upload test stats: Add info to the invoking file summary and some other changes (#164016) 2025-09-29 21:20:18 +00:00
test docstring_linter: Fix #151692 and other issues (#156596) 2025-08-25 16:04:14 +00:00
testing [BE] Remove bottleneck (#163210) 2025-09-18 12:08:13 +00:00
__init__.py
bazel.bzl Migrate c10/macros/cmake_macros.h.in to torch/headeronly (#158035) 2025-07-15 19:52:59 +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 Fix broken URLs (#152237) 2025-04-27 09:56:42 +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 [ROCm][CI/CD] upgrade nightly wheels to ROCm 7.0 (#163937) 2025-09-26 21:42:09 +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: