Merge pull request #619 from NVIDIA/fix-mig-get-minor-number

[bugfix] Process error type instead of nvml.Return
NVIDIA · Jul 22, 2024 · 87a63a6 · 87a63a6
2 parents c5c124b + d8ccd50
commit 87a63a6
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/internal/platform-support/dgpu/nvml.go b/internal/platform-support/dgpu/nvml.go
@@ -145,9 +145,9 @@ type toRequiredMigInfo struct {
 }
 
 func (d *toRequiredMigInfo) getPlacementInfo() (int, int, int, error) {
-	gpu, ret := d.parent.GetMinorNumber()
-	if ret != nvml.SUCCESS {
-		return 0, 0, 0, fmt.Errorf("error getting GPU minor: %v", ret)
+	gpu, err := d.parent.GetMinorNumber()
+	if err != nil {
+		return 0, 0, 0, fmt.Errorf("error getting GPU minor: %w", err)
 	}
 
 	gi, ret := d.GetGpuInstanceId()