update vfio validation

Author: JunAr7112

.github/workflows/ci.yaml

@@ -19,6 +19,7 @@ on:
     branches:
       - "pull-request/[0-9]+"
       - main
+      - vfio_changes
       - release-*
 
 jobs:

cmd/nvidia-validator/main.go

@@ -1733,7 +1733,44 @@ func (v *VGPUDevices) validate() error {
 	return nil
 }
 
+func (v *VGPUDevices) IsUbuntu2404() (bool, error) {
+    // Read from the host's /etc/os-release (mounted at /host in the container)
+    data, err := os.ReadFile("/host/etc/os-release")
+    if err != nil {
+        return false, fmt.Errorf("unable to read host OS release info: %v", err)
+    }
+    
+    content := string(data)
+    isUbuntu := strings.Contains(content, "ID=ubuntu")
+    is2404 := strings.Contains(content, `VERSION_ID="24.04"`)
+    
+    return isUbuntu && is2404, nil
+}
+
 func (v *VGPUDevices) runValidation() error {
+	isUbuntu2404, err := v.IsUbuntu2404()
+	if err != nil {
+		return fmt.Errorf("error checking if host is Ubuntu 24.04: %w", err)
+	}
+
+	if isUbuntu2404 {
+		nvpci := nvpci.New()
+		GPUDevices, err := nvpci.GetGPUs()
+		if err != nil {
+			return fmt.Errorf("error checking for devices on the host: %w", err)
+		}
+		for _, device := range GPUDevices {
+			totalVF := int(device.SriovInfo.PhysicalFunction.TotalVFs)
+			if totalVF > 0 {
+				log.Infof("Found GPU device with VF: %s", device.Address)
+				return nil
+			}
+		}
+		log.Info("No vGPU devices found")
+		return fmt.Errorf("no vGPU devices found")
+	}
+
+	// if not Ubuntu 24.04, check for vGPU devices using nvmdev
 	nvmdev := nvmdev.New()
 	vGPUDevices, err := nvmdev.GetAllDevices()
 	if err != nil {