From cc8ae13aa076e5bf245ca5121bfbab19f29341b1 Mon Sep 17 00:00:00 2001 From: Christopher Desiniotis Date: Wed, 27 May 2026 18:48:42 -0700 Subject: [PATCH] Add ability to disable CDI hooks in jit-cdi mode Signed-off-by: Christopher Desiniotis --- api/config/v1/runtime.go | 2 ++ internal/modifier/cdi.go | 1 + 2 files changed, 3 insertions(+) diff --git a/api/config/v1/runtime.go b/api/config/v1/runtime.go index 946f75a23..6f58f3764 100644 --- a/api/config/v1/runtime.go +++ b/api/config/v1/runtime.go @@ -49,6 +49,8 @@ type cdiModeConfig struct { type jitCDIModeConfig struct { // NVCDIFeatureFlags sets a list of nvcdi features explicitly. NVCDIFeatureFlags []nvcdi.FeatureFlag `toml:"nvcdi-feature-flags,omitempty"` + // NVCDIDisableHooks sets a list of nvcdi hooks to disable + NVCDIDisableHooks []nvcdi.HookName `toml:"nvcdi-disable-hooks,omitempty"` } type csvModeConfig struct { diff --git a/internal/modifier/cdi.go b/internal/modifier/cdi.go index 005c42f48..7b36ab590 100644 --- a/internal/modifier/cdi.go +++ b/internal/modifier/cdi.go @@ -199,6 +199,7 @@ func (f *Factory) newAutomaticCDISpecModifier(devices []string) (oci.SpecModifie nvcdi.WithFeatureFlags(f.cfg.NVIDIAContainerRuntimeConfig.Modes.JitCDI.NVCDIFeatureFlags...), nvcdi.WithCSVCompatContainerRoot(f.cfg.NVIDIAContainerRuntimeConfig.Modes.CSV.CompatContainerRoot), nvcdi.WithCSVFiles(csvFiles), + nvcdi.WithDisabledHooks(f.cfg.NVIDIAContainerRuntimeConfig.Modes.JitCDI.NVCDIDisableHooks...), ) if err != nil { return nil, fmt.Errorf("failed to construct CDI library for mode %q: %w", mode, err)