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
12daa4f663
pytorch/torch/csrc/jit
History
Zhengxu Chen 12daa4f663 [jit][edge] Enable CALL instruction in lite interpreter. (#65964) 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/65964

ghstack-source-id: 141425519

Test Plan: buck run xplat/caffe2:test_lite_interpreter

Reviewed By: cccclai

Differential Revision: D31326149

fbshipit-source-id: 8a599d92f3fa4e6c125100adb36d89592e71e547
2021-10-25 14:44:33 -07:00
..
api [PyTorch] Add tuple inline storage (#64066) 2021-10-15 12:16:51 -07:00
backends [iOS][CoreML] Check backend availability at runtime. (#65315) 2021-09-22 15:38:53 -07:00
codegen Deduplicate codegenOutputQuery to query maximum CUDA compute capabilities (#55901) 2021-10-18 07:42:15 -07:00
cuda Make pytorch clang-tidy clean (#60649) 2021-07-01 12:21:07 -07:00
docs s/foward/forward/g (#58497) 2021-05-19 11:42:42 -07:00
frontend Add SourceView which doesn't own source text as base class of Source (#65309) 2021-10-18 23:17:22 -07:00
ir [JIT] Add special cases for batch_norm, instance_norm in alias_analysis (#66554) 2021-10-20 10:22:10 -07:00
mobile [jit][edge] Enable CALL instruction in lite interpreter. (#65964) 2021-10-25 14:44:33 -07:00
passes [JIT] Freeze allows preservation of submodule attributes (#66102) 2021-10-25 07:56:20 -07:00
python jit trace (#59949) 2021-10-24 18:04:22 -07:00
runtime [jit][edge] Enable CALL instruction in lite interpreter. (#65964) 2021-10-25 14:44:33 -07:00
serialization [ONNX] Export nn.Module call as ONNX local function (#63589) (#66140) 2021-10-22 13:44:56 -07:00
tensorexpr Revert D31514095: Use kernel_func_name from aotCompiler 2021-10-23 17:17:53 -07:00
testing Add SourceView which doesn't own source text as base class of Source (#65309) 2021-10-18 23:17:22 -07:00
jit_log.cpp Added jit log stream changing function and some refactor (#65768) 2021-09-30 23:25:11 -07:00
jit_log.h Added jit log stream changing function and some refactor (#65768) 2021-09-30 23:25:11 -07:00
jit_opt_limit.cpp
jit_opt_limit.h
OVERVIEW.md Support Union in TorchScript (#64234) 2021-09-03 06:12:24 -07:00
README.md [JIT] improve documentation (#57991) 2021-05-19 11:47:32 -07:00
resource_guard.h Fix modernize-use-equals-default nolint failures in torch/csrcs (#61142) 2021-07-06 09:46:46 -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.)