xref: /aosp_15_r20/external/pytorch/aten/src/ATen/SequenceNumber.cpp (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1*da0073e9SAndroid Build Coastguard Worker #include <ATen/SequenceNumber.h>
2*da0073e9SAndroid Build Coastguard Worker 
3*da0073e9SAndroid Build Coastguard Worker namespace at::sequence_number {
4*da0073e9SAndroid Build Coastguard Worker 
5*da0073e9SAndroid Build Coastguard Worker namespace {
6*da0073e9SAndroid Build Coastguard Worker thread_local uint64_t sequence_nr_ = 0;
7*da0073e9SAndroid Build Coastguard Worker } // namespace
8*da0073e9SAndroid Build Coastguard Worker 
peek()9*da0073e9SAndroid Build Coastguard Worker uint64_t peek() {
10*da0073e9SAndroid Build Coastguard Worker   return sequence_nr_;
11*da0073e9SAndroid Build Coastguard Worker }
12*da0073e9SAndroid Build Coastguard Worker 
get_and_increment()13*da0073e9SAndroid Build Coastguard Worker uint64_t get_and_increment() {
14*da0073e9SAndroid Build Coastguard Worker   return sequence_nr_++;
15*da0073e9SAndroid Build Coastguard Worker }
16*da0073e9SAndroid Build Coastguard Worker 
17*da0073e9SAndroid Build Coastguard Worker } // namespace at::sequence_number
18