metrics

var (
	GpuTflopsRequest = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "gpu_tflops_request",
		},
		labels,
	)

	GpuTflopsLimit = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "gpu_tflops_limit",
		},
		labels,
	)

	VramBytesRequest = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "vram_bytes_request",
		},
		labels,
	)

	VramBytesLimit = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "vram_bytes_limit",
		},
		labels,
	)

	GpuCount = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "gpu_count",
			Help: "Number of GPUs allocated to the workload",
		},
		labels,
	)

	AllocatedTflopsPercent = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "allocated_compute_percentage",
		},
		nodeLabels,
	)

	AllocatedVramBytes = prometheus.NewGaugeVec(
		prometheus.GaugeOpts{
			Name: "allocated_memory_bytes",
		},
		nodeLabels,
	)
)