IdProvider.h

Go to the documentation of this file.

 
#pragma once
 
#include <tvm/api.h>
 
#include <mutex>
 
namespace tvm
{
 
namespace internal
{
class TVM_DLLAPI IdProvider
{
public:
  int makeId();
 
private:
  std::mutex mutex_;
  int id_ = 0;
};
 
inline int IdProvider::makeId()
{
  std::lock_guard<std::mutex> lock(mutex_);
  return id_++;
}
} // namespace internal
 
} // namespace tvm