mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-27 09:04:53 +08:00
ad8aef0f98
Mostly in torch/csrc/jit/runtime and in `ATen/cuda/` Pull Request resolved: https://github.com/pytorch/pytorch/pull/110314 Approved by: https://github.com/seemethere
20 lines
401 B
C++
20 lines
401 B
C++
#pragma once
|
|
|
|
#include <memory>
|
|
#include <unordered_map>
|
|
|
|
#include <torch/csrc/jit/ir/ir.h>
|
|
|
|
namespace torch::jit::interpreter {
|
|
|
|
// pre-processing that happens once per graph
|
|
struct PreprocessGraph {
|
|
explicit PreprocessGraph(Graph& g);
|
|
|
|
// Outputs of the preprocessing:
|
|
std::shared_ptr<Graph> graph;
|
|
std::unordered_map<Node*, bool> can_emit_inline;
|
|
};
|
|
|
|
} // namespace torch::jit::interpreter
|