frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
75651e3cc4a7683970075c40334896927e4beaff
pytorch/torch/csrc/jit
History
Sam Estep 75651e3cc4 Add remaining ToCs to ToC lint (#56487) 
Summary:
The lint was originally added in https://github.com/pytorch/pytorch/issues/54974, but at the time I didn't realize that these other Markdown files also each have a table of contents:

- `GLOSSARY.md`
- `torch/csrc/jit/OVERVIEW.md`
- `torch/csrc/jit/docs/serialization.md`
- `torch/fx/OVERVIEW.md`

This PR adds those files to the lint, and also changes the rule from using a fixed list of filenames to a `git grep` command that finds all Markdown files containing this magic comment:

```md

```

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

Test Plan: The "Lint / toc" job in GitHub Actions.

Reviewed By: janeyx99

Differential Revision: D27884885

Pulled By: samestep

fbshipit-source-id: 5462437502b17fba93abf5098e21754bf566a4fe
2021-04-20 10:28:47 -07:00
..
api
Remove notion of "level" from Module::dump_to_str. (#52539)
2021-03-09 05:45:57 -08:00
backends
[AutoAccept][Codemod][FBSourceClangFormatLinter] Daily arc lint --take CLANGFORMAT
2021-04-07 10:37:27 -07:00
codegen
[reland][ROCm] use hiprtc precompiled header (#55965)
2021-04-15 15:47:56 -07:00
cuda
Merge CUDA Streams and Events (#53902)
2021-04-05 08:19:55 -07:00
docs
Add remaining ToCs to ToC lint (#56487)
2021-04-20 10:28:47 -07:00
frontend
Move graph iterator to seperate utility file (#56211)
2021-04-16 15:51:26 -07:00
ir
Lazily initialize alias db in remove_mutation opt (#55949)
2021-04-19 09:45:33 -07:00
mobile
[PyTorch Edge] Move torch::jit::mobile::_export_operator_list() from serialization/export_module.cpp to mobile/import.cpp (#56044)
2021-04-15 17:53:36 -07:00
passes
[vulkan][jit_pass] Add optimized_for_vulkan attribute on vulkan pass (#56414)
2021-04-19 17:27:59 -07:00
python
[JIT] Put explicit error message on class attribute accesses. (#55723)
2021-04-16 15:47:10 -07:00
runtime
[static runtime] support aten::__getitem__ natively (#55310)
2021-04-19 23:08:19 -07:00
serialization
[PyTorch Edge] Move torch::jit::mobile::_export_operator_list() from serialization/export_module.cpp to mobile/import.cpp (#56044)
2021-04-15 17:53:36 -07:00
tensorexpr
[NNC] Change fuseLoops API to return bool flag and not throw any exceptions (#56353)
2021-04-19 17:20:40 -07:00
testing
Revert D27448156: irange for size_t
2021-04-03 19:14:00 -07:00
jit_log.cpp
[jit] Speed up saving in case of many classes (#44589)
2020-09-15 15:10:45 -07:00
jit_log.h
added macros in jit logging to check whether loggings are enabled; replaced similar checks in LLVM codegen with such macros (#49121)
2020-12-21 13:01:22 -08:00
jit_opt_limit.cpp
Implement optimization bisect (#49031)
2021-01-11 12:25:28 -08:00
jit_opt_limit.h
Update JIT_OPT macro for easier use (#50602)
2021-01-20 11:15:20 -08:00
OVERVIEW.md
Add remaining ToCs to ToC lint (#56487)
2021-04-20 10:28:47 -07:00
README.md
[jit] add top-level readme to csrc/jit (#33916)
2020-02-27 19:21:05 -08:00
resource_guard.h
[BE] Make torch/csrc/jit/tensorexpr/ clang-tidy clean (#55628)
2021-04-08 19:44:14 -07:00

README.md

PyTorch JIT

This folder contains (most of) the C++ code for the PyTorch JIT, a language and compiler stack for executing PyTorch models portably and efficiently. To learn more about the JIT from a user perspective, please consult our reference documentation and tutorials.

A brief summary of the source tree:

  • OVERVIEW.md: High-level technical overview of the JIT.
  • frontend/: Taking PyTorch modules in Python and translating them into the JIT IR.
  • ir/: Core IR abstractions.
  • runtime/: Interpreter, graph execution, and JIT operators.
  • codegen/: Generating efficient, hardware-specific code for JIT subgraphs.
  • serialization/: Saving and loading modules.
  • api/: Any user-facing C++ or Python interfaces.
  • python/: Binding stuff into Python or accessing information from the Python environment.
  • testing/: Utilities and helpers for testing.
  • mobile/: Mobile-specific implementations of runtime components.
  • passes/: IR-to-IR passes, generally for optimization and lowering.
  • generated/: This folder is generated by the PyTorch build, and contains bindings for native PyTorch operators into the JIT.

Refer to each folder for more in-depth documentation.

Other relevant parts of the codebase not contained here:

  • aten/src/ATen/core: contains JIT code re-used by other elements of the runtime system (eager, mobile, etc.)