Name | Date | Size | #Lines | LOC | ||
---|---|---|---|---|---|---|
.. | - | - | ||||
README.md | H A D | 25-Apr-2025 | 1.2 KiB | 19 | 13 |
README.md
1This folder contains generated sources for the lazy torchscript backend. 2 3The main input file that drives which operators get codegen support for torchscript backend is 4[../../../../aten/src/ATen/native/ts_native_functions.yaml](../../../../aten/src/ATen/native/ts_native_functions.yaml) 5 6The code generator lives at `torchgen/gen_lazy_tensor.py`. 7 8It is called automatically by the torch autograd codegen (`tools/setup_helpers/generate_code.py`) 9as a part of the build process in OSS builds (CMake/Bazel) and Buck. 10 11External backends (e.g. torch/xla) call `gen_lazy_tensor.py` directly, 12and feed it command line args indicating where the output files should go. 13 14For more information on codegen, see these resources: 15* Info about lazy tensor codegen: [gen_lazy_tensor.py docs](../../../../torchgen/gen_lazy_tensor.py) 16* Lazy TorchScript backend native functions: [ts_native_functions.yaml](../../../../aten/src/ATen/native/ts_native_functions.yaml) 17* Source of truth for native func definitions [ATen native_functions.yaml](../../../../aten/src/ATen/native/native_functions.yaml) 18* Info about native functions [ATen nativefunc README.md](../../../../aten/src/ATen/native/README.md) 19