pytorch/tools
Catherine Lee 946b50c788 [ez][TD] Increase logging (#124082)
increase logging during td
generate an artifact that says which tests got excluded
fix minor bug where filter test configs couldnt get commit messages

Pull Request resolved: https://github.com/pytorch/pytorch/pull/124082
Approved by: https://github.com/seemethere, https://github.com/malfet
2024-04-17 00:18:28 +00:00
..
alerts Revert "add oncall info individual info to failing alert job alert (#103915)" 2023-06-21 23:06:45 +00:00
amd_build [aot_inductor] move CppWrapperCodeGen into a separate file (#119871) 2024-02-16 08:14:20 +00:00
autograd Add mem-eff-attention's sliding window arg to align with xformers (#123571) 2024-04-09 02:19:12 +00:00
bazel_tools
build/bazel [CI] Update bazel deps (#124076) 2024-04-15 20:39:50 +00:00
build_defs Fix buck OSS build after #115570 (#115804) 2023-12-14 08:33:07 +00:00
code_analyzer [BE]: Enable RUFF PERF402 and apply fixes (#115505) 2023-12-20 18:01:24 +00:00
code_coverage [Reland] Update mypy to 1.4.1 (#105227) 2023-07-15 20:30:20 +00:00
config
coverage_plugins_package [BE] Enable ruff's UP rules and autoformat tools and scripts (#105428) 2023-07-19 01:24:44 +00:00
dynamo [dynamo, 3.12] enable tests disabled due to missing dynamo 3.12 support (#123300) 2024-04-05 20:13:17 +00:00
gdb [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
github fix upload alerts to rockset (#103995) 2023-06-22 23:33:10 +00:00
iwyu [2/N] Cleanup header inclusions in torch_cpu by iwyu (#109964) 2023-11-19 20:56:32 +00:00
jit [BE] Enable ruff's UP rules and autoformat tools and scripts (#105428) 2023-07-19 01:24:44 +00:00
linter Use uv in lintrunner init when it is available. (#124033) 2024-04-15 14:47:21 +00:00
lite_interpreter removing some redundant str splits (#106089) 2023-09-01 00:22:58 +00:00
lldb
onnx [torch/csrc/onnx] Use nested namespaces (3/N) (#113993) 2023-11-18 00:20:19 +00:00
pyi [AOTAutograd] Replay views on output using FunctionalTensor metas. (#121007) 2024-04-12 16:54:13 +00:00
rules
rules_cc
setup_helpers [Build] Add linker script optimization (#121975) 2024-04-09 20:22:25 +00:00
shared Test Reordering: Run previously failing tests first (#101123) 2023-05-16 19:57:54 +00:00
stats TD outside of test job (#118250) 2024-03-01 23:08:10 +00:00
test Fix round robin sharding (#121022) 2024-03-11 17:30:12 +00:00
testing [ez][TD] Increase logging (#124082) 2024-04-17 00:18:28 +00:00
__init__.py
bazel.bzl update rules_python and let bazel install its own pip dependencies (#101405) 2023-05-23 06:20:33 +00:00
BUCK.bzl Reify view_func() closures as ViewFuncs (#118404) 2024-02-14 22:00:43 +00:00
BUCK.oss
build_libtorch.py
build_pytorch_libs.py
build_with_debinfo.py [EZ] Fix argument parsing in build_with_debinfo (#120088) 2024-02-16 20:06:52 +00:00
download_mnist.py [BE] f-stringify torch/ and scripts (#105538) 2023-07-21 19:35:24 +00:00
extract_scripts.py
gen_flatbuffers.sh
gen_vulkan_spv.py [ET-VK] Enable Dynamic shape support via tensor virtual and physical resizing (#121598) 2024-03-12 14:32:00 +00:00
generate_torch_version.py Support for sapling scm (#122072) 2024-03-22 22:59:16 +00:00
generated_dirs.txt
git_add_generated_dirs.sh
git_reset_generated_dirs.sh
nightly.py removing some redundant str splits (#106089) 2023-09-01 00:22:58 +00:00
nvcc_fix_deps.py [BE][Ez]: FURB129: remove unneeded readlines() (#119796) 2024-02-13 21:21:22 +00:00
pytorch.version
README.md
render_junit.py
substitute.py [Codemod][python/main_function] caffe2: (#113357) 2023-11-15 22:17:31 +00:00
update_masked_docs.py Use strict to toggle strict options in MYPYSTRICT (#118479) 2024-01-28 19:22:22 +00:00
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: