NewAllocator creates a new Allocator using the given allocation policy.

NewBestEffortAllocator creates a new Allocator using the BestEffort allocation policy.

NewBestEffortPolicy creates a new BestEffortPolicy.

NewDevices creates a list of Devices from all available nvml.Devices using the specified options.

NewDeviceSet creates a new DeviceSet.

NewDevicesFrom creates a list of Devices from the specific set of GPU uuids passed in.

NewSimpleAllocator creates a new Allocator using the Simple allocation policy.

NewSimplePolicy creates a new SimplePolicy.

NewStaticDGX1Policy creates a new StaticDGX1Policy for gpuType.

NewStaticDGX2Policy creates a new StaticDGX2Policy.

WithDeviceLib provides an option to set the library used for device enumeration.

WithNvmlLib provides an option to set the nvml library.

Pascal GPUs.

Valid GPUTypes.

Allocator defines the primary object for allocating and freeing the available GPUs on a node.

Device represents a GPU device as reported by NVML, including all of its Point-to-Point link information.

P2PLink represents a Point-to-Point link between two GPU devices.

Policy defines an interface for pluggable allocation policies to be added to an Allocator.

DeviceList stores an ordered list of devices.

DeviceSet is used to hold and manipulate a set of unique GPU devices.

GPUType represents the valid set of GPU types a Static DGX policy can be created for.

Option defines a type for functional options for constructing device lists.