1#pragma once
2
3#include <ATen/ATen.h>
4#include <ATen/core/ivalue.h>
5#include <ATen/core/jit_type.h>
6#include <ATen/core/stack.h>
7#include <c10/util/sparse_bitset.h>
8#include <torch/csrc/Export.h>
9#include <torch/csrc/jit/ir/ir.h>
10#include <list>
11#include <unordered_map>
12#include <vector>
13namespace torch {
14namespace jit {
15
16using SparseBitVector = ::c10::SparseBitVector<256>;
17
18// BuildLivenessSets computes "bailout" liveness which is equivalent to
19// "{LIVE_IN} or {GEN}" or "{LIVE_OUT} - {KILL}"
20TORCH_API std::unordered_map<Node*, std::vector<Value*>> BuildLivenessSets(
21 std::shared_ptr<Graph> graph);
22} // namespace jit
23} // namespace torch
24