OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
d0ad848aa5
pytorch/torch/csrc/jit
History
Nikita Shulga ad8aef0f98 [BE] [3/N] Use nested namespaces (#110314) 
Mostly in torch/csrc/jit/runtime and in `ATen/cuda/`

Pull Request resolved: https://github.com/pytorch/pytorch/pull/110314
Approved by: https://github.com/seemethere
2023-09-30 02:23:48 +00:00
..
api [BE] [3/N] Use nested namespaces (#110314) 2023-09-30 02:23:48 +00:00
backends [PyTorch][Coreml] Bubble up NSError from loadModel (#109444) 2023-09-19 20:08:37 +00:00
codegen removing the functionality of nvfuser python APIs (#110124) 2023-09-29 01:45:00 +00:00
cuda add additional stream priority for cuda streams (#101956) 2023-05-27 02:36:16 +00:00
docs fix typo in serialization.md (#106191) 2023-07-29 00:01:59 +00:00
frontend Revert "[1/N] Cleanup header inclusions in torch_cpu by iwyu (#101178)" 2023-09-25 20:05:25 +00:00
ir [Reland] Move torch::make_unique to std::make_unique (#109780) 2023-09-21 18:30:21 +00:00
mobile [fuzzing result][fuzz_torch_jit_lite_interpreter] read-heap-use-after-free (size 8) in std::_Function_base::_M_empty() (#110289) 2023-09-29 22:32:38 +00:00
operator_upgraders [BE] [3/N] Use nested namespaces (#110314) 2023-09-30 02:23:48 +00:00
passes removing the functionality of nvfuser python APIs (#110124) 2023-09-29 01:45:00 +00:00
python [BE] [3/N] Use nested namespaces (#110314) 2023-09-30 02:23:48 +00:00
runtime [BE] [3/N] Use nested namespaces (#110314) 2023-09-30 02:23:48 +00:00
serialization [pytorch][jit] allow passing in obj loader in unpickle api (#109730) 2023-09-26 23:50:20 +00:00
tensorexpr [LLVM] Update apis incompatible with llvm versions in codegen (#110200) 2023-09-28 21:49:30 +00:00
testing [Reland] Eliminate invocations of c10::stoi,c10::stod,c10::stoull,c10::stoll (#109566) 2023-09-19 07:15:25 +00:00
jit_log.cpp [RELAND] Remove some unnecessary <iostream> includes from headers (#108150) 2023-09-20 21:55:15 +00:00
jit_log.h
jit_opt_limit.cpp [Reland] Eliminate invocations of c10::stoi,c10::stod,c10::stoull,c10::stoll (#109566) 2023-09-19 07:15:25 +00:00
jit_opt_limit.h Fix typo in jit_opt_limit.h (#106684) 2023-08-07 13:51:55 +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 removing the functionality of nvfuser python APIs (#110124) 2023-09-29 01:45:00 +00:00
README.md
resource_guard.h

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