xref: /aosp_15_r20/external/pytorch/torch/csrc/lazy/python/python_util.h (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1 #pragma once
2 #include <torch/csrc/Export.h>
3 #include <torch/csrc/lazy/core/ir_metadata.h>
4 #include <optional>
5 #include <vector>
6 
7 namespace torch {
8 namespace lazy {
9 
10 std::optional<SourceLocation> TORCH_PYTHON_API GetPythonFrameTop();
11 
12 std::vector<SourceLocation> TORCH_PYTHON_API GetPythonFrames();
13 
14 } // namespace lazy
15 } // namespace torch
16