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 37 Packages Projects Releases Wiki Activity
908daa8ae5
pytorch/torch/csrc/jit
History
David Berard 908daa8ae5 [nvfuser] avoid out of bounds error (#89584) 
Summary: update OOB check (https://github.com/csarofeen/pytorch/pull/2218) and skip tests that OOM on internal machines.

Test Plan:
```
buck2 test mode/dev-nosan //caffe2/torch/csrc/jit/codegen/cuda/test:nvfuser
```

Differential Revision: D41502369

Pull Request resolved: https://github.com/pytorch/pytorch/pull/89584
Approved by: https://github.com/jjsjann123
2022-11-29 02:03:59 +00:00
..
api
backends [xnnpack][lite-int] Handle Constant Data (#89445) 2022-11-22 02:20:54 +00:00
codegen [nvfuser] avoid out of bounds error (#89584) 2022-11-29 02:03:59 +00:00
cuda
docs Fix typo under torch directory (#87274) 2022-10-21 14:22:20 +00:00
frontend [JIT][Security] Do not blindly eval input string (#89189) 2022-11-17 22:05:30 +00:00
ir Fix typos in messages under torch (#88961) 2022-11-14 19:06:41 +00:00
mobile Fix typos in messages under torch (#88961) 2022-11-14 19:06:41 +00:00
operator_upgraders Fix typos in .md and .rst files (#88962) 2022-11-17 03:37:02 +00:00
passes quantization: deprecate observer compute_dtype and replace with is_dynamic (#85431) 2022-11-24 07:07:34 +00:00
python Use the Python frame safely in _pythonCallstack (#88993) 2022-11-17 00:59:15 +00:00
runtime Back out "[static-runtime] change the backend for permute_copy" (#89463) 2022-11-22 06:26:10 +00:00
serialization Fix typos in messages under torch (#88961) 2022-11-14 19:06:41 +00:00
tensorexpr nnc: fix Store if value is fp32 while buf is bf16 (#86788) 2022-11-24 02:52:34 +00:00
testing
jit_log.cpp
jit_log.h
jit_opt_limit.cpp
jit_opt_limit.h
JIT-AUTOCAST.md
OVERVIEW.md Fix typos under torch directory (#88172) 2022-11-01 22:58:22 +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.)