Disable enable-cuda-compat hook for management containers

elezar · elezar · commit 35e0dea1d35f · 2025-03-07T16:56:20.000+02:00
Management containers don't generally need forward compatibility.
We disable the enable-cuda-compat hook to not include this in the
generated CDI specifications.

Signed-off-by: Evan Lezar &lt;elezar@nvidia.com&gt;
diff --git a/pkg/nvcdi/lib.go b/pkg/nvcdi/lib.go
@@ -156,6 +156,8 @@ func New(opts ...Option) (Interface, error) {
 		if l.vendor == "" {
 			l.vendor = "management.nvidia.com"
 		}
+		// Management containers in general do not require CUDA Forward compatibility.
+		l.disabledHooks[HookEnableCudaCompat] = true
 		lib = (*managementlib)(l)
 	case ModeNvml:
 		lib = (*nvmllib)(l)

Original file line number	Diff line number	Diff line change
`@@ -156,6 +156,8 @@ func New(opts ...Option) (Interface, error) {`
`156`	`156`	`if l.vendor == "" {`
`157`	`157`	`l.vendor = "management.nvidia.com"`
`158`	`158`	`}`
	`159`	`+ // Management containers in general do not require CUDA Forward compatibility.`
	`160`	`+ l.disabledHooks[HookEnableCudaCompat] = true`
`159`	`161`	`lib = (*managementlib)(l)`
`160`	`162`	`case ModeNvml:`
`161`	`163`	`lib = (*nvmllib)(l)`