Skip to content

Commit 4624042

Browse files
committed
Add disable-imex-channel-creation feature flag
Signed-off-by: Evan Lezar <[email protected]>
1 parent c80e72e commit 4624042

File tree

2 files changed

+6
-0
lines changed

2 files changed

+6
-0
lines changed

cmd/nvidia-container-runtime-hook/main.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -95,6 +95,9 @@ func doPrestart() {
9595
if cli.LoadKmods {
9696
args = append(args, "--load-kmods")
9797
}
98+
if hook.Features.DisableImexChannelCreation.IsEnabled() {
99+
args = append(args, "--no-create-imex-channels")
100+
}
98101
if cli.NoPivot {
99102
args = append(args, "--no-pivot")
100103
}

internal/config/features.go

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,9 @@ type features struct {
2222
MOFED *featureMofed `toml:"mofed,omitempty"`
2323
NVSWITCH *featureNvswitch `toml:"nvswitch,omitempty"`
2424
GDRCopy *featureGdrCopy `toml:"gdrcopy,omitempty"`
25+
// DisableImexChannelCreation ensures that the implicit creation of
26+
// requested IMEX channels is skipped when invoking the nvidia-container-cli.
27+
DisableImexChannelCreation *feature `toml:"disable-imex-channel-creation,omitempty"`
2528
}
2629

2730
type feature bool

0 commit comments

Comments
 (0)