image-match/python/py/Lib/site-packages/torch/include/c10/util/Gauge.h

#if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
#pragma once

#include <memory>
#include <string_view>

#include <c10/macros/Macros.h>
#include <c10/util/SmallVector.h>

namespace c10::monitor {
namespace detail {

class GaugeImpl;

class GaugeBackendIf {
 public:
  virtual ~GaugeBackendIf() = default;
  virtual void record(int64_t value) noexcept = 0;
};

class GaugeBackendFactoryIf {
 public:
  virtual ~GaugeBackendFactoryIf() = default;

  // May return nullptr if the gauge will be ignored by the given backend.
  virtual std::unique_ptr<GaugeBackendIf> create(
      std::string_view key) noexcept = 0;
};

void C10_API
    registerGaugeBackend(std::unique_ptr<GaugeBackendFactoryIf> /*backend*/);
} // namespace detail

// A handle to a Gauge.
class C10_API GaugeHandle {
 public:
  explicit GaugeHandle(std::string_view key);
  void record(int64_t value);

 private:
  // NOLINTNEXTLINE(cppcoreguidelines-avoid-const-or-ref-data-members)
  detail::GaugeImpl& impl_;
};

} // namespace c10::monitor

#define STATIC_GAUGE(_key)                            \
  []() -> ::c10::monitor::GaugeHandle& {              \
    static ::c10::monitor::GaugeHandle handle(#_key); \
    return handle;                                    \
  }()

#else
#error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
#endif  // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)