From d96f5c92e3ebe9cdc4c2c401d07fcfcf2943d16f Mon Sep 17 00:00:00 2001 From: "github-actions[bot]" <41898282+github-actions[bot]@users.noreply.github.com> Date: Mon, 14 Sep 2026 19:28:41 +0000 Subject: [PATCH] Add ucode_*.bin firmware files to CDI spec introduced in R615 Signed-off-by: Christopher Desiniotis (cherry picked from commit 1b7fd56d6d774a82ba05ab940ce400a4bcebb8e3) --- pkg/nvcdi/driver-nvml.go | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/pkg/nvcdi/driver-nvml.go b/pkg/nvcdi/driver-nvml.go index ee19d0c3c..1da149dc4 100644 --- a/pkg/nvcdi/driver-nvml.go +++ b/pkg/nvcdi/driver-nvml.go @@ -249,22 +249,23 @@ func getCustomFirmwareClassPath(logger logger.Interface) string { return strings.TrimSpace(string(customFirmwareClassPath)) } -// newDriverFirmwareDiscoverer creates a discoverer for GSP firmware associated with the specified driver version. +// newDriverFirmwareDiscoverer creates a discoverer for the GSP and ucode firmware associated with the specified driver version. func (l *nvcdilib) newDriverFirmwareDiscoverer(version string) (discover.Discover, error) { - gspFirmwareSearchPaths, err := getFirmwareSearchPaths(l.logger) + firmwareSearchPaths, err := getFirmwareSearchPaths(l.logger) if err != nil { return nil, fmt.Errorf("failed to get firmware search paths: %v", err) } gspFirmwarePaths := filepath.Join("nvidia", version, "gsp*.bin") + ucodeFirmwarePaths := filepath.Join("nvidia", version, "ucodes*.bin") return discover.NewMounts( l.logger, lookup.NewFileLocator( lookup.WithLogger(l.logger), lookup.WithRoot(l.driver.Root), - lookup.WithSearchPaths(gspFirmwareSearchPaths...), + lookup.WithSearchPaths(firmwareSearchPaths...), ), l.driver.Root, - []string{gspFirmwarePaths}, + []string{gspFirmwarePaths, ucodeFirmwarePaths}, ), nil }