pytorch/torch/csrc/jit
Yuanyuan Chen e2dc32f4ba Replace decltype(auto) with auto (#166537)
This PR replaces `decltype(auto)` with `auto` for C++ return type deduction and simplifies some templates.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/166537
Approved by: https://github.com/Skylion007
2025-11-01 00:30:23 +00:00
..
api [2/N] Mark unused parameters in C++ code (#165121) 2025-10-15 03:04:39 +00:00
backends [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
codegen [ROCm] Reduce duplication in bfloat16_support_literal definition (#166147) 2025-10-29 16:59:03 +00:00
cuda
docs [BE][10/16] fix typos in torch/ (torch/csrc/jit/) (#156320) 2025-07-02 22:55:29 +00:00
frontend [TorchScript] clearer debug for ConcreteModuleType::findSubmoduleConcreteType (#166192) 2025-10-25 14:07:54 +00:00
ir add support for ir scalar literal parsing for inf/-inf/True/False (#163924) 2025-10-27 05:10:21 +00:00
mobile [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
operator_upgraders [BE][10/16] fix typos in torch/ (torch/csrc/jit/) (#156320) 2025-07-02 22:55:29 +00:00
passes [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
python Save Python refcount bump on each arg in maybe_handle_torch_function (#164625) 2025-10-21 02:40:12 +00:00
runtime Replace decltype(auto) with auto (#166537) 2025-11-01 00:30:23 +00:00
serialization [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
tensorexpr [2/N] Mark unused parameters in C++ code (#165121) 2025-10-15 03:04:39 +00:00
testing
jit_log.cpp
jit_log.h
jit_opt_limit.cpp
jit_opt_limit.h
JIT-AUTOCAST.md
OVERVIEW.md [3/N] fix typo in other folders (#166606) 2025-10-30 10:30:40 +00:00
README.md [BE][8/16] fix typos in torch/ (torch/csrc/jit/) (#156318) 2025-07-02 22:55:29 +00:00
resource_guard.h

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 reused by other elements of the runtime system (eager, mobile, etc.)