Add the memory addressing mode as an attribute for each device

On coherent-memory systems, the possible modes are:
  * HMM  - (Hardware Memory Management mode)
  * ATS  - (Address Translation Service mode)
  * None - (Supported by the platform but currently inactive)

On others, its explicitly set to empty to indicate that it is not supported by the platform

Signed-off-by: Shiva Krishna, Merla <smerla@nvidia.com>

Author: shivamerla

cmd/gpu-kubelet-plugin/deviceinfo.go

@@ -53,6 +53,7 @@ type GpuInfo struct {
 	pcieRootAttr          *deviceattribute.DeviceAttribute
 	migProfiles           []*MigProfileInfo
 	Health                HealthStatus
+	addressingMode        string
 }
 
 type MigDeviceInfo struct {
@@ -139,6 +140,9 @@ func (d *GpuInfo) GetDevice() resourceapi.Device {
 			"pcieBusID": {
 				StringValue: &d.pcieBusID,
 			},
+			"addressingMode": {
+				StringValue: &d.addressingMode,
+			},
 		},
 		Capacity: map[resourceapi.QualifiedName]resourceapi.DeviceCapacity{
 			"memory": {
@@ -189,6 +193,9 @@ func (d *MigDeviceInfo) GetDevice() resourceapi.Device {
 			"pcieBusID": {
 				StringValue: &d.pcieBusID,
 			},
+			"addressingMode": {
+				StringValue: &d.parent.addressingMode,
+			},
 		},
 		Capacity: map[resourceapi.QualifiedName]resourceapi.DeviceCapacity{
 			"multiprocessors": {

cmd/gpu-kubelet-plugin/nvlib.go

@@ -304,6 +304,17 @@ func (l deviceLib) getGpuInfo(index int, device nvdev.Device) (*GpuInfo, error)
 		return nil, fmt.Errorf("error getting PCIe bus ID for device %d: %w", index, err)
 	}
 
+	// Get the memory-addressing mode supported by the device.
+	// On coherent-memory systems, the possible modes are:
+	//   - HMM  (Hardware Memory Management)
+	//   - ATS  (Address Translation Service)
+	//   - None (Supported by the platform but currently inactive)
+	//   - ""   (Not supported by the platform)
+	addressingMode, err := device.GetAddressingModeAsString()
+	if err != nil {
+		return nil, fmt.Errorf("error getting addressing mode for device %d: %w", index, err)
+	}
+
 	var pcieRootAttr *deviceattribute.DeviceAttribute
 	if attr, err := deviceattribute.GetPCIeRootAttributeByPCIBusID(pcieBusID); err == nil {
 		pcieRootAttr = &attr
@@ -379,6 +390,7 @@ func (l deviceLib) getGpuInfo(index int, device nvdev.Device) (*GpuInfo, error)
 		pcieRootAttr:          pcieRootAttr,
 		migProfiles:           migProfiles,
 		Health:                Healthy,
+		addressingMode:        addressingMode,
 	}
 
 	return gpuInfo, nil