1#pragma once
2
3#include <c10/macros/Export.h>
4#include <cstdint>
5
6namespace at {
7
8// A simple thread local enumeration, used to link forward and backward pass
9// ops and is used by autograd and observers framework
10namespace sequence_number {
11
12TORCH_API uint64_t peek();
13TORCH_API uint64_t get_and_increment();
14
15} // namespace sequence_number
16} // namespace at
17