Add --disable-hook flag to cdi generate command #1077

ArangoGutierrez · 2025-05-12T16:53:55Z

When running the nvidia-ctk cdi generate command, a user should be able to opt out of specific hooks. We propose to add a flag --disable-hooks that will take a comma-separated list of hooks that will be skipped when creating the CDI spec.

Fixes: #1074

Copilot

Pull Request Overview

This PR adds a new CLI flag (--skip-hooks) to allow users to opt out of specific hooks when generating the CDI specification.

Updated tests to include a skip-hook option.
Modified the command initialization and spec generation logic to support disabling hooks based on the provided flag.

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated no comments.

File	Description
cmd/nvidia-ctk/cdi/generate/generate_test.go	Added a test case to validate the skip-hook functionality and updated expected spec/options accordingly.
cmd/nvidia-ctk/cdi/generate/generate.go	Introduced a new CLI flag and added logic to disable hooks based on user input.

Comments suppressed due to low confidence (1)

cmd/nvidia-ctk/cdi/generate/generate_test.go:40

[nitpick] The variable name 'skipdHook' may be a typo and could be renamed to 'skipHook' for consistency and clarity.

skipdHook := cli.NewStringSlice("enable-cuda-compat")

Copilot

Pull Request Overview

This PR adds support for a new flag (--skip-hooks) to the nvidia-ctk cdi generate command so that users can opt out of specific hooks when creating the CDI spec.

In generate.go, a new skipHook option and corresponding CLI flag are introduced and processed when creating CDI library options.
In generate_test.go, a new test case verifies the behavior when a non-default hook is skipped.

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 1 comment.

File	Description
cmd/nvidia-ctk/cdi/generate/generate_test.go	Added test case for --skip-hooks flag and its effect.
cmd/nvidia-ctk/cdi/generate/generate.go	Introduced skipHook flag and integrated it into initialization options.

cmd/nvidia-ctk/cdi/generate/generate.go

Copilot

Pull Request Overview

This PR adds a new flag --skip-hooks to the cdi generate command to allow users to disable specific hooks while generating the CDI specification.

Introduces a new CLI flag in generate.go to capture the hooks to skip
Updates the options struct and test cases in generate_test.go to include and validate the new skipHook functionality

Reviewed Changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 1 comment.

File	Description
cmd/nvidia-ctk/cdi/generate/generate_test.go	Adds a test case for skipHook using a new CLI string slice
cmd/nvidia-ctk/cdi/generate/generate.go	Registers the new --skip-hook flag and passes the provided hooks to disable options during CDI spec generation

Comments suppressed due to low confidence (1)

cmd/nvidia-ctk/cdi/generate/generate_test.go:40

[nitpick] Consider adding a test case to verify that when a hook name is provided via skipHook, the generated CDI specification omits or disables that hook, ensuring the new flag's functionality is fully verified.

skipHook := cli.NewStringSlice("enable-cuda-compat")

cmd/nvidia-ctk/cdi/generate/generate.go

elezar · 2025-05-13T13:57:11Z

cmd/nvidia-ctk/cdi/generate/generate.go

@@ -176,6 +177,12 @@ func (m command) build() *cli.Command {
 			Usage:       "Specify a pattern the CSV mount specifications.",
 			Destination: &opts.csv.ignorePatterns,
 		},
