core/psutil/gpu/gpu.go

package gpu

import "errors"

type Process struct {
	PID     int32
	Index   int
	Memory  uint64  // bytes
	Usage   float64 // percent 0-100
	Encoder float64 // percent 0-100
	Decoder float64 // percent 0-100
}

type Stats struct {
	ID           string
	Name         string
	Architecture string

	MemoryTotal uint64 // bytes
	MemoryUsed  uint64 // bytes

	Usage   float64 // percent 0-100
	Encoder float64 // percent 0-100
	Decoder float64 // percent 0-100

	Process []Process

	Extension interface{}
}

type GPU interface {
	// Count returns the number of GPU in the system.
	Count() (int, error)

	// Stats returns current GPU stats.
	Stats() ([]Stats, error)

	// Process returns a Process.
	Process(pid int32) (Process, error)

	// Close stops all GPU collection processes
	Close()
}

var ErrProcessNotFound = errors.New("process not found")