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
13d2af2a9b
pytorch/torch/csrc/lazy/python
History
Jiewen Tan 13d2af2a9b [LTC] Metrics can be reset too (#89606) 
Summary:
This change allow MetricsArena to ResetMetrics too. And then rename Reset to ResetCounters given that's what it does for real.

This matches pytorch/xla#4109, and is paired with pytorch/xla#4245.

Test Plan:
CI.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/89606
Approved by: https://github.com/JackCaoG
2022-11-28 21:44:12 +00:00
..
init.cpp [LTC] Metrics can be reset too (#89606) 2022-11-28 21:44:12 +00:00
init.h Revert "Revert "Add a lint rule for torch/csrc/util/pybind.h include (#82552)"" (#82599) 2022-08-02 19:37:02 +00:00
python_util.cpp Handle python frame is empty in GetPythonFrames (#83643) 2022-08-18 16:36:54 +00:00
python_util.h [lint] autoformat test/cpp and torch/csrc 2022-06-11 21:11:16 +00:00
README.md

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.