Bump github.com/NVIDIA/go-nvlib from 0.3.0 to 0.6.0

go.mod

@@ -5,7 +5,7 @@ go 1.21
 toolchain go1.22.5
 
 require (
-	github.com/NVIDIA/go-nvlib v0.3.0
+	github.com/NVIDIA/go-nvlib v0.6.0
 	github.com/NVIDIA/go-nvml v0.12.4-0
 	github.com/sirupsen/logrus v1.9.3
 	github.com/stretchr/testify v1.9.0

go.sum

@@ -1,5 +1,5 @@
-github.com/NVIDIA/go-nvlib v0.3.0 h1:vd7jSOthJTqzqIWZrv317xDr1+Mnjoy5X4N69W9YwQM=
-github.com/NVIDIA/go-nvlib v0.3.0/go.mod h1:NasUuId9hYFvwzuOHCu9F2X6oTU2tG0JHTfbJYuDAbA=
+github.com/NVIDIA/go-nvlib v0.6.0 h1:zAMBzCYT9xeyRQo0tb7HJbStkzajD6e5joyaQqJ2OGU=
+github.com/NVIDIA/go-nvlib v0.6.0/go.mod h1:9UrsLGx/q1OrENygXjOuM5Ey5KCtiZhbvBlbUIxtGWY=
 github.com/NVIDIA/go-nvml v0.12.4-0 h1:4tkbB3pT1O77JGr0gQ6uD8FrsUPqP1A/EOEm2wI1TUg=
 github.com/NVIDIA/go-nvml v0.12.4-0/go.mod h1:8Llmj+1Rr+9VGGwZuRer5N/aCjxGuR5nPb/9ebBiIEQ=
 github.com/cpuguy83/go-md2man/v2 v2.0.4 h1:wfIWP927BUkWJb2NmU/kNDYIBTh/ziUX91+lVfRxZq4=

pkg/mig/mode/pci_test.go

@@ -36,7 +36,7 @@ func NewMockPciA100Device() (*mockPciMigModeManager, error) {
 		return nil, fmt.Errorf("error creating Mock A100 PCI device: %v", err)
 	}
 
-	err = nvpci.AddMockA100("0000:80:05.1", 0)
+	err = nvpci.AddMockA100("0000:80:05.1", 0, nil)
 	if err != nil {
 		return nil, fmt.Errorf("error adding Mock A100 device to MockNvpci: %v", err)
 	}

pkg/types/nvdev.go

@@ -41,7 +41,7 @@ func nvdevNewMigProfile(giProfileID, ciProfileID, ciEngProfileID int, migMemoryS
 		nvmllib = nvml.New()
 	}
 	if nvdevlib == nil {
-		nvdevlib = nvdev.New(nvdev.WithNvml(nvmllib))
+		nvdevlib = nvdev.New(nvmllib)
 	}
 
 	mp, err := nvdevlib.NewMigProfile(giProfileID, ciProfileID, ciEngProfileID, migMemorySizeMB, deviceMemorySizeBytes)
@@ -57,7 +57,7 @@ func nvdevAssertValidMigProfileFormat(profile string) error {
 		nvmllib = nvml.New()
 	}
 	if nvdevlib == nil {
-		nvdevlib = nvdev.New(nvdev.WithNvml(nvmllib))
+		nvdevlib = nvdev.New(nvmllib)
 	}
 
 	return nvdevlib.AssertValidMigProfileFormat(profile)
@@ -68,7 +68,7 @@ func nvdevParseMigProfile(profile string) (nvdev.MigProfile, error) {
 		nvmllib = nvml.New()
 	}
 	if nvdevlib == nil {
-		nvdevlib = nvdev.New(nvdev.WithNvml(nvmllib))
+		nvdevlib = nvdev.New(nvmllib)
 	}
 
 	ret := nvmllib.Init()
@@ -141,8 +141,7 @@ func SetMockNVdevlib() {
 		},
 	}
 
-	nvdevlib = nvdev.New(
-		nvdev.WithNvml(nvmllib),
+	nvdevlib = nvdev.New(nvmllib,
 		nvdev.WithVerifySymbols(false),
 	)
 }