Skip to content

Commit 9943dbb

Browse files
committed
Disable enable-cuda-compat hook for IMEX driver
In order to support older NVIDIA Container Toolkit installations we explicitly disable the enable-cuda-compat hooks for the IMEX driver. Signed-off-by: Evan Lezar <[email protected]>
1 parent 776d03e commit 9943dbb

File tree

2 files changed

+6
-0
lines changed
  • cmd

2 files changed

+6
-0
lines changed

cmd/compute-domain-kubelet-plugin/cdi.go

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -120,6 +120,8 @@ func NewCDIHandler(opts ...cdiOption) (*CDIHandler, error) {
120120
nvcdi.WithVendor(h.vendor),
121121
nvcdi.WithClass(h.claimClass),
122122
nvcdi.WithNVIDIACDIHookPath(h.nvidiaCTKPath),
123+
// TODO: Enable when https://github.com/NVIDIA/nvidia-container-toolkit/pull/968 is merged
124+
// nvcdi.WithDisabledHook(nvcdi.HookEnableCudaCompat),
123125
)
124126
if err != nil {
125127
return nil, fmt.Errorf("unable to create CDI library for claims: %w", err)

cmd/gpu-kubelet-plugin/cdi.go

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -103,6 +103,8 @@ func NewCDIHandler(opts ...cdiOption) (*CDIHandler, error) {
103103
nvcdi.WithVendor(h.vendor),
104104
nvcdi.WithClass(h.deviceClass),
105105
nvcdi.WithNVIDIACDIHookPath(h.nvidiaCTKPath),
106+
// TODO: Enable when https://github.com/NVIDIA/nvidia-container-toolkit/pull/968 is merged
107+
// nvcdi.WithDisabledHook(nvcdi.HookEnableCudaCompat),
106108
)
107109
if err != nil {
108110
return nil, fmt.Errorf("unable to create CDI library for devices: %w", err)
@@ -120,6 +122,8 @@ func NewCDIHandler(opts ...cdiOption) (*CDIHandler, error) {
120122
nvcdi.WithVendor(h.vendor),
121123
nvcdi.WithClass(h.claimClass),
122124
nvcdi.WithNVIDIACDIHookPath(h.nvidiaCTKPath),
125+
// TODO: Enable when https://github.com/NVIDIA/nvidia-container-toolkit/pull/968 is merged
126+
// nvcdi.WithDisabledHook(nvcdi.HookEnableCudaCompat),
123127
)
124128
if err != nil {
125129
return nil, fmt.Errorf("unable to create CDI library for claims: %w", err)

0 commit comments

Comments
 (0)