pytorch/tools
Guilherme Leobas e107ef5ca2 Add type annotations for torch.nn.utils.* (#43080)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/43013

Redo of gh-42954

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

Reviewed By: albanD

Differential Revision: D23681334

Pulled By: malfet

fbshipit-source-id: 20ec78aa3bfecb7acffc12eb89d3ad833024394c
2020-09-14 17:52:37 -07:00
..
amd_build [ROCm] update hip library name (#41813) 2020-07-22 09:42:45 -07:00
autograd Updates div to perform true division (#42907) 2020-09-14 15:50:38 -07:00
clang_format_hash
code_analyzer [pytorch] remove code analyzer build folder between builds (#44148) 2020-09-04 10:38:12 -07:00
code_coverage add lcov to oss for beautiful html report (#44568) 2020-09-11 15:29:24 -07:00
codegen Add unary ops: exp and sqrt (#42537) 2020-09-07 19:57:34 -07:00
config
docker
jit [pytorch] deprecate static dispatch (#43564) 2020-08-27 14:52:48 -07:00
pyi Add type annotations for torch.nn.utils.* (#43080) 2020-09-14 17:52:37 -07:00
rules remediation of S205607 2020-07-17 17:19:47 -07:00
setup_helpers Rewrite of ATen code generator (#42629) 2020-08-31 09:00:22 -07:00
shared Rewrite of ATen code generator (#42629) 2020-08-31 09:00:22 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
aten_mirror.sh
build_libtorch.py
build_pytorch_libs.py
build_variables.bzl [quant][pyper] Support quantization of ops in fork-wait subgraph (#44048) 2020-09-05 12:06:19 -07:00
clang_format_all.py
clang_format_ci.sh Removed whitelist reference from tools/clang_format_ci.sh (#41636) 2020-07-21 12:32:14 -07:00
clang_format_utils.py
clang_tidy.py
download_mnist.py
flake8_hook.py
generated_dirs.txt
git-clang-format
git-pre-commit [ONNX] Utilize ONNX shape inference for ONNX exporter (#40628) 2020-08-30 18:35:46 -07:00
git_add_generated_dirs.sh
git_reset_generated_dirs.sh
nightly.py Nightly Pull (#43294) 2020-08-20 08:34:18 -07:00
pytorch.version
README.md
update_disabled_tests.sh

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.

Legacy infrastructure (we should kill this):

  • cwrap - Implementation of legacy code generation for THNN/THCUNN. This is used by nnwrap.

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: