pytorch/tools
Allen Goodman 64e0d0c4fe Laguerre polynomial (#78366)
Adds:

```Python
laguerre_polynomial_l(input, n, *, out=None) -> Tensor
```

Laguerre polynomial $L_{n}(\text{input})$.

## Derivatives

Recommended $k$-derivative formula with respect to $\text{input}$:

$$\frac{d^{k}}{d \times \text{input}^{k}} L_{n}(\text{input}) = -1^{k} \times L_{-k + n}^{k}(\text{input})$$

where $L_{n}^{\alpha}$ is the associated Laguerre polynomial.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/78366
Approved by: https://github.com/mruberry
2022-05-30 17:24:00 +00:00
..
amd_build [PyTorch][AMD] fix hipify_python (#76720) 2022-05-03 22:59:10 +00:00
autograd Laguerre polynomial (#78366) 2022-05-30 17:24:00 +00:00
build_defs Re-land BUCK build for pytorch mobile (#77612) 2022-05-17 00:30:13 +00:00
code_analyzer Re-land BUCK build for pytorch mobile (#77612) 2022-05-17 00:30:13 +00:00
code_coverage fix 'pytorch/tools/code_coverage/README.md' for renamed options 2022-04-08 15:57:40 +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 default argument handling for mobile unboxing codegen 2022-05-27 06:41:23 +00:00
linter Make s3_init linter adapter work in abscense of git (#77466) 2022-05-14 00:23:36 +00:00
lite_interpreter Re-land BUCK build for pytorch mobile (#77612) 2022-05-17 00:30:13 +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 Fix a few issues on assert/double error/legacy constructor (#77966) 2022-05-20 20:25:12 +00:00
rules [Bazel] Add CUDA build to CI (#66241) 2021-12-17 13:44:29 -08:00
setup_helpers Pass WITH_BLAS option from environment to CMake (#78037) 2022-05-25 13:38:41 +00:00
shared [lint] upgrade mypy to latest version 2022-05-03 20:51:34 +00:00
stats [ci] improve upload test stats logging 2022-05-24 17:32:20 +00:00
test Codegen Non-Native IR Nodes (#76535) 2022-05-24 19:29:23 +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
build_variables.bzl [PyTorch] Integrate Execution Graph Observer into PyTorch Profiler (#75358) 2022-05-26 08:06:27 +00: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 Revert "tools: add ability to grab git tag (#77279)" 2022-05-13 20:07:38 +00:00
generated_dirs.txt
git_add_generated_dirs.sh
git_reset_generated_dirs.sh
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
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
ufunc_defs.bzl ufunc codegen (#65851) 2022-03-01 00:33:40 +00: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: