Use nvinfo package from go-nvlib

Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar
2022-08-08 17:11:42 +02:00
parent ffd6ec3c54
commit a9dc6550d5
3 changed files with 6 additions and 69 deletions

View File

@@ -18,15 +18,15 @@ package modifier
import (
"github.com/NVIDIA/nvidia-container-toolkit/internal/discover"
"github.com/NVIDIA/nvidia-container-toolkit/internal/info"
"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup"
"github.com/NVIDIA/nvidia-container-toolkit/internal/oci"
"github.com/sirupsen/logrus"
"gitlab.com/nvidia/cloud-native/go-nvlib/pkg/nvinfo"
)
// NewTegraPlatformFiles creates a modifier to inject the Tegra platform files into a container.
func NewTegraPlatformFiles(logger *logrus.Logger) (oci.SpecModifier, error) {
isTegra, _ := info.IsTegraSystem()
isTegra, _ := nvinfo.IsTegraSystem()
if !isTegra {
return nil, nil
}