+		&cli.StringSliceFlag{
+			Name:        "skip-hook",
+			Usage:       "Comma-separated list of hooks to skip when generating the CDI specification.",


Technically the flag can also be specified multiple times.

elezar · 2025-05-13T13:57:41Z

cmd/nvidia-ctk/cdi/generate/generate.go

@@ -57,6 +57,7 @@ type options struct {

 	configSearchPaths  cli.StringSlice
 	librarySearchPaths cli.StringSlice
+	skipHook           cli.StringSlice


This should be:

Suggested change

skipHook cli.StringSlice

skipHooks cli.StringSlice

Since it contains 0 or more entries.

elezar · 2025-05-13T13:58:21Z

cmd/nvidia-ctk/cdi/generate/generate.go

+	if len(opts.skipHook.Value()) > 0 {
+		for _, hook := range opts.skipHook.Value() {
+			initOpts = append(initOpts, nvcdi.WithDisabledHook(nvcdi.HookName(hook)))
+		}
+	}


Did we want to check for validity of the names, or do we want to add that as a follow-up?

let's hold this PR until:

Ensure that libcuda.so is in the ldcache #947

Disable device node creation in CDI mode #927
get in, then I will think on how to add a name check.

let's hold this PR until:

Should this PR not just work automatically with those PRs -- assuming those hooks can be opted out of using the same option?

It now allows for disabling all hooks

elezar · 2025-05-13T20:37:59Z

cmd/nvidia-ctk/cdi/generate/generate.go

@@ -176,6 +177,12 @@ func (m command) build() *cli.Command {
 			Usage:       "Specify a pattern the CSV mount specifications.",
 			Destination: &opts.csv.ignorePatterns,
 		},
+		&cli.StringSliceFlag{
+			Name:        "skip-hook",


What about disable-hook (we could even add an alias as disable-hooks) instead? Thinking ahead, it may be useful to allow users to do something like:

nvidia-ctk cdi generate \ --disable-hook=update-ldcache \ --disable-hook=enable-cuda-compat \ --mode=nvml

Even further forward, more conservative users may want to consider:

nvidia-ctk cdi generate --disable-hooks=all

although that is out of scope of this PR.

@klueska @jgehrcke what are your thoughts on the UX?

Copilot

Pull Request Overview

This PR adds support for a new flag that allows users to disable specific hooks during the CDI spec generation. Key changes include:

Adding conditional logic to check for supported hooks in the NVML driver library.
Introducing the new CLI flag (and corresponding option) to allow disabling hooks.
Updating tests to reflect the new --disable-hook(s) functionality.

Reviewed Changes

Copilot reviewed 3 out of 3 changed files in this pull request and generated no comments.

File	Description
pkg/nvcdi/driver-nvml.go	Wraps hook-based discoverer creation in conditionals based on hook support.
cmd/nvidia-ctk/cdi/generate/generate_test.go	Adds test cases and sample disable-hook values in accordance with the new functionality.
cmd/nvidia-ctk/cdi/generate/generate.go	Introduces a new flag and integrates hook disabling into the initialization options.

Comments suppressed due to low confidence (1)

cmd/nvidia-ctk/cdi/generate/generate.go:181

[nitpick] Consider renaming the flag from "disable-hook" to "disable-hooks" to match the PR title and description for consistency.

Name:        "disable-hook",

When running the nvidia-ctk cdi generate command, a user should be able to opt out of specific hooks. We propose to add a flag --disable-hook that will take a comma-separated list of hooks that will be skipped when creating the CDI spec. Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

ArangoGutierrez requested review from elezar and Copilot May 12, 2025 16:53

ArangoGutierrez self-assigned this May 12, 2025

Copilot AI reviewed May 12, 2025

View reviewed changes

ArangoGutierrez force-pushed the 1074 branch from d9c00f7 to ef09bbf Compare May 12, 2025 17:59

ArangoGutierrez requested a review from Copilot May 12, 2025 18:00

Copilot AI reviewed May 12, 2025

View reviewed changes

cmd/nvidia-ctk/cdi/generate/generate.go Outdated Show resolved Hide resolved

ArangoGutierrez force-pushed the 1074 branch from ef09bbf to 2b37211 Compare May 12, 2025 18:15

ArangoGutierrez requested a review from Copilot May 12, 2025 18:15

Copilot AI reviewed May 12, 2025

View reviewed changes

cmd/nvidia-ctk/cdi/generate/generate.go Show resolved Hide resolved

elezar reviewed May 13, 2025

View reviewed changes

ArangoGutierrez force-pushed the 1074 branch from 2b37211 to 194c1e6 Compare May 14, 2025 09:18

ArangoGutierrez changed the title ~~Add --skip-hooks flag to cdi generate command~~ Add --disable-hook flag to cdi generate command May 14, 2025

ArangoGutierrez requested review from elezar and Copilot May 14, 2025 10:09

Copilot AI reviewed May 14, 2025

View reviewed changes

ArangoGutierrez force-pushed the 1074 branch from b6e5911 to 64375df Compare May 14, 2025 10:36

ArangoGutierrez added 3 commits May 14, 2025 12:36

rename flag to disable-hooks

d657e0a

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

Add all option to disable all hooks

b32d8a9

Signed-off-by: Carlos Eduardo Arango Gutierrez <[email protected]>

ArangoGutierrez force-pushed the 1074 branch from 64375df to b32d8a9 Compare May 14, 2025 10:37

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Add --disable-hook flag to cdi generate command #1077

Add --disable-hook flag to cdi generate command #1077

ArangoGutierrez commented May 12, 2025 •

edited

Loading

Copilot AI left a comment

Copilot AI left a comment

Copilot AI left a comment

elezar May 13, 2025

elezar May 13, 2025

elezar May 13, 2025

ArangoGutierrez May 13, 2025

elezar May 13, 2025

ArangoGutierrez May 14, 2025

elezar May 13, 2025

elezar May 14, 2025

Copilot AI left a comment

Add --disable-hook flag to cdi generate command #1077

Are you sure you want to change the base?

Add --disable-hook flag to cdi generate command #1077

Conversation

ArangoGutierrez commented May 12, 2025 • edited Loading

Copilot AI left a comment

Choose a reason for hiding this comment

Pull Request Overview

Reviewed Changes

Copilot AI left a comment

Choose a reason for hiding this comment

Pull Request Overview

Reviewed Changes

Copilot AI left a comment

Choose a reason for hiding this comment

Pull Request Overview

Reviewed Changes

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Choose a reason for hiding this comment

Copilot AI left a comment

Choose a reason for hiding this comment

Pull Request Overview

Reviewed Changes

ArangoGutierrez commented May 12, 2025 •

edited

Loading