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
bc68625151
pytorch/torch/csrc/jit
History
Salil Desai bc68625151 [Vulkan] Add support for Optimization Blocklist to Vulkan Rewrite (#87431) 
Optimization Blocklist will be used in a future diff (D40315730) to make the rewrite to transfer input/output backends optional

Differential Revision: [D40315729](https://our.internmc.facebook.com/intern/diff/D40315729/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/87431
Approved by: https://github.com/mcr229, https://github.com/digantdesai
2022-10-31 14:15:51 +00:00
..
api
backends [xnnpack][lite-int][graph-build] graph passes and op checking (#87128) 2022-10-25 22:08:29 +00:00
codegen Check all CUDA API calls for errors in torch/ (#81560) 2022-10-28 00:40:48 +00:00
cuda
docs Fix typo under torch directory (#87274) 2022-10-21 14:22:20 +00:00
frontend Enable max.unary_out (#86855) 2022-10-13 17:14:53 +00:00
ir Add 'share_from_this' to 'torch::jit::Graph' (#87343) 2022-10-28 23:51:44 +00:00
mobile Revert "Back out "Revert D40198461: [pytorch][PR] Backport currently dont work with some models if:" (#87124)" 2022-10-21 16:03:00 +00:00
operator_upgraders Fix typo under torch directory (#87274) 2022-10-21 14:22:20 +00:00
passes [Vulkan] Add support for Optimization Blocklist to Vulkan Rewrite (#87431) 2022-10-31 14:15:51 +00:00
python [Vulkan] Add support for Optimization Blocklist to Vulkan Rewrite (#87431) 2022-10-31 14:15:51 +00:00
runtime Fix bugs found by static analysis (#85705) 2022-10-28 23:51:55 +00:00
serialization [BE] Delete TH_DISALLOW_COPY_AND_ASSIGN (#87743) 2022-10-26 03:31:56 +00:00
tensorexpr Fix bugs found by static analysis (#85705) 2022-10-28 23:51:55 +00:00
testing
jit_log.cpp
jit_log.h
jit_opt_limit.cpp
jit_opt_limit.h
JIT-AUTOCAST.md [JIT][Autocast] document that scripted autocast context cannot disable eager-enabled autocast (#81747) 2022-07-21 17:44:44 +00:00
OVERVIEW.md Fix typo under torch directory (#87274) 2022-10-21 14:22:20 +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.)