You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
56 lines
1.4 KiB
56 lines
1.4 KiB
#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)
|