mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
f782500ee0
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/28473 Test Plan: Imported from OSS Differential Revision: D18121007 Pulled By: jamesr66a fbshipit-source-id: 4c4a4344ad9bcc4630b945d2a645a0b05928933c
38 lines
896 B
C++
38 lines
896 B
C++
#pragma once
|
|
|
|
#include <torch/csrc/jit/tracer.h>
|
|
#include <torch/csrc/python_headers.h>
|
|
#include <torch/csrc/utils/pybind.h>
|
|
#include <torch/csrc/jit/source_range.h>
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
namespace torch {
|
|
namespace jit {
|
|
|
|
namespace script {
|
|
struct Module;
|
|
}
|
|
|
|
namespace tracer {
|
|
void initPythonTracerBindings(PyObject* module);
|
|
|
|
std::string getPythonInterpreterStackTrace();
|
|
SourceRange getPythonInterpreterSourceRange();
|
|
|
|
Node* preRecordPythonTrace(
|
|
THPObjectPtr pyobj,
|
|
const std::string& arg_types,
|
|
at::ArrayRef<autograd::Variable> inputs,
|
|
std::vector<THPObjectPtr> scalar_args);
|
|
|
|
std::pair<std::shared_ptr<Graph>, Stack> createGraphByTracing(
|
|
const py::function& func,
|
|
Stack inputs,
|
|
const py::function& var_name_lookup_fn,
|
|
bool force_outplace,
|
|
script::Module* self = nullptr);
|
|
} // namespace tracer
|
|
} // namespace jit
|
|
} // namespace torch
|