pytorch/tools
Edward Z. Yang 213a8fc992 Put symint overloads on a different name
Due to implicit conversion shenanigans, having both IntArrayRef
and SymIntArrayRef overloads makes {} ambiguous.  While we could
fix this by making a single unified type that accepts all the overloads
we want, an easier fix was to just push the SymIntArrayRef overload
to its own name.

Signed-off-by: Edward Z. Yang <ezyangfb.com>

Pull Request resolved: https://github.com/pytorch/pytorch/pull/79281

Approved by: https://github.com/suo
2022-06-12 14:36:39 +00:00
..
amd_build [PyTorch][AMD] fix hipify_python (#76720) 2022-05-03 22:59:10 +00:00
autograd Put symint overloads on a different name 2022-06-12 14:36:39 +00:00
build_defs [Test] create shared targets for xplat aten (#78345) 2022-06-08 22:20:03 +00:00
code_analyzer Re-land BUCK build for pytorch mobile (#77612) 2022-05-17 00:30:13 +00:00
code_coverage [codemod][usort] apply import merging for fbcode (1 of 11) (#78973) 2022-06-06 23:44:28 +00:00
config [bazel] GPU-support: add @local_config_cuda and @cuda (#63604) 2021-08-27 09:33:42 -07:00
coverage_plugins_package Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
fast_nvcc Revert autoformat of tools/fast_nvcc/fast_nvcc.py 2022-05-12 04:02:46 +00:00
gdb Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
iwyu Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
jit Revert "[mobile] Fix lightweight dispatch OOM error by introducing selective build" 2022-06-09 05:16:42 +00:00
linter Add option to skip binaries when doing pip install for lintrunner (#78668) 2022-06-02 20:50:57 +00:00
lite_interpreter rename BUILD.buck to BUCK.oss (#78792) 2022-06-03 07:23:16 +00:00
lldb Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
onnx [ONNX] Refactor to remove inline imports - attempt 2 (#77448) 2022-05-16 14:44:24 +00:00
pyi ENH: Add a force argument to numpy() (#78564) 2022-06-06 14:14:17 +00:00
rules Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
setup_helpers Revert "[mobile] Fix lightweight dispatch OOM error by introducing selective build" 2022-06-09 05:16:42 +00:00
shared [lint] upgrade mypy to latest version 2022-05-03 20:51:34 +00:00
stats [ci] fix GITHUB_ACTIONS env var checks 2022-06-11 17:16:30 +00:00
test [ci] write test suites to rockset 2022-06-10 15:38:09 +00:00
testing Allow sharding for distributed tests 2022-04-29 03:55:07 +00:00
__init__.py
bazel.bzl add support for filtering out Bazel targets from common structure (#76173) 2022-05-03 12:13:19 +00:00
build_libtorch.py Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
build_pytorch_libs.py Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
cpuinfo_target_definition.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
download_mnist.py Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
extract_scripts.py Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
gen_flatbuffers.sh Update generated header to use flatbuffer v1.12; (#71279) 2022-01-13 17:23:30 -08:00
generate_torch_version.py tools: Add ability to grab release versions 2022-06-01 17:19:17 +00:00
generated_dirs.txt
git_add_generated_dirs.sh
git_reset_generated_dirs.sh
miniz_target_definition.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
nightly.py Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
nvcc_fix_deps.py Add workaround for nvcc header dependecies bug (#62550) 2021-10-11 09:07:12 -07:00
perf_kernel_defs.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
pytorch.version
README.md [ci] delete old linter stuff 2022-05-11 07:40:44 +00:00
render_junit.py Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
sgx_aten_target_definitions.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
sgx_caffe2_target_definitions.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
sgx_target_definitions.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
substitute.py move substitute lib to open source (#79093) 2022-06-08 23:00:47 +00:00
target_definitions.bzl Add all bzl files per D36874458 2022-06-06 09:40:19 -07:00
update_masked_docs.py Run Black on all of tools/ 2022-04-20 17:29:41 +00:00
vscode_settings.py Run Black on all of tools/ 2022-04-20 17:29:41 +00:00

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.
  • fast_nvcc - Mostly-transparent wrapper over nvcc that parallelizes compilation when used to build CUDA files for multiple architectures at once.
    • fast_nvcc.py - Python script, entrypoint to the fast nvcc wrapper.

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: