pytorch/tools
Brian Hirsh 7d710403b0 Reapply "Make functionalization ViewMeta serializable with pickle. (#143712)" (#163769)
### Summary:
NOTE: This is a re-export of https://github.com/pytorch/pytorch/pull/161994 ; the changes between these two PRs is exclusively to the buck/build files

(Summary from #161994 )
Attempted rebase of https://github.com/pytorch/pytorch/pull/143712.

This reverts commit 6c713ccb5e.

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx chenyang78 kadeng chauhang amjames Lucaskabela

imported-using-ghimport

Test Plan: Imported from OSS

Differential Revision: D81524507

Pulled By: Lucaskabela

Pull Request resolved: https://github.com/pytorch/pytorch/pull/163769
Approved by: https://github.com/dolpm

Co-authored-by: Brian Hirsh <hirsheybar@fb.com>
2025-09-25 10:27:37 +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
autograd Revert "[Reland] Use std::string_view in torchgen (#158625)" 2025-09-13 07:52: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 [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] print out tensor descriptor as comments in codegen (#163739) 2025-09-25 01:29:29 +00:00
flight_recorder [Flight Recorder][WP] Added mismatch tail as an arg (#162991) 2025-09-16 04:46:05 +00:00
gdb
github
iwyu
jit
linter [Code Clean] Remove deadcodes about Python3.9 [5/N] (#163644) 2025-09-24 07:30:50 +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
setup_helpers Reapply "Make functionalization ViewMeta serializable with pickle. (#143712)" (#163769) 2025-09-25 10:27:37 +00:00
shared
stats Track monitor (#156907) 2025-07-18 22:54:13 +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
BUCK.oss
build_libtorch.py
build_pytorch_libs.py [RELAND] Always build USE_DISTRIBUTED (#160449) and Make distributed modules importable even when backend not built (#159889) (#162594) 2025-09-22 21:12:18 +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
extract_scripts.py mypy 1.16.0 (#155821) 2025-06-14 18:18:43 +00:00
gen_flatbuffers.sh
gen_vulkan_spv.py
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
nightly.py [BE][Easy] respect os.environ in subprocess calls in tools/nightly.py (#159572) 2025-08-01 20:40:31 +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: