Function GetGPUMemoryUsage

pkg/xsysinfo/gpu.go:290–343

GetGPUMemoryUsage returns real-time GPU memory usage for all detected GPUs. It tries multiple vendor-specific tools in order: NVIDIA, AMD, Intel, Vulkan. Returns an empty slice if no GPU monitoring tools are available.

()

// It tries multiple vendor-specific tools in order: NVIDIA, AMD, Intel, Vulkan.
// Returns an empty slice if no GPU monitoring tools are available.
func GetGPUMemoryUsage() []GPUMemoryInfo {
	var gpus []GPUMemoryInfo

	// Try NVIDIA first
	nvidiaGPUs := getNVIDIAGPUMemory()
	if len(nvidiaGPUs) > 0 {
		gpus = append(gpus, nvidiaGPUs...)
	}

	// XXX: Note - I could not test this with AMD and Intel GPUs, so I'm not sure if it works and it was added with the help of AI.

	// Try AMD ROCm
	amdGPUs := getAMDGPUMemory()
	if len(amdGPUs) > 0 {
		// Adjust indices to continue from NVIDIA GPUs
		startIdx := len(gpus)
		for i := range amdGPUs {
			amdGPUs[i].Index = startIdx + i
		}
		gpus = append(gpus, amdGPUs...)
	}

	// Try Intel
	intelGPUs := getIntelGPUMemory()
	if len(intelGPUs) > 0 {
		startIdx := len(gpus)
		for i := range intelGPUs {
			intelGPUs[i].Index = startIdx + i
		}
		gpus = append(gpus, intelGPUs...)
	}

	// Try NVIDIA integrated GPUs (Tegra Jetson, DGX Spark, Thor — unified memory).
	// These either lack nvidia-smi or have it behave unreliably, so we detect
	// them via SoC sysfs and report system RAM figures.
	if len(gpus) == 0 {
		integratedGPUs := getNVIDIAIntegratedGPUMemory()
		gpus = append(gpus, integratedGPUs...)
	}

	// Try Vulkan as fallback for device detection (limited real-time data)
	if len(gpus) == 0 {
		vulkanGPUs := getVulkanGPUMemory()
		gpus = append(gpus, vulkanGPUs...)
	}

	// Try Apple Silicon (macOS only)
	if len(gpus) == 0 {
		appleGPUs := getAppleGPUMemory()
		gpus = append(gpus, appleGPUs...)
	}

	return gpus
}

// GetGPUAggregateInfo returns aggregate GPU information across all GPUs
func GetGPUAggregateInfo() GPUAggregateInfo {