add warning when set memory limits valide is pre-volta arch

Signed-off-by: rongfu.leng <rongfu.leng@daocloud.io>
NVIDIA · Oct 2, 2024 · 5978161 · 5978161
1 parent 37d2609
commit 5978161
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cmd/mps-control-daemon/mps/daemon.go b/cmd/mps-control-daemon/mps/daemon.go
@@ -254,6 +254,10 @@ func (m *Daemon) perDevicePinnedDeviceMemoryLimits() map[string]string {
 	totalMemoryInBytesPerDevice := make(map[string]uint64)
 	replicasPerDevice := make(map[string]uint64)
 	for _, device := range m.Devices() {
+		if isVoltaDevice := (*mpsDevice)(device).isAtLeastVolta(); !isVoltaDevice {
+			klog.Warningf("Skipping memory limits for pre-volta device %s", device.ID)
+			continue
+		}
 		index := device.Index
 		totalMemoryInBytesPerDevice[index] = device.TotalMemory
 		replicasPerDevice[index] += 1