Fix nil-pointer mistake

kubernetes-sigs · May 30, 2024 · bc85b61 · bc85b61
1 parent 122d879
commit bc85b61
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pkg/services/govmomi/pci/device.go b/pkg/services/govmomi/pci/device.go
@@ -77,7 +77,7 @@ func ConstructDeviceSpecs(pciDeviceSpecs []infrav1.PCIDeviceSpec) []types.BaseVi
 }
 
 func createBackingInfo(spec infrav1.PCIDeviceSpec) types.BaseVirtualDeviceBackingInfo {
-	if spec.VGPUProfile == "" {
+	if spec.DeviceID != nil && spec.VendorID != nil {
 		return &types.VirtualPCIPassthroughDynamicBackingInfo{
 			AllowedDevice: []types.VirtualPCIPassthroughAllowedDevice{
 				{
@@ -95,7 +95,7 @@ func createBackingInfo(spec infrav1.PCIDeviceSpec) types.BaseVirtualDeviceBackin
 }
 
 func constructKey(pciDeviceSpec infrav1.PCIDeviceSpec) string {
-	if pciDeviceSpec.VGPUProfile == "" {
+	if pciDeviceSpec.DeviceID != nil && pciDeviceSpec.VendorID != nil {
 		return fmt.Sprintf("%d-%d", *pciDeviceSpec.DeviceID, *pciDeviceSpec.VendorID)
 	}