mirror of
				https://github.com/NVIDIA/nvidia-container-toolkit
				synced 2025-06-26 18:18:24 +00:00 
			
		
		
		
	Merge branch 'goimports-local' into 'main'
run goimports -local against the entire codebase See merge request nvidia/container-toolkit/container-toolkit!512
This commit is contained in:
		
						commit
						ffe7ed313a
					
				| @ -15,6 +15,10 @@ linters: | ||||
|     - staticcheck | ||||
|     - unconvert | ||||
| 
 | ||||
| linters-settings: | ||||
|   goimports: | ||||
|     local-prefixes: github.com/NVIDIA/nvidia-container-toolkit | ||||
| 
 | ||||
| issues: | ||||
|   exclude-rules: | ||||
|   # Exclude the gocritic dupSubExpr issue for cgo files. | ||||
|  | ||||
							
								
								
									
										5
									
								
								Makefile
									
									
									
									
									
								
							
							
						
						
									
										5
									
								
								Makefile
									
									
									
									
									
								
							| @ -78,6 +78,11 @@ fmt: | ||||
| 	go list -f '{{.Dir}}' $(MODULE)/... \
 | ||||
| 		| xargs gofmt -s -l -w | ||||
| 
 | ||||
| # Apply goimports -local github.com/NVIDIA/container-toolkit to the codebase
 | ||||
| goimports: | ||||
| 	go list -f {{.Dir}} $(MODULE)/... \
 | ||||
| 		| xargs goimports -local $(MODULE) -w | ||||
| 
 | ||||
| golangci-lint: | ||||
| 	golangci-lint run ./... | ||||
| 
 | ||||
|  | ||||
| @ -9,9 +9,10 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	"golang.org/x/mod/semver" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -5,8 +5,9 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| ) | ||||
| 
 | ||||
