1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
|
#pragma once
#include <c10/util/Flags.h>
#include <torch/csrc/jit/api/module.h>
#include <torch/csrc/jit/runtime/graph_executor_impl.h>
namespace torch {
namespace jit {
struct TORCH_API SimpleGraphExecutorImpl : public GraphExecutorImplBase {
SimpleGraphExecutorImpl(
const std::shared_ptr<Graph>& graph,
std::string function_name);
const ExecutionPlan& getPlanFor(
Stack& stack,
c10::optional<size_t> remaining_bailout_depth) override;
GraphExecutorState getDebugState() override;
~SimpleGraphExecutorImpl() override = default;
private:
c10::optional<ExecutionPlan> execution_plan_;
};
} // namespace jit
} // namespace torch
|
