GPU information cannot be queried after virtual machine GPU pass through. By traversing the PCI information, it is found that the driver is not NVIDIA but vfio PCI. Can I add a vendor Name judge whether the PCI identification is GPU

[huyuan1999]

package main

import (
	"fmt"
	"github.com/jaypipes/ghw"
	"github.com/sirupsen/logrus"
	"os"
	"strings"
)

func Exists(path string) bool {
	_, err := os.Stat(path)
	if err != nil {
		if os.IsExist(err) {
			return true
		}
		return false
	}
	return true
}

func IsDir(path string) bool {
	s, err := os.Stat(path)
	if err != nil {
		return false
	}
	return s.IsDir()
}

func IsFile(path string) bool {
	if Exists(path) {
		return !IsDir(path)
	}
	return false
}

func main() {
	pcidbPath := "/tmp/pci.ids"
	if IsFile(pcidbPath) {
		os.Setenv("PCIDB_PATH", "/tmp/pci.ids")
	}

	pci, err := ghw.PCI()
	if err != nil {
		logrus.Errorln(err)
		return
	}

	for _, devices := range pci.ListDevices() {
		if strings.Index(strings.ToLower(devices.Driver), "nvidia") != -1 || strings.Index(strings.ToLower(devices.Vendor.Name), "nvidia") != -1 {
			fmt.Printf("==================== nvidia %s ====================\n", devices.Address)
			fmt.Println("Address", devices.Address)
			fmt.Println("Vendor.ID", devices.Vendor.ID)
			fmt.Println("Vendor.Name", devices.Vendor.Name)
			fmt.Println("Driver", devices.Driver)
			fmt.Println("devices.Product.ID", devices.Product.ID)
			fmt.Println("devices.Product.VendorID", devices.Product.VendorID)
			fmt.Println("devices.Product.Name", devices.Product.Name)
		}
	}
}

[ffromani]

Hi @huyuan1999 ! Thanks for the report. I understand parts of the problem, because indeed inside a VM a VFIO device can be misreported, but I'm not completely sure what's the question here, could you please elaborate?
Are you requesting for example a way to override the pci.db?

[red-0ne]

I think what @huyuan1999 is looking for is a way to determine if a PCI device with a nvidia device.Vendor.Name is a GPU or not.
Since device.Driver does not return nvidia but vfio-pci, we cannot rely on that property.

We are facing the same problem here. A possible workaround would be to check if device.Class.Name is a Display controller. @fromanirh is it reasonable to do it that way?

[ydcool]

Same issue here, any updates?

[ffromani]

Thanks for elaborating. Rather than make logic on the name, I'd indeed make logic on device class. I'd use ids rather than names.

[ffromani]

Is the gpu subpackage returning the correct information?