frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 12:54:11 +08:00
Code Issues Packages Projects Releases Wiki Activity
Files
501d0729cb7cfd25aebb2c0e82cef035bf4e7e78
pytorch/torch/csrc/jit
History
dzdang 5874a31169 [quant][core][better-engineering] Rename files in quantized/cpu directory to conform with non-quantized countertpart filenames 
Summary:
Names of analogous files in quantized directory (previously snake case) were inconsistent with
their non-quantized filename counterparts (pascal case). This is the second of a series of PRs that changes
all files in quantized (and sub-directories) dir to have pascal case.

Some files have not been renamed as it is causing issues related to
custom class with `import torch` at runtime. See
https://github.com/pytorch/pytorch/pull/77037 for additional details

Test Plan:
```
python test/test_quantization.py
```

Reviewers:

Subscribers:

Tasks:

Tags:

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77422

Approved by: https://github.com/jerryzh168
2022-06-01 21:20:30 +00:00
..
api
Add option to disable autocast pass
2022-05-18 14:57:25 +00:00
backends
[coreml] Use special throw macro when encountering CoreML API errors (#77429)
2022-05-16 18:02:48 +00:00
codegen
retry - enable NVFuser by default
2022-05-20 14:21:18 +00: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
Static initializer update (#78052)
2022-05-22 23:14:03 +00:00
ir
Retry "[JIT] parse prim::Constant[value=annotate()] and prim::Constant[value={0}]"
2022-05-13 15:12:07 +00:00
mobile
Populate bytecode version and operator version (#77685)
2022-05-19 23:51:51 +00:00
operator_upgraders
Move other div variants to upgraders map
2022-05-16 22:32:15 +00:00
passes
[quant][core][better-engineering] Rename files in quantized/cpu directory to conform with non-quantized countertpart filenames
2022-06-01 21:20:30 +00:00
python
[PyTorchEdge] Extend Flatbuffer to get mobile_info for NMLML workflows
2022-06-01 20:09:09 +00:00
runtime
Revert "Port index.Tensor to structured kernels."
2022-06-01 00:12:15 +00:00
serialization
Temp fix for upgrader (#78589)
2022-06-01 01:31:51 +00:00
tensorexpr
[quant][core][better-engineering] Rename files in quantized/cpu directory to conform with non-quantized countertpart filenames
2022-06-01 21:20:30 +00:00
testing
Reland "Make debug_pkl smaller by only emitting unique traces." (#73368)
2022-04-18 22:34:21 +00:00
jit_log.cpp
Remove copies in jit_log.cpp (#67841)
2022-01-25 20:32:12 +00:00
jit_log.h
[JIT] script & logging for extracting IR from logs (#72889)
2022-03-02 18:34:35 +00:00
jit_opt_limit.cpp
Implement optimization bisect (#49031)
2021-01-11 12:25:28 -08:00
jit_opt_limit.h
Remove WindowsTorchApiMacro.h in favor of Export.h (#69585)
2021-12-09 17:30:09 -08:00
JIT-AUTOCAST.md
Add fp16/fp32 autocasting to JIT/TorchScript (#63939)
2021-10-27 12:11:36 -07:00
OVERVIEW.md
aliasing fixes (#66977)
2021-11-09 18:33:37 -08: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.)