pytorch/tools
mfkasim91 6eaa324c9f Implement torch.igamma (#46183)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/41637
This is regularized lower incomplete gamma function, equivalent to scipy's `gammainc` and tensorflow `igamma`.

cc fritzo mruberry

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

Reviewed By: gchanan

Differential Revision: D24479126

Pulled By: mruberry

fbshipit-source-id: fdf8ea289fe4ca1b408810732192411e948fcdfe
2020-10-29 11:40:18 -07:00
..
amd_build Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00
autograd Implement torch.igamma (#46183) 2020-10-29 11:40:18 -07:00
clang_format_hash
code_analyzer Make add_relu an internal function (#46676) 2020-10-22 18:08:15 -07:00
code_coverage add lcov to oss for beautiful html report (#44568) 2020-09-11 15:29:24 -07:00
codegen [pytorch] get rid of cpp_type_str from pybind codegen (#46977) 2020-10-28 21:25:55 -07:00
config
docker
jit [RFC] Switch PyTorch Selective Build (Custom Build) to use the SelectiveBuilder abstraction (#45722) 2020-10-18 15:10:42 -07:00
pyi [py][vulkan][reland] Add is_vulkan to py api, add vulkan to device type parsing (#46655) 2020-10-22 09:35:50 -07:00
rules remediation of S205607 2020-07-17 17:19:47 -07:00
setup_helpers [pytorch] rewrite of the python binding codegen with the v2 API (#46244) 2020-10-19 17:36:45 -07:00
shared Replace map(lambda constructs (#46462) 2020-10-22 09:50:22 -07:00
__init__.py remediation of S205607 2020-07-17 17:19:47 -07:00
aten_mirror.sh
build_libtorch.py Remove Incorrect Comment in tools/build_libtorch and remove Python2 support in the module import (#44888) 2020-09-18 10:03:36 -07:00
build_pytorch_libs.py
build_variables.bzl [Gradient Compression] Provide 2 default C++ comm hooks (#46701) 2020-10-27 21:43:15 -07:00
clang_format_all.py Replaced whitelist with allowlist (#45796) 2020-10-06 09:18:51 -07:00
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 Remove __future__ imports for legacy Python2 supports (#45033) 2020-09-23 17:57:02 -07:00
download_mnist.py Remove Incorrect Comment in tools/build_libtorch and remove Python2 support in the module import (#44888) 2020-09-18 10:03:36 -07:00
flake8_hook.py
generate_torch_version.py Fix torch.version.debug generation (#47006) 2020-10-28 12:48:30 -07:00
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 robustness fixes for linking across devices (#43771) 2020-09-21 12:32:32 -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: