pytorch/torch/csrc/jit
2024-10-17 16:23:36 +00:00
..
api Use C10_UNUSED instead of (void)X (#137239) 2024-10-15 14:32:59 +00:00
backends [Reland] Check function declarations of COREML code (#136070) 2024-09-26 03:52:06 +00:00
codegen [22/N] Fix clang-tidy warnings in jit (#134829) 2024-09-19 19:24:42 +00:00
cuda [20/N] Fix clang-tidy warnings in jit (#133399) 2024-08-26 17:43:52 +00:00
docs Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
frontend Revert "Avoid some dangling reference warnings (#132535)" 2024-10-17 16:23:36 +00:00
ir Revert "Avoid some dangling reference warnings (#132535)" 2024-10-17 16:23:36 +00:00
mobile Revert "Avoid some dangling reference warnings (#132535)" 2024-10-17 16:23:36 +00:00
operator_upgraders Add None return type to init (#132335) 2024-08-01 15:26:45 +00:00
passes Revert "Avoid some dangling reference warnings (#132535)" 2024-10-17 16:23:36 +00:00
python Revert "Expose option to disable CRC-32 computation during torch.save (#137735)" 2024-10-16 17:03:06 +00:00
runtime Revert "Avoid some dangling reference warnings (#132535)" 2024-10-17 16:23:36 +00:00
serialization Revert "Avoid some dangling reference warnings (#132535)" 2024-10-17 16:23:36 +00:00
tensorexpr Revert "Avoid some dangling reference warnings (#132535)" 2024-10-17 16:23:36 +00:00
testing Fix the output of FileCheck when not run and add unit tests (#135345) 2024-09-11 04:13:24 +00:00
jit_log.cpp [20/N] Fix clang-tidy warnings in jit (#133399) 2024-08-26 17:43:52 +00:00
jit_log.h [8/N] Fix clang-tidy warnings in jit (#131997) 2024-07-29 12:40:42 +00:00
jit_opt_limit.cpp [8/N] Fix clang-tidy warnings in jit (#131997) 2024-07-29 12:40:42 +00:00
jit_opt_limit.h [8/N] Fix clang-tidy warnings in jit (#131997) 2024-07-29 12:40:42 +00:00
JIT-AUTOCAST.md Fix missing mandatory device_type argument in autocast docstring (#97223) 2023-06-27 01:54:54 +00:00
OVERVIEW.md c10::optional -> std::optional in PyTorch (#137333) 2024-10-11 00:16:10 +00:00
README.md
resource_guard.h [8/N] Fix clang-tidy warnings in jit (#131997) 2024-07-29 12:40:42 +00:00

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.)