mirror of
https://github.com/NVIDIA/nvidia-container-toolkit
synced 2024-11-22 16:29:18 +00:00
583793b7ae
This change adds a Requirements abstraction that can be used to check an images' NVIDIA_REQUIRE_* envvars against the host properties such as CUDA version or architecture. Signed-off-by: Evan Lezar <elezar@nvidia.com>
70 lines
2.2 KiB
Go
70 lines
2.2 KiB
Go
/**
|
|
# Copyright (c) 2022, NVIDIA CORPORATION. All rights reserved.
|
|
#
|
|
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
# you may not use this file except in compliance with the License.
|
|
# You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
# See the License for the specific language governing permissions and
|
|
# limitations under the License.
|
|
**/
|
|
|
|
package requirements
|
|
|
|
import (
|
|
"github.com/NVIDIA/nvidia-container-toolkit/internal/requirements/constraints"
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
// Requirements represents a collection of requirements that can be compared to properties
|
|
type Requirements struct {
|
|
logger *logrus.Logger
|
|
requirements []string
|
|
properties map[string]constraints.Property
|
|
}
|
|
|
|
// New creates a new set of requirements
|
|
func New(logger *logrus.Logger, requirements []string) *Requirements {
|
|
r := Requirements{
|
|
logger: logger,
|
|
requirements: requirements,
|
|
properties: map[string]constraints.Property{
|
|
// Set up the supported properties. These are overridden with actual values.
|
|
CUDA: constraints.NewVersionProperty(CUDA, ""),
|
|
ARCH: constraints.NewVersionProperty(ARCH, ""),
|
|
DRIVER: constraints.NewVersionProperty(DRIVER, ""),
|
|
BRAND: constraints.NewStringProperty(BRAND, ""),
|
|
},
|
|
}
|
|
|
|
return &r
|
|
}
|
|
|
|
// AddVersionProperty adds the specified property (name, value pair) to the requirements
|
|
func (r *Requirements) AddVersionProperty(name string, value string) {
|
|
r.properties[name] = constraints.NewVersionProperty(name, value)
|
|
}
|
|
|
|
// AddStringProperty adds the specified property (name, value pair) to the requirements
|
|
func (r *Requirements) AddStringProperty(name string, value string) {
|
|
r.properties[name] = constraints.NewStringProperty(name, value)
|
|
}
|
|
|
|
// Assert checks the specified requirements
|
|
func (r Requirements) Assert() error {
|
|
if len(r.requirements) == 0 {
|
|
return nil
|
|
}
|
|
|
|
c, err := constraints.New(r.logger, r.requirements, r.properties)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return c.Assert()
|
|
}
|