| func TestGetNvidiaConfig(t *testing.T) { | ||||
|  | ||||
| @ -21,8 +21,9 @@ import ( | ||||
| 	"os" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| ) | ||||
| 
 | ||||
| func TestGetHookConfig(t *testing.T) { | ||||
|  | ||||
| @ -11,11 +11,12 @@ import ( | ||||
| 	"strings" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/modifier" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/modifier" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -17,11 +17,12 @@ | ||||
| package cdi | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/cdi/generate" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/cdi/list" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/cdi/transform" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -22,14 +22,15 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 	cdi "tags.cncf.io/container-device-interface/pkg/parser" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/platform-support/tegra/csv" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 	cdi "tags.cncf.io/container-device-interface/pkg/parser" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -19,9 +19,10 @@ package list | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -21,11 +21,12 @@ import ( | ||||
| 	"io" | ||||
| 	"os" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	transformroot "github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform/root" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -17,9 +17,10 @@ | ||||
| package transform | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/cdi/transform/root" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -23,11 +23,12 @@ import ( | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	createdefault "github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/config/create-default" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/config/flags" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -19,10 +19,11 @@ package defaultsubcommand | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/config/flags" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -25,9 +25,10 @@ import ( | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -22,12 +22,13 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup/symlinks" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/platform-support/tegra/csv" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	chmod "github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/hook/chmod" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	symlinks "github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/hook/create-symlinks" | ||||
| 	ldcache "github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/hook/update-ldcache" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type hookCommand struct { | ||||
|  | ||||
| @ -23,10 +23,11 @@ import ( | ||||
| 	"strings" | ||||
| 	"syscall" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -17,8 +17,9 @@ | ||||
| package info | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -19,6 +19,8 @@ package main | ||||
| import ( | ||||
| 	"os" | ||||
| 
 | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/cdi" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/hook" | ||||
| @ -26,7 +28,6 @@ import ( | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/runtime" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/system" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 
 | ||||
| 	cli "github.com/urfave/cli/v2" | ||||
| ) | ||||
|  | ||||
| @ -17,9 +17,10 @@ | ||||
| package runtime | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/runtime/configure" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type runtimeCommand struct { | ||||
|  | ||||
| @ -21,6 +21,7 @@ import ( | ||||
| 	"path/filepath" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvpci" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info/proc/devices" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/nvcaps" | ||||
|  | ||||
| @ -24,11 +24,12 @@ import ( | ||||
| 	"strings" | ||||
| 	"syscall" | ||||
| 
 | ||||
| 	"github.com/fsnotify/fsnotify" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/system/nvdevices" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/system/nvmodules" | ||||
| 	"github.com/fsnotify/fsnotify" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"golang.org/x/sys/unix" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	"golang.org/x/sys/unix" | ||||
| ) | ||||
| 
 | ||||
| type nodeLister interface { | ||||
|  | ||||
| @ -19,10 +19,11 @@ package createdevicenodes | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/system/nvdevices" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/system/nvmodules" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -19,9 +19,10 @@ package createdevicenodes | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/ldcache" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -17,11 +17,12 @@ | ||||
| package system | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	devchar "github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/system/create-dev-char-symlinks" | ||||
| 	devicenodes "github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/system/create-device-nodes" | ||||
| 	ldcache "github.com/NVIDIA/nvidia-container-toolkit/cmd/nvidia-ctk/system/print-ldcache" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| type command struct { | ||||
|  | ||||
| @ -22,10 +22,11 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| ) | ||||
| 
 | ||||
| func TestCharDevices(t *testing.T) { | ||||
|  | ||||
| @ -19,9 +19,10 @@ package discover | ||||
| import ( | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| ) | ||||
| 
 | ||||
| func TestIPCMounts(t *testing.T) { | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 
 | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| ) | ||||
| 
 | ||||
|  | ||||
| @ -17,9 +17,10 @@ | ||||
| package edits | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| ) | ||||
| 
 | ||||
| type device discover.Device | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| ) | ||||
| 
 | ||||
| func TestDeviceToSpec(t *testing.T) { | ||||
|  | ||||
| @ -19,12 +19,13 @@ package edits | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	ociSpecs "github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| ) | ||||
| 
 | ||||
| type edits struct { | ||||
|  | ||||
| @ -19,8 +19,9 @@ package edits | ||||
| import ( | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| ) | ||||
| 
 | ||||
| func TestFromDiscovererAllowsMountsToIterate(t *testing.T) { | ||||
|  | ||||
| @ -17,9 +17,10 @@ | ||||
| package edits | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| ) | ||||
| 
 | ||||
| type hook discover.Hook | ||||
|  | ||||
| @ -17,9 +17,10 @@ | ||||
| package edits | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| ) | ||||
| 
 | ||||
| type mount discover.Mount | ||||
|  | ||||
| @ -20,6 +20,7 @@ import ( | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/info" | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| ) | ||||
|  | ||||
| @ -19,10 +19,11 @@ package info | ||||
| import ( | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| ) | ||||
| 
 | ||||
| func TestResolveAutoMode(t *testing.T) { | ||||
|  | ||||
| @ -23,9 +23,10 @@ import ( | ||||
| 	"strings" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| ) | ||||
| 
 | ||||
| func TestLocate(t *testing.T) { | ||||
|  | ||||
| @ -22,9 +22,10 @@ import ( | ||||
| 	"strings" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/ldcache" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/ldcache" | ||||
| ) | ||||
| 
 | ||||
| func TestLDCacheLocator(t *testing.T) { | ||||
|  | ||||
| @ -19,10 +19,11 @@ package cdi | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| ) | ||||
| 
 | ||||
| type builder struct { | ||||
|  | ||||
| @ -20,10 +20,11 @@ import ( | ||||
| 	"errors" | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| ) | ||||
| 
 | ||||
| // fromRegistry represents the modifications performed using a CDI registry.
 | ||||
|  | ||||
| @ -19,9 +19,10 @@ package cdi | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| ) | ||||
| 
 | ||||
| // fromCDISpec represents the modifications performed from a raw CDI spec.
 | ||||
|  | ||||
| @ -19,11 +19,12 @@ package modifier | ||||
| import ( | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| ) | ||||
| 
 | ||||
| func TestNewCSVModifier(t *testing.T) { | ||||
|  | ||||
| @ -19,11 +19,12 @@ package modifier | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| ) | ||||
| 
 | ||||
| type discoverModifier struct { | ||||
|  | ||||
| @ -20,10 +20,11 @@ import ( | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| ) | ||||
| 
 | ||||
| func TestDiscoverModifier(t *testing.T) { | ||||
|  | ||||
| @ -19,8 +19,9 @@ package modifier | ||||
| import ( | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config/image" | ||||
| ) | ||||
| 
 | ||||
| func TestGraphicsModifier(t *testing.T) { | ||||
|  | ||||
| @ -19,10 +19,11 @@ package modifier | ||||
| import ( | ||||
| 	"path/filepath" | ||||
| 
 | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| ) | ||||
| 
 | ||||
| // nvidiaContainerRuntimeHookRemover is a spec modifer that detects and removes inserted nvidia-container-runtime hooks
 | ||||
|  | ||||
| @ -17,8 +17,9 @@ | ||||
| package modifier | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| ) | ||||
| 
 | ||||
| type list struct { | ||||
|  | ||||
| @ -19,9 +19,10 @@ package modifier | ||||
| import ( | ||||
| 	"path/filepath" | ||||
| 
 | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/oci" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| ) | ||||
| 
 | ||||
| // NewStableRuntimeModifier creates an OCI spec modifier that inserts the NVIDIA Container Runtime Hook into an OCI
 | ||||
|  | ||||
| @ -22,10 +22,11 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| ) | ||||
| 
 | ||||
| type testConfig struct { | ||||
|  | ||||
| @ -19,8 +19,9 @@ package oci | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| ) | ||||
| 
 | ||||
| // SpecModifier defines an interface for modifying a (raw) OCI spec
 | ||||
|  | ||||
| @ -4,8 +4,9 @@ | ||||
| package oci | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	"sync" | ||||
| 
 | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| ) | ||||
| 
 | ||||
| // Ensure, that SpecMock does implement Spec.
 | ||||
|  | ||||
| @ -5,8 +5,9 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| ) | ||||
| 
 | ||||
| func TestMaintainSpec(t *testing.T) { | ||||
|  | ||||
| @ -20,8 +20,9 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| ) | ||||
| 
 | ||||
| func TestGetFileList(t *testing.T) { | ||||
|  | ||||
| @ -20,11 +20,12 @@ import ( | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/platform-support/tegra/csv" | ||||
| ) | ||||
|  | ||||
| @ -26,8 +26,9 @@ import ( | ||||
| 	"strconv" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| ) | ||||
| 
 | ||||
| // Logger adds a way to manage output to a log file to a logrus.Logger
 | ||||
|  | ||||
| @ -22,9 +22,10 @@ import ( | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| ) | ||||
| 
 | ||||
| // Run is an entry point that allows for idiomatic handling of errors
 | ||||
|  | ||||
| @ -24,11 +24,12 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| 	"github.com/opencontainers/runtime-spec/specs-go" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/test" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	"errors" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info/proc/devices" | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info/proc/devices" | ||||
| ) | ||||
| 
 | ||||
| func TestCreateControlDevices(t *testing.T) { | ||||
|  | ||||
| @ -17,8 +17,9 @@ | ||||
| package nvdevices | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"golang.org/x/sys/unix" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| ) | ||||
| 
 | ||||
| //go:generate moq -stub -out mknod_mock.go . mknoder
 | ||||
|  | ||||
| @ -19,8 +19,9 @@ package containerd | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| 	"github.com/pelletier/go-toml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| ) | ||||
| 
 | ||||
| // ConfigV1 represents a version 1 containerd config
 | ||||
|  | ||||
| @ -19,8 +19,9 @@ package containerd | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| 	"github.com/pelletier/go-toml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| ) | ||||
| 
 | ||||
| // AddRuntime adds a runtime to the containerd config
 | ||||
|  | ||||
| @ -17,9 +17,10 @@ | ||||
| package containerd | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/pelletier/go-toml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| 	"github.com/pelletier/go-toml" | ||||
| ) | ||||
| 
 | ||||
| // Config represents the containerd config
 | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 
 | ||||
| 	"github.com/pelletier/go-toml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| 	"github.com/pelletier/go-toml" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -19,8 +19,9 @@ package crio | ||||
| import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| 	"github.com/pelletier/go-toml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| ) | ||||
| 
 | ||||
| // Config represents the cri-o config
 | ||||
|  | ||||
| @ -20,8 +20,9 @@ import ( | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/pelletier/go-toml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| ) | ||||
| 
 | ||||
| type builder struct { | ||||
|  | ||||
| @ -18,9 +18,10 @@ package nvcdi | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -23,12 +23,13 @@ import ( | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| 	"golang.org/x/sys/unix" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup/cuda" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup/root" | ||||
| 	"golang.org/x/sys/unix" | ||||
| ) | ||||
| 
 | ||||
| // NewDriverDiscoverer creates a discoverer for the libraries and binaries associated with a driver installation.
 | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	"errors" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 
 | ||||
| 	testlog "github.com/sirupsen/logrus/hooks/test" | ||||
| ) | ||||
| 
 | ||||
|  | ||||
| @ -24,12 +24,13 @@ import ( | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info/drm" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| ) | ||||
| 
 | ||||
| // GetGPUDeviceSpecs returns the CDI device specs for the full GPU represented by 'device'.
 | ||||
|  | ||||
| @ -20,11 +20,12 @@ import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| ) | ||||
| 
 | ||||
| type gdslib nvcdilib | ||||
|  | ||||
| @ -20,12 +20,13 @@ import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/platform-support/tegra" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| ) | ||||
| 
 | ||||
| type csvlib nvcdilib | ||||
|  | ||||
| @ -21,10 +21,11 @@ import ( | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| ) | ||||
| 
 | ||||
| type nvmllib nvcdilib | ||||
|  | ||||
| @ -20,10 +20,11 @@ import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| ) | ||||
| 
 | ||||
| type wsllib nvcdilib | ||||
|  | ||||
| @ -22,6 +22,7 @@ import ( | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/info" | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup/root" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/platform-support/tegra/csv" | ||||
|  | ||||
| @ -22,12 +22,13 @@ import ( | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup/cuda" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| ) | ||||
| 
 | ||||
| type managementlib nvcdilib | ||||
|  | ||||
| @ -21,12 +21,13 @@ import ( | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/nvcaps" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| ) | ||||
| 
 | ||||
| // GetMIGDeviceSpecs returns the CDI device specs for the full GPU represented by 'device'.
 | ||||
|  | ||||
| @ -20,11 +20,12 @@ import ( | ||||
| 	"fmt" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/discover" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/edits" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/spec" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| ) | ||||
| 
 | ||||
| type mofedlib nvcdilib | ||||
|  | ||||
| @ -4,8 +4,9 @@ | ||||
| package nvcdi | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| 	"sync" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| ) | ||||
| 
 | ||||
| // Ensure, that nvmlUUIDerMock does implement nvmlUUIDer.
 | ||||
|  | ||||
| @ -19,6 +19,7 @@ package nvcdi | ||||
| import ( | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvlib/device" | ||||
| 	"github.com/NVIDIA/go-nvlib/pkg/nvml" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/logger" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| ) | ||||
|  | ||||
| @ -20,10 +20,11 @@ import ( | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/parser" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| ) | ||||
| 
 | ||||
| type builder struct { | ||||
|  | ||||
| @ -17,8 +17,9 @@ | ||||
| package noop | ||||
| 
 | ||||
| import ( | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| ) | ||||
| 
 | ||||
| type noop struct{} | ||||
|  | ||||
| @ -20,8 +20,9 @@ import ( | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| ) | ||||
| 
 | ||||
| // containerRootTransformer transforms the roots of container paths in a CDI spec.
 | ||||
|  | ||||
| @ -20,8 +20,9 @@ import ( | ||||
| 	"fmt" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| 	"tags.cncf.io/container-device-interface/specs-go" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform" | ||||
| ) | ||||
| 
 | ||||
| // hostRootTransformer transforms the roots of host paths in a CDI spec.
 | ||||
|  | ||||
| @ -21,10 +21,11 @@ import ( | ||||
| 	"os" | ||||
| 	"os/exec" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container/operator" | ||||
| 	"github.com/sirupsen/logrus" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container/operator" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -20,10 +20,11 @@ import ( | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine/containerd" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container" | ||||
| 	"github.com/pelletier/go-toml" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine/containerd" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container" | ||||
| ) | ||||
| 
 | ||||
| func TestUpdateV1ConfigDefaultRuntime(t *testing.T) { | ||||
|  | ||||
| @ -20,10 +20,11 @@ import ( | ||||
| 	"fmt" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine/containerd" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container" | ||||
| 	"github.com/pelletier/go-toml" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine/containerd" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -20,11 +20,12 @@ import ( | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 
 | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	cli "github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine/containerd" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	cli "github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -21,13 +21,14 @@ import ( | ||||
| 	"os" | ||||
| 	"path/filepath" | ||||
| 
 | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	cli "github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine/crio" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/ocihook" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	cli "github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -20,11 +20,12 @@ import ( | ||||
| 	"fmt" | ||||
| 	"os" | ||||
| 
 | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	cli "github.com/urfave/cli/v2" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/info" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine/docker" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	cli "github.com/urfave/cli/v2" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
| @ -20,9 +20,10 @@ import ( | ||||
| 	"encoding/json" | ||||
| 	"testing" | ||||
| 
 | ||||
| 	"github.com/stretchr/testify/require" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/config/engine/docker" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/tools/container" | ||||
| 	"github.com/stretchr/testify/require" | ||||
| ) | ||||
| 
 | ||||
| func TestUpdateConfigDefaultRuntime(t *testing.T) { | ||||
|  | ||||
| @ -23,15 +23,16 @@ import ( | ||||
| 	"path/filepath" | ||||
| 	"strings" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/system/nvdevices" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi" | ||||
| 	transformroot "github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform/root" | ||||
| 	toml "github.com/pelletier/go-toml" | ||||
| 	log "github.com/sirupsen/logrus" | ||||
| 	"github.com/urfave/cli/v2" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/cdi" | ||||
| 	"tags.cncf.io/container-device-interface/pkg/parser" | ||||
| 
 | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/config" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/internal/system/nvdevices" | ||||
| 	"github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi" | ||||
| 	transformroot "github.com/NVIDIA/nvidia-container-toolkit/pkg/nvcdi/transform/root" | ||||
| ) | ||||
| 
 | ||||
| const ( | ||||
|  | ||||
		Loading…
	
		Reference in New Issue
	
	Block a user