From d7fecb3a1c6b088473e4530d7aa0ab0f2a096361 Mon Sep 17 00:00:00 2001 From: Evan Lezar Date: Tue, 18 Nov 2025 13:52:24 +0100 Subject: [PATCH] Use requested devices for CSV CDI spec generation Signed-off-by: Evan Lezar --- internal/modifier/csv.go | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/internal/modifier/csv.go b/internal/modifier/csv.go index 1f8a12f83..c8cf4ead3 100644 --- a/internal/modifier/csv.go +++ b/internal/modifier/csv.go @@ -33,7 +33,8 @@ import ( // NewCSVModifier creates a modifier that applies modications to an OCI spec if required by the runtime wrapper. // The modifications are defined by CSV MountSpecs. func NewCSVModifier(logger logger.Interface, cfg *config.Config, container image.CUDA) (oci.SpecModifier, error) { - if devices := container.VisibleDevices(); len(devices) == 0 { + devices := container.VisibleDevices() + if len(devices) == 0 { logger.Infof("No modification required; no devices requested") return nil, nil } @@ -63,7 +64,7 @@ func NewCSVModifier(logger logger.Interface, cfg *config.Config, container image return nil, fmt.Errorf("failed to construct CDI library: %v", err) } - spec, err := cdilib.GetSpec() + spec, err := cdilib.GetSpec(devices...) if err != nil { return nil, fmt.Errorf("failed to get CDI spec: %v", err) }