Print invalid version on parse error

elezar · elezar · commit 6750ce16670a · 2023-07-11T13:47:39.000+02:00
Signed-off-by: Evan Lezar &lt;elezar@nvidia.com&gt;
diff --git a/internal/config/image/cuda_image.go b/internal/config/image/cuda_image.go
@@ -148,9 +148,10 @@ func (i CUDA) GetDriverCapabilities() DriverCapabilities {
 }
 
 func (i CUDA) legacyVersion() (string, error) {
-	majorMinor, err := parseMajorMinorVersion(i[envCUDAVersion])
+	cudaVersion := i[envCUDAVersion]
+	majorMinor, err := parseMajorMinorVersion(cudaVersion)
 	if err != nil {
-		return "", fmt.Errorf("invalid CUDA version: %v", err)
+		return "", fmt.Errorf("invalid CUDA version %v: %v", cudaVersion, err)
 	}
 
 	return majorMinor, nil

Original file line number	Diff line number	Diff line change
`@@ -148,9 +148,10 @@ func (i CUDA) GetDriverCapabilities() DriverCapabilities {`
`148`	`148`	`}`
`149`	`149`
`150`	`150`	`func (i CUDA) legacyVersion() (string, error) {`
`151`		`- majorMinor, err := parseMajorMinorVersion(i[envCUDAVersion])`
	`151`	`+ cudaVersion := i[envCUDAVersion]`
	`152`	`+ majorMinor, err := parseMajorMinorVersion(cudaVersion)`
`152`	`153`	`if err != nil {`
`153`		`- return "", fmt.Errorf("invalid CUDA version: %v", err)`
	`154`	`+ return "", fmt.Errorf("invalid CUDA version %v: %v", cudaVersion, err)`
`154`	`155`	`}`
`155`	`156`
`156`	`157`	`return majorMinor, nil`