OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
e35dd4f603
pytorch/torch/csrc/jit
History
davidriazati e35dd4f603 [jit] Include call stack in OSError message (#34669) 
Summary:
Previously there was no indication of why you would get an `OSError` for something (such as the generated methods of a `dataclass`).
](https://our.intern.facebook.com/intern/diff/20426570/)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/34669

Pulled By: driazati

Differential Revision: D20426570

fbshipit-source-id: 45d63631984fa26a87c03de5523fb10d8abbc6db
2020-03-18 15:10:23 -07:00
..
api [jit] small cleanups after script:: removal (#34677) 2020-03-13 17:56:16 -07:00
codegen/fuser Delete OperatorOptions, absorb AliasAnalysisKind into FunctionSchema. (#34588) 2020-03-11 20:59:46 -07:00
docs [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
frontend [jit] Include call stack in OSError message (#34669) 2020-03-18 15:10:23 -07:00
ir [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
mobile [JIT][mobile] Support built-in Function call in lite interpreter (#34676) 2020-03-13 18:24:18 -07:00
passes [quant][graphmode] Add quantization support for aten::dropout (#34347) 2020-03-18 14:35:27 -07:00
python [jit] Include call stack in OSError message (#34669) 2020-03-18 15:10:23 -07:00
runtime Added functionality for all to take Lists as input (#34582) 2020-03-18 12:01:30 -07:00
serialization [JIT][mobile] Support built-in Function call in lite interpreter (#34676) 2020-03-13 18:24:18 -07:00
tensorexpr [TensorExpr] Pull changes from bertmaher/pytorch_fusion. (#34842) 2020-03-17 11:02:48 -07:00
testing [jit] kill script namespace (#34515) 2020-03-11 23:32:48 -07:00
jit_log.cpp [JIT] Virtualize Function (#33921) 2020-03-07 10:03:50 -08:00
jit_log.h Add traces to specialize_autograd and lower_grad_of (2nd try) 2019-09-24 09:58:43 -07:00
OVERVIEW.md Fix typo (#33925) 2020-03-02 08:13:55 -08:00
README.md [jit] add top-level readme to csrc/jit (#33916) 2020-02-27 19:21:05 -08:00
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.)