pytorch/torch/csrc/lazy/python
2023-11-07 15:53:32 +00:00
..
init.cpp Revert "Add support for torch.Generator type in TorchScript (#110413)" 2023-11-07 15:53:32 +00:00
init.h Cleanup Windows warning suppression in CMake and fix some warnings in the source code (#94927) 2023-02-27 19:22:20 +00:00
python_util.cpp
python_util.h Cleanup Windows warning suppression in CMake and fix some warnings in the source code (#94927) 2023-02-27 19:22:20 +00:00
README.md

Lazy Tensor Python Code

Lazy Tensor Core is part of libtorch, which can not depend on python.

Parts of lazy tensor core use python for 2 purposes A) py bindings let python programs call into lazy tensor c++ code B) lazy tensor core calls into python to use it (e.g. for grabbing stack traces)

(A) is trivial since the python bindings only depend on libtorch; (B) requires making libtorch_python register a function with libtorch if loaded, and having a default (no-op) function otherwise. Any functionality that strictly needs to depend on python should be part of the 'python' folder.